<?php

namespace app\lib\dns;

use app\lib\DnsInterface;

class cloudflare implements DnsInterface
{
    private $Email;
    private $ApiKey;
    private $auth;
    private $baseUrl = 'https://api.cloudflare.com/client/v4';
    private $error;
    private $domain;
    private $domainid;
    private $proxy;

    function __construct($config)
    {
        $this->Email = $config['email'];
        $this->ApiKey = $config['apikey'];
        $this->domain = $config['domain'];
        $this->domainid = $config['domainid'];
        $this->proxy = isset($config['proxy']) ? $config['proxy'] == 1 : false;
        $this->auth = isset($config['auth']) ? intval($config['auth']) : (preg_match('/^[0-9a-f]+$/i', $this->ApiKey) ? 0 : 1);
    }

    /**
     * 从 Cloudflare API 返回的完整域名中提取子域名（主机记录）
     * 兼容 Emoji/IDN 域名：Cloudflare API 返回 Punycode 格式，数据库存储 UTF-8
     */
    private function extractName($fullName)
    {
        $domainAscii = idn_to_ascii($this->domain, IDNA_DEFAULT, INTL_IDNA_VARIANT_UTS46);
        if ($domainAscii === false) $domainAscii = $this->domain;

        if ($fullName === $domainAscii || $fullName === $this->domain) {
            return '@';
        }
        if (str_ends_with($fullName, '.' . $domainAscii)) {
            return substr($fullName, 0, -(strlen($domainAscii) + 1));
        }
        if (str_ends_with($fullName, '.' . $this->domain)) {
            return substr($fullName, 0, -(strlen($this->domain) + 1));
        }
        return $fullName;
    }

    public function getError()
    {
        return $this->error;
    }

    public function check()
    {
        if ($this->getDomainList() !== false) {
            return true;
        }
        return false;
    }

    //获取域名列表
    public function getDomainList($KeyWord = null, $PageNumber = 1, $PageSize = 20)
    {
        $param = ['page' => $PageNumber, 'per_page' => $PageSize];
        if (!empty($KeyWord)) {
            $param['name'] = $KeyWord;
        }
        $data = $this->send_reuqest('GET', '/zones', $param);
        if ($data) {
            $list = [];
            foreach ($data['result'] as $row) {
                $list[] = [
                    'DomainId' => $row['id'],
                    'Domain' => $row['name'],
                    'RecordCount' => 0,
                ];
            }
            return ['total' => $data['result_info']['total_count'], 'list' => $list];
        }
        return false;
    }

    //获取解析记录列表
    public function getDomainRecords($PageNumber = 1, $PageSize = 20, $KeyWord = null, $SubDomain = null, $Value = null, $Type = null, $Line = null, $Status = null)
    {
        if (!isNullOrEmpty($Value)) $KeyWord = $Value;
        $param = ['type' => $Type, 'search' => $KeyWord, 'page' => $PageNumber, 'per_page' => $PageSize];
        if (!isNullOrEmpty($SubDomain)) {
            $domainAscii = idn_to_ascii($this->domain, IDNA_DEFAULT, INTL_IDNA_VARIANT_UTS46) ?: $this->domain;
            if ($SubDomain == '@') $SubDomain = $domainAscii;
            else $SubDomain .= '.' . $domainAscii;
            $param['name'] = $SubDomain;
        }
        if (!isNullOrEmpty($Line)) {
            $param['proxied'] = $Line == '1' ? 'true' : 'false';
        }
        $data = $this->send_reuqest('GET', '/zones/'.$this->domainid.'/dns_records', $param);
        if ($data) {
            $list = [];
            foreach ($data['result'] as $row) {
                $name = $this->extractName($row['name']);
                $status = str_ends_with($name, '_pause') ? '0' : '1';
                $name = $name == '__root__' ? '@' : $name;
                $name = $status == '0' ? substr($name, 0, -6) : $name;
                if ($row['type'] == 'SRV' && isset($row['priority'])) {
                    $row['content'] = $row['priority'] . ' ' . $row['content'];
                }
                $list[] = [
                    'RecordId' => $row['id'],
                    'Domain' => $this->domain,
                    'Name' => $name,
                    'Type' => $row['type'],
                    'Value' => $row['content'],
                    'Line' => $row['proxied'] ? '1' : '0',
                    'TTL' => $row['ttl'],
                    'MX' => isset($row['priority']) ? $row['priority'] : null,
                    'Status' => $status,
                    'Weight' => null,
                    'Remark' => $row['comment'],
                    'UpdateTime' => $row['modified_on'],
                ];
            }
            return ['total' => $data['result_info']['total_count'], 'list' => $list];
        }
        return false;
    }

