<?php

declare(strict_types=1);

namespace app\common\command\build;

use app\common\tools\PathTools;
use app\common\tools\phpparser\MinifyPrinterTools;
use app\common\tools\phpparser\NodeVisitorTools;
use League\Flysystem\Adapter\Local;
use League\Flysystem\Filesystem;
use PhpParser\Node;
use PhpParser\Node\Arg;
use PhpParser\Node\Const_;
use PhpParser\Node\Expr\BinaryOp\Concat;
use PhpParser\Node\Expr\ConstFetch;
use PhpParser\Node\Expr\FuncCall;
use PhpParser\Node\Expr\Include_;
use PhpParser\Node\Name;
use PhpParser\Node\Scalar\MagicConst\Dir;
use PhpParser\Node\Scalar\String_;
use PhpParser\Node\Stmt;
use PhpParser\Node\Stmt\Class_;
use PhpParser\Node\Stmt\ClassLike;
use PhpParser\Node\Stmt\Expression;
use PhpParser\Node\Stmt\Namespace_;
use PhpParser\Node\Stmt\Use_;
use PhpParser\Node\Stmt\UseUse;
use PhpParser\NodeTraverser;
use PhpParser\NodeVisitorAbstract;
use PhpParser\ParserFactory;
use app\common\tools\phpparser\PrettyPrinterTools as Standard;
use PhpParser\Comment;
use think\console\Command;
use think\console\Input;
use think\console\input\Argument;
use think\console\input\Option;
use think\console\Output;
use think\facade\App;
use think\facade\Config;
use think\facade\View;
use think\helper\Str;

class Dist extends Command
{

    public $packList = [];
    public $usedClass = [];
    public $hasClass = [];
    public $newPackList = [];
    public $classExtendList = [];

    protected $distPath;

    public $constDirList = [];

    /**
     * @var FileSystem
     */
    protected $appFilesystem;

    /**
     * @var FileSystem
     */
    protected $distFilesystem;

    protected function configure()
    {
        // 指令配置
        $this->setName('build:dist')
            ->setDescription('the build:dist command');
    }

    protected function execute(Input $input, Output $output)
    {
        // 指令输出
        $output->writeln('build:dist');

        $app_path = App::getRootPath();

        $dist_path = $app_path . 'dist';

        PathTools::intiDir($dist_path . '.temp');


        $this->distPath = $dist_path;

        $app_adapter = new Local($app_path);

        $app_filesystem = new Filesystem($app_adapter);

        $this->appFilesystem = $app_filesystem;

        $dist_adapter = new Local($dist_path);

        $dist_filesystem = new Filesystem($dist_adapter);

        $this->distFilesystem = $dist_filesystem;

        $list_dist = $dist_filesystem->listContents();

        foreach ($list_dist as  $file_info) {
            if ($file_info['type'] == 'file') {
                $dist_filesystem->delete($file_info['path']);
            } else {
                $dist_filesystem->deleteDir($file_info['path']);
            }
        }

        $list_content = $app_filesystem->listContents('', true);
        foreach ($list_content as  $file_info) {
            if ($file_info['type'] == 'dir') {
                continue;
            }

            $file_path = $file_info['path'];

            if ($this->isSkip($file_path)) {
                continue;
            }

            $file_content = $app_filesystem->read($file_path);
            $path_info = pathinfo($file_path);

            if (!$this->isIgnored($file_path)) {

                if (isset($path_info['extension'])) {
                    if ($path_info['extension'] == 'php') {
                        $file_content = $this->buildPhpContent($file_content, $file_path);
                    }
                }
            }
            if (!is_null($file_content)) {

                $dist_filesystem->write($file_path, $file_content);
            }
        }


        $lib_php_file = '/lib.' . uniqid() . '.php';
        $lib_php_path = $this->distPath . '/lib' . $lib_php_file;

        $this->buildMainClassFile($lib_php_path);

        $lib_function_file = '/lib.' . uniqid() . '.php';
        $lib_function_path = $this->distPath . '/lib' . $lib_function_file;
        $this->buildFunctionFile($lib_function_path);

        $this->buildMigrateFile();

        $lib_dir_const_file = '/lib.' . uniqid() . '.php';
        $lib_dir_const_path = $this->distPath . '/lib' . $lib_dir_const_file;
        $this->buildDirConstFile($lib_dir_const_path);


        PathTools::intiDir($lib_php_path);

        $this->buildIncludeIndexFile([
            $lib_dir_const_file,
            $lib_php_file,
            $lib_function_file,
        ]);


        $this->buildAllAppDir();


        $output->info('打包完成');
    }

