#!/usr/bin/env python
# -*- coding: utf-8 -*-
# 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.

# Description
#
# This script is invoked by the build system and does not need to be executed directly by the developer.
# First, it checks if --release is provided as an argument. This is the only allowed type for stateMgmt that is included in the build image.
# It then verifies if the node_modules folder exists. If not, npm install is executed.
# Afterward, npm run build_release is performed, which also generates generateGni.js
# The files_to_watch.gni file contains a list of input files from tsconfig.base.json.
# When any of these files are modified, the build system triggers this script to regenerate stateMgmt.js.

import os
import sys
import time
import shutil
import subprocess
import re


def main(argv):
    if len(argv) < 4:
        print("Usage: python script.py <path_to_project> <path_to_node_modules> <js_output_path> [--release]")
        sys.exit(1)

    # Determine if it's a release build
    is_release = len(argv) > 4 and argv[4] == "--release"
    print(f": is_release set to {is_release}")

    # If not a release build, print an error and exit
    if not is_release:
        print("ERROR: state_mgmt/build.py requires a release build. Exiting...")
        sys.exit(1)

    project_path = os.path.abspath(argv[1])
    node_modules_path = os.path.abspath(argv[2])
    js_output_path = os.path.abspath(argv[3])

    output_file = os.path.join(js_output_path, "stateMgmt.js")

    print(f"StateMgmt: Changing directory to {project_path}. Out dir = {js_output_path}")
    os.chdir(project_path)

    # Check if `node_modules` exists. If yes skip npm install
    if not os.path.exists(node_modules_path):

        print(f"StateMgmt: node_modules directory not found at {node_modules_path}, running npm install")
        secondary_npm_registry = "https://cmc.centralrepo.rnd.huawei.com/artifactory/api/npm/npm-central-repo/"
        try:
            subprocess.check_call(["npm", "install"])
        except subprocess.CalledProcessError as e:
            print(f"Error: npm install failed with exit code {e.returncode}. Retry...")
            print(e.stderr)
            try:
                subprocess.check_call(["npm", "install", "--registry", secondary_npm_registry, "--loglevel=verbose"])
            except subprocess.CalledProcessError as e2:
                print(e2.stderr)
                sys.exit(e2.returncode)
    else:
        print(f"StateMgmt: node_modules directory exists at {node_modules_path}")

    # Determine the npm script to run. Currently only build_release supported.
    script = "build_release"
    print(f"StateMgmt: Running npm script '{script}'")

    try:
        subprocess.check_call(["npm", "run", script])
    except subprocess.CalledProcessError as e:
        print(f"Error: npm run {script} failed with exit code {e.returncode}.")
        print("Error: State managemant build failed. See log output for failing .ts files")
        sys.exit(e.returncode)

    source_folder = "distRelease"
    built_file = os.path.join(project_path, source_folder, "stateMgmt.js")

    if not os.path.exists(built_file):
        print(f"Error: Built file not found at {built_file}")
        sys.exit(1)

    # Ensure the output directory exists
    if not os.path.exists(js_output_path):
        os.makedirs(js_output_path)

    print(f"StateMgmt: Copying {built_file} to {output_file}")

    try:
        shutil.copy(built_file, output_file)
        print(f"StateMgmt: File successfully copied to {output_file}")
    except Exception as e:
        print(f"Error: Failed to copy file: {e}")
        sys.exit(1)

if __name__ == '__main__':
    start_time = time.time()
    main(sys.argv)
    end_time = time.time()
    elapsed_time = end_time - start_time
    print(f"StateMgmt: build time: {elapsed_time:.2f} seconds")