/*
 * Copyright (c) 2024 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import * as fs from 'node:fs'
import * as path from 'node:path'
import { exit } from 'node:process'

const STANDARD_LIBRARIES = [
    'lib.d.ts',
    'lib.dom.iterable.d.ts',
    'lib.dom.d.ts',
    'lib.es2015.symbol.wellknown.d.ts',
    'lib.es2015.symbol.d.ts',
    'lib.es2015.reflect.d.ts',
    'lib.es2015.proxy.d.ts',
    'lib.es2015.promise.d.ts',
    'lib.es2015.iterable.d.ts',
    'lib.es2015.generator.d.ts',
    'lib.es2015.d.ts',
    'lib.es2015.core.d.ts',
    'lib.es2015.collection.d.ts',
    'lib.es2016.full.d.ts',
    'lib.es2016.d.ts',
    'lib.es2016.array.include.d.ts',
    'lib.es2017.typedarrays.d.ts',
    'lib.es2017.string.d.ts',
    'lib.es2017.sharedmemory.d.ts',
    'lib.es2017.object.d.ts',
    'lib.es2017.intl.d.ts',
    'lib.es2017.full.d.ts',
    'lib.es2017.d.ts',
    'lib.es2018.regexp.d.ts',
    'lib.es2018.promise.d.ts',
    'lib.es2018.intl.d.ts',
    'lib.es2018.full.d.ts',
    'lib.es2018.d.ts',
    'lib.es2018.asynciterable.d.ts',
    'lib.es2018.asyncgenerator.d.ts',
    'lib.es2019.symbol.d.ts',
    'lib.es2019.string.d.ts',
    'lib.es2019.object.d.ts',
    'lib.es2019.intl.d.ts',
    'lib.es2019.full.d.ts',
    'lib.es2019.d.ts',
    'lib.es2019.array.d.ts',
    'lib.es2020.symbol.wellknown.d.ts',
    'lib.es2020.string.d.ts',
    'lib.es2020.sharedmemory.d.ts',
    'lib.es2020.promise.d.ts',
    'lib.es2020.number.d.ts',
    'lib.es2020.intl.d.ts',
    'lib.es2020.full.d.ts',
    'lib.es2020.date.d.ts',
    'lib.es2020.d.ts',
    'lib.es2020.bigint.d.ts',
    'lib.es2021.weakref.d.ts',
    'lib.es2021.string.d.ts',
    'lib.es2021.promise.d.ts',
    'lib.es2021.intl.d.ts',
    'lib.es2021.full.d.ts',
    'lib.es2021.d.ts',
    'lib.es2022.string.d.ts',
    'lib.es2022.sharedmemory.d.ts',
    'lib.es2022.object.d.ts',
    'lib.es2022.intl.d.ts',
    'lib.es2022.full.d.ts',
    'lib.es2022.error.d.ts',
    'lib.es2022.d.ts',
    'lib.es2022.array.d.ts',
    'lib.es6.d.ts',
    'lib.es5.d.ts',
    'lib.esnext.weakref.d.ts',
    'lib.esnext.string.d.ts',
    'lib.esnext.promise.d.ts',
    'lib.esnext.intl.d.ts',
    'lib.esnext.full.d.ts',
    'lib.esnext.d.ts',
    'lib.scripthost.d.ts',
    'lib.webworker.iterable.d.ts',
    'lib.webworker.importscripts.d.ts',
    'lib.webworker.d.ts'
];

function reportErrorAndExit(msg) {
    console.log(msg);
    exit(1);
}

function copyDirectorySync(srcPath, destPath, options) {
    const entries = fs.readdirSync(srcPath, { withFileTypes: true });

    if (!fs.existsSync(destPath)) {
        fs.mkdirSync(destPath, { recursive: true });
    }

    for (const entry of entries) {
        const srcEntry = path.join(srcPath, entry.name);
        const destEntry = path.join(destPath, entry.name);

        if (typeof options.filter === 'function' && !options.filter(srcEntry, destEntry)) {
            continue;
        }

        if (entry.isDirectory()) {
            if (options.recursive) {
                copyDirectorySync(srcEntry, destEntry, options);
            }
        } else {
            fs.copyFileSync(srcEntry, destEntry);
        }
    }
}

function copyTypescriptLibDeclarationsToDist() {
    const typescript_lib = path.join('node_modules', 'typescript', 'lib');
    const dist = path.join('dist');

    if (!fs.existsSync(typescript_lib)) {
        reportErrorAndExit('Failed to locate ' + typescript_lib + ' directory');
    }
    if (!fs.existsSync(dist)) {
        reportErrorAndExit('Failed to locate ' + dist + ' directory');
    }

    const srcPath = path.resolve(typescript_lib);
    const distPath = path.resolve(dist);

    copyDirectorySync(srcPath, distPath, {
        recursive: true,
        filter: (src, dest) => {
            if (!src?.length) return false;
            const stats = fs.statSync(src);
            if (stats.isDirectory()) {
                const normalizedSrc = path.resolve(src);
                return normalizedSrc === path.resolve(srcPath) || normalizedSrc.endsWith(srcPath);
            }
            return STANDARD_LIBRARIES.includes(path.basename(src));
        }
    });
}

copyTypescriptLibDeclarationsToDist();