    public function buildMigrateFile()
    {
        $database_dir =  'database';

        $list_files = $this->appFilesystem->listContents($database_dir, true);

        $parser = (new ParserFactory)->create(ParserFactory::PREFER_PHP7);


        $prettyPrinter = new  MinifyPrinterTools();

        foreach ($list_files as  $item_file) {
            if ($item_file['type'] == 'dir') {
                continue;
            }

            $file_content = $this->appFilesystem->read($item_file['path']);

            $file_stmts = $parser->parse($file_content);

            $traverser = new NodeTraverser();
            $traverser->addVisitor(new class($item_file['path'], $this) extends NodeVisitorAbstract
            {
                protected $name;
                protected $mainClass;
                public function __construct($name, $main_class)
                {
                    $this->name = $name;
                    $this->mainClass = $main_class;
                }

                public function leaveNode(Node $node)
                {
                    $name = $this->name;
                    if ($node instanceof Dir) {
                        // Clean out the function body
                        $const_key = 'dirconst' . uniqid();

                        $this->mainClass->constDirList[$const_key] = dirname($name);
                        return new ConstFetch(new Name($const_key));
                    }

                    if ($node->hasAttribute('comments')) {
                        $node->setAttribute('comments', []);
                    }
                }
            });


            $file_stmts = $traverser->traverse($file_stmts);

            $function_code = $prettyPrinter->prettyPrintFile($file_stmts);

            $this->distFilesystem->put($item_file['path'], $function_code);
        }
    }


    /**
     * 打包标准类文件（核心）
     *
     * @param string $lib_php_path
     * @return void
     */
    public function buildMainClassFile($lib_php_path)
    {
        $prettyPrinter = new  Standard();
        // 根据调用次数排序
        $this->parsePackList();

        $stmts = [];
        foreach ($this->packList as $class_name => $class_item) {
            foreach ($class_item['stmts'] as $stmt_item) {
                $stmts[] = $stmt_item;
            }
        }


        $newCode = $prettyPrinter->prettyPrintFile($stmts);

        file_put_contents($lib_php_path, $newCode);
    }

    /**
     * 统一声明目录魔术常量
     *
     * @param string $lib_dir_const_path
     * @return void
     */
    public function buildDirConstFile($lib_dir_const_path)
    {
        $dir_const_stmts = [];

        foreach ($this->constDirList as $const_key => $const_value) {

            $dir_const_stmts[] = new Expression(new FuncCall(
                new Name('define'),
                [
                    new Arg(new String_($const_key)),
                    new Arg(new Concat(
                        new Dir(),
                        new String_('/../' . $const_value)
                    )),
                ]
            ));
        }

        $prettyPrinter = new  MinifyPrinterTools();

        $dir_const_code = $prettyPrinter->prettyPrintFile($dir_const_stmts);

        file_put_contents($lib_dir_const_path, $dir_const_code);
    }

