/**
 * @license
 * Copyright Google LLC All Rights Reserved.
 *
 * Use of this source code is governed by an MIT-style license that can be
 * found in the LICENSE file at https://angular.dev/license
 */
import ts from 'typescript';

import {getFileSystem, PathManipulation} from '../../file_system';
import {TestFile} from '../../file_system/testing';
import {makeProgram} from '../../testing';

/**
 * Construct a TS program consisting solely of an import graph, from a string-based representation
 * of the graph.
 *
 * The `graph` string consists of semicolon separated files, where each file is specified
 * as a name and (optionally) a list of comma-separated imports or exports. For example:
 *
 * "a:b,c;b;c"
 *
 * specifies a program with three files (a.ts, b.ts, c.ts) where a.ts imports from both b.ts and
 * c.ts.
 *
 * A more complicated example has a dependency from b.ts to c.ts: "a:b,c;b:c;c".
 *
 * A * preceding a file name in the list of imports indicates that the dependency should be an
 * "export" and not an "import" dependency. For example:
 *
 * "a:*b,c;b;c"
 *
 * represents a program where a.ts exports from b.ts and imports from c.ts.
 *
 * An import can be suffixed with ! to make it a type-only import.
 */
export function makeProgramFromGraph(
  fs: PathManipulation,
  graph: string,
): {
  program: ts.Program;
  host: ts.CompilerHost;
  options: ts.CompilerOptions;
} {
  const files: TestFile[] = graph.split(';').map((fileSegment) => {
    const [name, importList] = fileSegment.split(':');
    const contents =
      (importList ? importList.split(',') : [])
        .map((i) => {
          if (i.startsWith('*')) {
            const sym = i.slice(1);
            return `export {${sym}} from './${sym}';`;
          } else if (i.endsWith('!')) {
            const sym = i.slice(0, -1);
            return `import type {${sym}} from './${sym}';`;
          } else {
            return `import {${i}} from './${i}';`;
          }
        })
        .join('\n') + `export const ${name} = '${name}';\n`;
    return {
      name: fs.resolve(`/${name}.ts`),
      contents,
    };
  });
  return makeProgram(files);
}

export function importPath(files: ts.SourceFile[]): string {
  const fs = getFileSystem();
  return files.map((sf) => fs.basename(sf.fileName).replace('.ts', '')).join(',');
}