    //获取子域名解析记录列表
    public function getSubDomainRecords($SubDomain, $PageNumber = 1, $PageSize = 20, $Type = null, $Line = null)
    {
        return $this->getDomainRecords($PageNumber, $PageSize, null, $SubDomain, null, $Type, $Line);
    }

    //获取解析记录详细信息
    public function getDomainRecordInfo($RecordId)
    {
        $data = $this->send_reuqest('GET', '/zones/'.$this->domainid.'/dns_records/'.$RecordId);
        if ($data) {
            $name = $this->extractName($data['result']['name']);
            $status = str_ends_with($name, '_pause') ? '0' : '1';
            $name = $status == '0' ? substr($name, 0, -6) : $name;
            $name = $name == '__root__' ? '@' : $name;
            if ($data['result']['type'] == 'SRV' && isset($data['result']['priority'])) {
                $data['result']['content'] = $data['result']['priority'] . ' ' . $data['result']['content'];
            }
            return [
                'RecordId' => $data['result']['id'],
                'Domain' => $this->domain,
                'Name' => $name,
                'Type' => $data['result']['type'],
                'Value' => $data['result']['content'],
                'Line' => $data['result']['proxied'] ? '1' : '0',
                'TTL' => $data['result']['ttl'],
                'MX' => isset($data['result']['priority']) ? $data['result']['priority'] : null,
                'Status' => $status,
                'Weight' => null,
                'Remark' => $data['result']['comment'],
                'UpdateTime' => $data['result']['modified_on'],
            ];
        }
        return false;
    }

    //添加解析记录
    public function addDomainRecord($Name, $Type, $Value, $Line = '0', $TTL = 600, $MX = 1, $Weight = null, $Remark = null)
    {
        $param = ['name' => $Name, 'type' => $this->convertType($Type), 'content' => $Value, 'proxied' => $Line == '1', 'ttl' => intval($TTL), 'comment' => $Remark];
        if ($Type == 'MX') $param['priority'] = intval($MX);
        if ($Type == 'CAA' || $Type == 'SRV') {
            unset($param['content']);
            $param['data'] = $this->convertValue($Value, $Type);
        }
        $data = $this->send_reuqest('POST', '/zones/'.$this->domainid.'/dns_records', $param);
        return is_array($data) ? $data['result']['id'] : false;
    }

    //修改解析记录
    public function updateDomainRecord($RecordId, $Name, $Type, $Value, $Line = '0', $TTL = 600, $MX = 1, $Weight = null, $Remark = null)
    {
        $param = ['name' => $Name, 'type' => $this->convertType($Type), 'content' => $Value, 'proxied' => $Line == '1', 'ttl' => intval($TTL), 'comment' => $Remark];
        if ($Type == 'MX') $param['priority'] = intval($MX);
        if ($Type == 'CAA' || $Type == 'SRV') {
            unset($param['content']);
            $param['data'] = $this->convertValue($Value, $Type);
        }
        $data = $this->send_reuqest('PATCH', '/zones/'.$this->domainid.'/dns_records/'.$RecordId, $param);
        return is_array($data);
    }

    //修改解析记录备注
    public function updateDomainRecordRemark($RecordId, $Remark)
    {
        return false;
    }

    //删除解析记录
    public function deleteDomainRecord($RecordId)
    {
        $data = $this->send_reuqest('DELETE', '/zones/'.$this->domainid.'/dns_records/'.$RecordId);
        return is_array($data);
    }