    /**
     * 打包类库文件
     *
     * @param string $lib_function_path
     * @return void
     */
    public function buildFunctionFile($lib_function_path)
    {
        $function_path = Config::get('dist.function_path', []);

        $function_stmts = [];

        $parser = (new ParserFactory)->create(ParserFactory::PREFER_PHP7);

        foreach ($function_path as  $function_file) {

            $function_file_path = App::getRootPath() . '' . $function_file;


            $function_content = file_get_contents($function_file_path);

            $stmts = $parser->parse($function_content);

            $traverser = new NodeTraverser();
            $traverser->addVisitor(new class($function_file, $this) extends NodeVisitorAbstract
            {
                protected $name;
                protected $mainClass;
                public function __construct($name, $main_class)
                {
                    $this->name = $name;
                    $this->mainClass = $main_class;
                }

                public function leaveNode(Node $node)
                {
                    $name = $this->name;
                    if ($node instanceof Dir) {
                        // Clean out the function body
                        $const_key = 'dirconst' . uniqid();

                        $this->mainClass->constDirList[$const_key] = dirname($name);
                        return new ConstFetch(new Name($const_key));
                    }

                    if ($node->hasAttribute('comments')) {
                        $node->setAttribute('comments', []);
                    }
                }
            });


            $stmts = $traverser->traverse($stmts);

            foreach ($stmts as  $stmt_item) {
                $function_stmts[] = $stmt_item;
            }
        }


        $prettyPrinter = new  MinifyPrinterTools();

        $function_code = $prettyPrinter->prettyPrintFile($function_stmts);

        file_put_contents($lib_function_path, $function_code);
    }

    /**
     * 创建多应用空间
     *
     * @return void
     */
    public function buildAllAppDir()
    {
        $list = $this->appFilesystem->listContents('');

        foreach ($list as  $item) {
            if ($item['type'] != 'dir') {
                continue;
            }

            if ($this->distFilesystem->has($item['path'])) {
                continue;
            }

            $this->distFilesystem->createDir($item['path']);
        }
    }

    /**
     * 生成引入文件
     *
     * @param string[] $files
     * @return void
     */
    public function buildIncludeIndexFile($files)
    {
        $file_stmts = [];

        $file_stmts[] = new Expression(new FuncCall(
            new Name('define'),
            [
                new Arg(new String_('ULTHON_ADMIN_BUILD_DIST')),
                new Arg(new String_('1'))
            ]
        ));

        foreach ($files as  $file_name) {
            $file_stmts[] = new Expression(new Include_(new Concat(new Dir, new String_($file_name)), Include_::TYPE_REQUIRE_ONCE));
        }
        $prettyPrinter = new  MinifyPrinterTools();


        $newCode = $prettyPrinter->prettyPrintFile($file_stmts);

        file_put_contents($this->distPath . '/lib/index.php', $newCode);
    }

    /**
     * 后处理核心类库
     *
     * @return void
     */
    public function parsePackList()
    {

        foreach ($this->packList as $class_name => $class_item) {

            $extend_name_orginal = $this->usedClass[$class_item['extend_name']] ?? '';

            $this->classExtendList[$class_name] = $extend_name_orginal;
        }

        foreach ($this->packList as $class_name => $class_item) {
            $this->insertToNewPackList($class_name, $class_item);
        }

        $this->packList = $this->newPackList;

        $this->newPackList = [];
    }

    /**
     * 调整类加载顺序
     *
     * @param string $class_name
     * @param array $class_item
     * @return void
     */
    public function insertToNewPackList($class_name, $class_item)
    {
        if (isset($this->newPackList[$class_name])) {
            return;
        }

        $extend_name = $this->classExtendList[$class_name] ?? '';

        if (!empty($extend_name)) {
            if (isset($this->packList[$extend_name])) {
                $this->insertToNewPackList($extend_name, $this->packList[$extend_name]);
            }
        } else {
            $extend_name = $class_item['extend_name'];


            if (!empty($extend_name)) {
                $try_namse_extend_name = $class_item['namespace_name'] . '\\' . $extend_name;

                if (isset($this->packList[$try_namse_extend_name])) {
                    $this->insertToNewPackList($try_namse_extend_name, $this->packList[$try_namse_extend_name]);
                }
            }
        }


        $this->newPackList[$class_name] = $class_item;
    }

