<?php

class tldDB extends SQLite3 {
    function __construct() {
        $this->open('./db/allTlds.db'); // 顶级域名数据库
    }
}

class icpDB extends SQLite3 {
    function __construct() {
        $this->open('./db/icpTlds.db'); // 顶级域名数据库
    }
}

class extractDomain {
    private function getAllTlds() { // 获取所有顶级域 含次级域
        $db = new tldDB;
        $res = $db->query('SELECT tld FROM `tlds`;');
        while ($row = $res->fetchArray(SQLITE3_ASSOC)) {
            $tlds[] = $row['tld'];
        }
        $tlds = array_merge($tlds, $this->getIcpTlds());
        $tlds = array_unique($tlds);
        foreach ($tlds as $tld) {
            echo $tld . PHP_EOL;
        }
        return $tlds; // Unicode字符使用Punycode编码
    }

    private function getIcpTlds() { // 获取所有可ICP备案的顶级域
        $db = new icpDB;
        $punycode = new Punycode();
        $res = $db->query('SELECT tld FROM `tlds`;');
        while ($row = $res->fetchArray(SQLITE3_ASSOC)) {
            $tlds[] = $punycode->encode($row['tld']); // 转为Punycode编码
        }
        return $tlds; // Unicode字符使用Punycode编码
    }

    private function isDomain($domain) { // 检测是否为域名
        preg_match('/^(?=^.{3,255}$)[a-zA-Z0-9][-a-zA-Z0-9]{0,62}(\.[a-zA-Z0-9][-a-zA-Z0-9]{0,62})+$/', $domain, $match);
        return (count($match) != 0);
    }

    private function getDomain($url) { // 从URL获取域名
        $url = preg_replace('/^[\w]+:\/\//', '', $url); // 去除协议字段
        $url = explode('?', $url)[0]; // 去除请求参数内容
        $domain = explode('/', $url)[0]; // 分离域名
        return (new Punycode)->encode($domain);
    }

    private function getTld($domain) { // 搜索域名TLD
        $tlds = $this->getAllTlds(); // 获取TLD列表
        foreach ($tlds as $tld) {
            if (substr($domain, -strlen($tld)) === $tld) { // 匹配测试
                $target[] = $tld;
            }
        }
        if (count($target) === 0) {
            return ''; // 匹配不到TLD
        };
        $type = 0;
        foreach ($target as $tld) { // 遍历可能的结果
            $num = substr_count($tld, '.');
            if ($type < $num) { // 获取.个数最多的
                $type = $num;
                $result = $tld;
            }
        }
        return $result; // 返回网站顶级域名
    }

    private function getSite($domain, $tld) { // 获取主域名
        $domain = explode('.', $domain);
        $num = count($domain) - substr_count($tld, '.');
        return $domain[$num - 1] . $tld;
    }

    public function analyse($url) { // 分析域名信息
        $domain = $this->getDomain($url);
        if (!$this->isDomain($domain)) { // 域名不合格
            return array();
        }
        $tld = $this->getTld($domain);
        if ($tld == '') { // 匹配不到TLD
            return array(
                'domain' => $domain
            );
        }
        return array(
            'domain' => $domain,
            'tld' => $tld,
            'site' => $this->getSite($domain, $tld)
        );
    }
}

?>