    //设置解析记录状态
    public function setDomainRecordStatus($RecordId, $Status)
    {
        $info = $this->getDomainRecordInfo($RecordId);
        $Name = $Status == '1' ? str_replace('_pause', '', $info['Name']) : $info['Name'] . '_pause';
        // @ 作为特殊字符不能设置为解析, 故设置暂停解析的时候, 替换为 __root__
        if ($Name == '__root__') {
            $Name = '@';
        } elseif ($Name == '@_pause') {
            $Name = '__root___pause';
        }
        return $this->updateDomainRecord($RecordId, $Name, $info['Type'], $info['Value'], $info['Line'], $info['TTL'], $info['MX'], $info['Weight'], $info['Remark']);
    }

    //获取解析记录操作日志
    public function getDomainRecordLog($PageNumber = 1, $PageSize = 20, $KeyWord = null, $StartDate = null, $endDate = null)
    {
        return false;
    }

    //获取解析线路列表
    public function getRecordLine()
    {
        return ['0' => ['name' => '仅DNS', 'parent' => null], '1' => ['name' => '已代理', 'parent' => null]];
    }

    //获取域名信息
    public function getDomainInfo()
    {
        $data = $this->send_reuqest('GET', '/zones/'.$this->domainid);
        if ($data) {
            return $data['result'];
        }
        return false;
    }

    //获取域名最低TTL
    public function getMinTTL()
    {
        return false;
    }

    public function addDomain($Domain)
    {
        $param = ['name' => $Domain];
        $data = $this->send_reuqest('POST', '/zones', $param);
        if ($data) {
            return ['id' => $data['result']['id'], 'name' => $data['result']['name']];
        }
        return false;
    }

    private function convertType($type)
    {
        $convert_dict = ['REDIRECT_URL' => 'URI', 'FORWARD_URL' => 'URI'];
        if (array_key_exists($type, $convert_dict)) {
            return $convert_dict[$type];
        }
        return $type;
    }

    private function convertValue($value, $type)
    {
        if ($type == 'SRV') {
            $arr = explode(' ', $value);
            if (count($arr) > 3) {
                $data = [
                    'priority' => intval($arr[0]),
                    'weight' => intval($arr[1]),
                    'port' => intval($arr[2]),
                    'target' => $arr[3],
                ];
            } else {
                $data = [
                    'weight' => intval($arr[0]),
                    'port' => intval($arr[1]),
                    'target' => $arr[2],
                ];
            }
        } elseif ($type == 'CAA') {
            $arr = explode(' ', $value);
            $data = [
                'flags' => intval($arr[0]),
                'tag' => $arr[1],
                'value' => trim($arr[2], '"'),
            ];
        }
        return $data;
    }

    private function send_reuqest($method, $path, $params = null)
    {
        $url = $this->baseUrl . $path;

        if ($this->auth == 0) {
            $headers = [
                'X-Auth-Email' => $this->Email,
                'X-Auth-Key' => $this->ApiKey,
            ];
        } else {
            $headers = [
                'Authorization' => 'Bearer ' . $this->ApiKey,
            ];
        }

        $body = '';
        if ($method == 'GET' || $method == 'DELETE') {
            if ($params) {
                $url .= '?' . http_build_query($params);
            }
        } else {
            $body = json_encode($params);
            $headers['Content-Type'] = 'application/json';
        }

        try {
            $response = http_request($url, $body, null, null, $headers, $this->proxy, $method);
        } catch (\Exception $e) {
            $this->setError($e->getMessage());
            return false;
        }

        $arr = json_decode($response['body'], true);
        if ($arr) {
            if ($arr['success']) {
                return $arr;
            } else {
                $this->setError(isset($arr['errors'][0]) ? $arr['errors'][0]['message'] : '未知错误');
                return false;
            }
        } else {
            $this->setError('返回数据解析失败');
            return false;
        }
    }

    private function setError($message)
    {
        $this->error = $message;
        //file_put_contents('logs.txt',date('H:i:s').' '.$message."\r\n", FILE_APPEND);
    }
}
