|
| 1 | +package transip |
| 2 | + |
| 3 | +import ( |
| 4 | + "strings" |
| 5 | + |
| 6 | + "github.com/TimothyYe/godns/internal/settings" |
| 7 | + "github.com/TimothyYe/godns/internal/utils" |
| 8 | + log "github.com/sirupsen/logrus" |
| 9 | + "github.com/transip/gotransip/v6" |
| 10 | + "github.com/transip/gotransip/v6/domain" |
| 11 | +) |
| 12 | + |
| 13 | +const defaultTTL int = 60 // 60 seconds. |
| 14 | + |
| 15 | +// DNSProvider struct. |
| 16 | +type DNSProvider struct { |
| 17 | + configuration *settings.Settings |
| 18 | + clientConfig gotransip.ClientConfiguration |
| 19 | +} |
| 20 | + |
| 21 | +// Init passes DNS settings and store it to the provider instance. |
| 22 | +func (provider *DNSProvider) Init(conf *settings.Settings) { |
| 23 | + provider.configuration = conf |
| 24 | + provider.clientConfig = gotransip.ClientConfiguration{ |
| 25 | + AccountName: conf.Email} |
| 26 | + if strings.HasPrefix(conf.LoginToken, "-----BEGIN PRIVATE KEY-----") { // Private Key. |
| 27 | + provider.clientConfig.PrivateKeyReader = strings.NewReader(conf.LoginToken) |
| 28 | + } else { // JWT. |
| 29 | + provider.clientConfig.Token = conf.LoginToken |
| 30 | + } |
| 31 | +} |
| 32 | + |
| 33 | +// UpdateIP updates the IP address of the given subdomain. |
| 34 | +func (provider *DNSProvider) UpdateIP(domainName, subDomainName, ip string) error { |
| 35 | + client, err := gotransip.NewClient(provider.clientConfig) |
| 36 | + if err != nil { |
| 37 | + return err |
| 38 | + } |
| 39 | + domainRepo := domain.Repository{Client: client} |
| 40 | + |
| 41 | + exists, ttl, err := checkExistence(domainRepo, subDomainName, domainName) |
| 42 | + if err != nil { |
| 43 | + return err |
| 44 | + } |
| 45 | + |
| 46 | + if exists { // Update. |
| 47 | + err = domainRepo.UpdateDNSEntry(domainName, domain.DNSEntry{ |
| 48 | + Name: subDomainName, |
| 49 | + Type: provider.setType(), |
| 50 | + Content: ip, |
| 51 | + Expire: ttl}) |
| 52 | + if err != nil { |
| 53 | + log.Error("failed to update domain:", subDomainName) |
| 54 | + return err |
| 55 | + } |
| 56 | + } else { // Create. |
| 57 | + err = domainRepo.AddDNSEntry(domainName, domain.DNSEntry{ |
| 58 | + Name: subDomainName, |
| 59 | + Type: provider.setType(), |
| 60 | + Content: ip, |
| 61 | + Expire: defaultTTL}) |
| 62 | + if err != nil { |
| 63 | + log.Error("failed to add domain:", subDomainName) |
| 64 | + return err |
| 65 | + } |
| 66 | + } |
| 67 | + return nil |
| 68 | +} |
| 69 | + |
| 70 | +func checkExistence(repo domain.Repository, subdomain, domainName string) (bool, int, error) { |
| 71 | + records, err := repo.GetDNSEntries(domainName) |
| 72 | + if err == nil { |
| 73 | + for _, record := range records { |
| 74 | + if record.Name == subdomain { |
| 75 | + return true, record.Expire, nil |
| 76 | + } |
| 77 | + } |
| 78 | + } |
| 79 | + log.Error("failed to get domain:", domainName) |
| 80 | + return false, defaultTTL, err |
| 81 | +} |
| 82 | + |
| 83 | +// defaults to A record (ipv4). |
| 84 | +func (provider *DNSProvider) setType() string { |
| 85 | + if strings.ToUpper(provider.configuration.IPType) == utils.IPV6 { |
| 86 | + return utils.IPTypeAAAA |
| 87 | + } |
| 88 | + return utils.IPTypeA |
| 89 | +} |
0 commit comments