/*
 * Copyright (c) 2022-2023 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 ts from 'typescript';
import { logTscDiagnostic } from './Utils';
import { consoleLog } from './TypeScriptLinter';
import { CommandLineOptions } from './CommandLineOptions';
import { LintOptions } from './LintOptions';

export function compile(options: LintOptions, extraOptions?: any): ts.Program {
  const createProgramOptions = formTscOptions(options.cmdOptions, extraOptions);
  const program = ts.createProgram(createProgramOptions);
  // Log Tsc errors if needed
  if (options.cmdOptions.logTscErrors) {
    const diagnostics = ts.getPreEmitDiagnostics(program);
    logTscDiagnostic(diagnostics, consoleLog);
    diagnostics.forEach((diagnostic) => {
      if (diagnostic.file && diagnostic.start) {
        const { line, character } = ts.getLineAndCharacterOfPosition(diagnostic.file, diagnostic.start);
        const message = ts.flattenDiagnosticMessageText(diagnostic.messageText, '\n');
        consoleLog(`${diagnostic.file.fileName} (${line + 1}, ${character + 1}): ${message}`);
      } else {
        consoleLog(ts.flattenDiagnosticMessageText(diagnostic.messageText, '\n'));
      }
    });
  }
  return program;
}

function formTscOptions(cmdOptions: CommandLineOptions, extraOptions?: any): ts.CreateProgramOptions {
  if (cmdOptions.parsedConfigFile) {
    let options: ts.CreateProgramOptions = {
      rootNames: cmdOptions.parsedConfigFile.fileNames,
      options: cmdOptions.parsedConfigFile.options,
      projectReferences: cmdOptions.parsedConfigFile.projectReferences,
      configFileParsingDiagnostics: ts.getConfigFileParsingDiagnostics(cmdOptions.parsedConfigFile),
    };

    if (extraOptions) {
      options.options = Object.assign(options.options, extraOptions);
    }
    
    return options;
  }

  let options: ts.CreateProgramOptions = {
    rootNames: cmdOptions.inputFiles,
    options: {
      target: ts.ScriptTarget.Latest,
      module: ts.ModuleKind.CommonJS,
      allowJs: true,
      checkJs: true,
    },
  };

  if (extraOptions) {
    options.options = Object.assign(options.options, extraOptions);
  }

  return options;
}