"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    var desc = Object.getOwnPropertyDescriptor(m, k);
    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
      desc = { enumerable: true, get: function() { return m[k]; } };
    }
    Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
    var ownKeys = function(o) {
        ownKeys = Object.getOwnPropertyNames || function (o) {
            var ar = [];
            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
            return ar;
        };
        return ownKeys(o);
    };
    return function (mod) {
        if (mod && mod.__esModule) return mod;
        var result = {};
        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
        __setModuleDefault(result, mod);
        return result;
    };
})();
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.getLayoutFiles = getLayoutFiles;
exports.updateLayouts = updateLayouts;
exports.patchLayoutFile = patchLayoutFile;
const fs_extra_1 = __importDefault(require("fs-extra"));
const path_1 = __importDefault(require("path"));
const parser_1 = require("@babel/parser");
const traverse_1 = __importDefault(require("@babel/traverse"));
const generator_1 = __importDefault(require("@babel/generator"));
const t = __importStar(require("@babel/types"));
async function patchLayoutFile(filePath) {
    const code = await fs_extra_1.default.readFile(filePath, "utf8");
    if (!code.includes("AntdLayout")) {
        return;
    }
    const ast = (0, parser_1.parse)(code, {
        sourceType: "module",
        plugins: [
            "typescript",
            "jsx"
        ]
    });
    let hasImport = false;
    let changed = false;
    (0, traverse_1.default)(ast, {
        ImportDeclaration(path) {
            if (path.node.source.value ===
                "../components/warden-skin") {
                hasImport = true;
            }
        },
        JSXOpeningElement(path) {
            if (!t.isJSXIdentifier(path.node.name) ||
                path.node.name.name !== "AntdLayout") {
                return;
            }
            const hasThemeSkins = path.node.attributes.some(attr => {
                return (t.isJSXAttribute(attr) &&
                    t.isJSXIdentifier(attr.name) &&
                    attr.name.name === "themeSkins");
            });
            if (hasThemeSkins) {
                return;
            }
            path.node.attributes.push(t.jsxAttribute(t.jsxIdentifier("themeSkins"), t.jsxExpressionContainer(t.identifier("wardenSkin"))));
            changed = true;
        }
    });
    if (!changed) {
        return;
    }
    if (!hasImport) {
        ast.program.body.unshift(t.importDeclaration([
            t.importDefaultSpecifier(t.identifier("wardenSkin"))
        ], t.stringLiteral("../components/warden-skin")));
    }
    const output = (0, generator_1.default)(ast, {
        retainLines: true
    }, code);
    await fs_extra_1.default.writeFile(filePath, output.code);
    console.log(`✔ updated ${path_1.default.relative(process.cwd(), filePath)}`);
}
async function updateLayouts(projectRoot) {
    const layoutsDir = path_1.default.join(projectRoot, "src/layouts");
    if (!(await fs_extra_1.default.pathExists(layoutsDir))) {
        return;
    }
    const files = await getLayoutFiles(layoutsDir);
    for (const file of files) {
        await patchLayoutFile(file);
    }
}
async function getLayoutFiles(dir) {
    const result = [];
    const items = await fs_extra_1.default.readdir(dir);
    for (const item of items) {
        const full = path_1.default.join(dir, item);
        const stat = await fs_extra_1.default.stat(full);
        if (stat.isDirectory()) {
            result.push(...await getLayoutFiles(full));
            continue;
        }
        if (/\.(tsx|ts)$/.test(item)) {
            result.push(full);
        }
    }
    return result;
}
