models.go

package westcn

import (
	"fmt"
	"time"

	"github.com/libdns/libdns"
)

// APIResponse API响应结构体
// Result: 200为成功，其他均为失败
// Msg: 错误或提示信息
// https://www.west.cn/CustomerCenter/doc/apiv2.html#12u3001u8eabu4efdu9a8cu8bc10a3ca20id3d12u3001u8eabu4efdu9a8cu8bc13e203ca3e56
type APIResponse struct {
	Result    int    `json:"result,omitempty"`
	ClientID  string `json:"clientid,omitempty"`
	Msg       string `json:"msg,omitempty"`
	ErrorCode int    `json:"errcode,omitempty"`
}

type RecordIDResponse struct {
	APIResponse
	Data RecordID `json:"data,omitempty"`
}

type RecordsResponse struct {
	APIResponse
	Data Records `json:"data,omitempty"`
}

type RecordID struct {
	ID int `json:"id,omitempty"`
}

type Records struct {
	Records []Record `json:"items,omitempty"`
}

type Record struct {
	ID    int    `json:"id,omitempty"`
	Item  string `json:"item,omitempty"` // 主机名称，获取记录时返回
	Host  string `json:"host,omitempty"` // 主机名称，添加记录时需要
	Type  string `json:"type,omitempty"`
	Value string `json:"value,omitempty"`
	TTL   int    `json:"ttl,omitempty"` // 60~86400 seconds
	Level int    `json:"level,omitempty"`
}

func (r Record) libdnsRecord(zone string) (libdns.Record, error) {
	// format host name
	if r.Host == "" {
		r.Host = r.Item
	}
	// format MX record
	if r.Type == "MX" {
		r.Value = fmt.Sprintf("%d %s", r.Level, r.Value)
	}
	return libdns.RR{
		Name: libdns.RelativeName(r.Host, zone),
		TTL:  time.Duration(r.TTL) * time.Second,
		Type: r.Type,
		Data: r.Value,
	}.Parse()
}

func westcnRecord(zone string, r libdns.Record) (Record, error) {
	rr := r.RR()
	westcnRec := Record{
		Host:  rr.Name,
		Type:  rr.Type,
		TTL:   int(rr.TTL.Seconds()),
		Value: rr.Data,
	}
	// Set default values for TTL and Level if not provided
	if westcnRec.TTL <= 0 {
		westcnRec.TTL = 600
	}
	if westcnRec.Level <= 0 {
		westcnRec.Level = 10
	}
	return westcnRec, nil
}