    /**
     * 
     *
     * @param Node\Stmt[] $stmts
     * @return void
     */
    public function checkStmts($stmts, $name)
    {

        $class_count = 0;
        $namsepace_count = 0;
        foreach ($stmts as  $stmt_item) {
            if ($stmt_item instanceof Namespace_) {
                $namsepace_count++;
                foreach ($stmt_item->stmts as  $class_item) {

                    if ($class_item instanceof ClassLike) {
                        $class_count++;
                    }
                }
            }
        }

        if ($namsepace_count !== 1) {
            throw new \Exception('一个文件只能有一个命名空间：' . $name);
        }
        if ($class_count !== 1) {
            throw new \Exception('一个文件只能有一个类：' . $name);
        }
    }

    /**
     * 扫描解析代码
     *
     * @param string $content
     * @param string $name
     * @return void
     */
    public function buildPhpContent($content, $name)
    {

        $parser = (new ParserFactory)->create(ParserFactory::PREFER_PHP7);

        $stmts = $parser->parse($content);

        $this->scanForMagicConstDir($stmts, $name);

        $this->checkStmts($stmts, $name);


        $stmts = $this->parseStmts($stmts, $name);

        $class_name = null;
        $extend_name = null;
        $namespace_name = null;
        foreach ($stmts as  $stmt_item) {
            if ($stmt_item instanceof Namespace_) {
                $namespace_name = $stmt_item->name->toString();
                foreach ($stmt_item->stmts as  $class_item) {

                    if ($class_item instanceof Use_) {
                        foreach ($class_item->uses as $stmt_use) {
                            $use_class = $stmt_use->name->toString();
                            $alias = $use_class;

                            if (!empty($stmt_use->alias)) {
                                $alias = $stmt_use->alias->toString();
                            }
                            $this->usedClass[$alias] = $use_class;
                        }
                    }

                    if ($class_item instanceof ClassLike) {
                        $class_name = $namespace_name . '\\' . $class_item->name->toString();

                        if ($class_item instanceof Class_) {

                            if (!empty($class_item->extends)) {
                                $extend_name = $class_item->extends->toString();
                            }
                        }
                    }
                }
            }
        }

        $this->hasClass[] = $class_name;
        $pack_item = [
            'file_name' => $name,
            'class_name' => $class_name,
            'extend_name' => $extend_name,
            'namespace_name' => $namespace_name,
            'stmts' => $stmts
        ];

        $this->packList[$class_name] = $pack_item;

        return null;
    }


    /**
     * 遍历扫描代码
     *
     * @param Node\Stmt[]|null $stmts
     * @param string $name
     * @return Node\Stmt[]
     */
    public function parseStmts($stmts, $name)
    {

        $traverser = new NodeTraverser();

        $node_visitor = new NodeVisitorTools($this, $name);

        $traverser->addVisitor($node_visitor);

        $stmts = $traverser->traverse($stmts);

        return $stmts;
    }

    /**
     * 扫描魔术变量
     *
     * @param Node\Stmt[]|null $stmts
     * @param string $name
     * @return void
     */
    public function scanForMagicConstDir($stmts, $name)
    {

        $traverser = new NodeTraverser();
        $traverser->addVisitor(new class($name, $this) extends NodeVisitorAbstract
        {
            protected $name;
            protected $mainClass;
            public function __construct($name, $main_class)
            {
                $this->name = $name;
                $this->mainClass = $main_class;
            }

            public function leaveNode(Node $node)
            {
                $name = $this->name;
                if ($node instanceof Dir) {
                    // Clean out the function body
                    $const_key = 'dirconst' . uniqid();

                    $this->mainClass->constDirList[$const_key] = dirname($name);
                    return new ConstFetch(new Name($const_key));
                }
            }
        });

        $stmts = $traverser->traverse($stmts);
    }

    public function isSkip($path)
    {

        $system_skip_path = ['/app\/common.php/'];

        $skip_path = Config::get('dist.skip_path', []);

        $skip_path = array_merge($system_skip_path, $skip_path);

        foreach ($skip_path as $rule) {
            if (preg_match($rule, $path)) {
                return true;
            }
        }

        return false;
    }

    public function isIgnored($path)
    {

        $ignore_path = Config::get('dist.ignore_path', []);

        foreach ($ignore_path as  $rule) {

            if (preg_match($rule, $path)) {
                return true;
            }
        }

        return false;
    }
}