shopware / platform

 * @private

 * @package admin

 */

import { CallExpression, Project, SourceFile, ts } from "ts-morph";

import * as path from "path";

import * as fs from "fs";

const project = new Project({

  skipAddingFilesFromTsConfig: true,

});

// load all the source files from the "src" directory

project.addSourceFilesAtPaths([

  "src/**/*{.js,.ts}",

  "!src/**/*{.spec.js,.spec.vue3.js,.d.ts,.types.ts}",

  "!src/meta/**/*",

]);

type componentInfo = {

  p: string, // path to import

  r: boolean, // needs register

  en?: string, // extends component name

  e?: boolean, // needs extends

}

function isComponentCall(call: CallExpression<ts.CallExpression>, functionString: string): boolean {

    const expression = call.getExpression();

    if (expression === null) {

      return false;

    }

    return [

      `Shopware.Component.${functionString}`,

      `Component.${functionString}`

    ].includes(expression.getText());

}

function throwIfComponentIsAlreadyRegistered(componentName: string, sourceFile: SourceFile): void {

  if (componentImportMap.hasOwnProperty(componentName)) {

    throw new Error(`Component ${componentName} already exists. Found again in file ${sourceFile.getFilePath()}`);

  }

}

function procsessComponentRegisterCall(sourceFile: SourceFile, call: CallExpression<ts.CallExpression>): void {

    const componentName = getComponentNameFromArgumentNumber(call, 1);

    throwIfComponentIsAlreadyRegistered(componentName, sourceFile);

    const secondArgument = call.getArguments()[1];

    // If the secondArgument is a ArrowFunction

    if (secondArgument.getKind() === ts.SyntaxKind.ArrowFunction) {

      // Get the import path inside the ArrowFunction

      const importPath = secondArgument

        .getDescendantsOfKind(ts.SyntaxKind.StringLiteral)[0]

        .getText()

        // remove all single and double quotes

        .replace(/['"]/g, '');

      // Check if the import path is relative

      let aliasPath = '';

      if (importPath.includes('./')) {

        // Get the path of the parent directory of this file

        const parentDirectory = sourceFile.getDirectoryPath();

        // Remove everything before and including "/app/administration/" from the parent directory

        const relativePath = parentDirectory.replace(/.*\/app\/administration\//, '');

        // Combine the relative path with the import path

        aliasPath = path.join(relativePath, importPath);

      } else {

        aliasPath = importPath;

      }

      componentImportMap[componentName] = {

        p: aliasPath,

        r: true,

      };

      return;

    }

    // If the secondArgument is a ObjectLiteralExpression

    if (secondArgument.getKind() === ts.SyntaxKind.ObjectLiteralExpression) {

      // Get the path of the parent directory of this file

      const path = sourceFile.getDirectoryPath().replace(/.*\/app\/administration\//, '');

      componentImportMap[componentName] = {

        p: path,

        r: false,

      };

    }

}

function procsessComponentExtendCall(sourceFile: SourceFile, call: CallExpression<ts.CallExpression>): void {

  const componentName = getComponentNameFromArgumentNumber(call, 1);

  const extendedComponentName = getComponentNameFromArgumentNumber(call, 2);

  throwIfComponentIsAlreadyRegistered(componentName, sourceFile);

  const thirdArgument = call.getArguments()[2];

  // If the thirdArgument is a ArrowFunction

  if (thirdArgument.getKind() === ts.SyntaxKind.ArrowFunction) {

    // Get the import path inside the ArrowFunction

    const importPath = thirdArgument

      .getDescendantsOfKind(ts.SyntaxKind.StringLiteral)[0]

      .getText()

      .replace(/['"]/g, '');

    // Check if the import path is relative

    let aliasPath = '';

    if (importPath.includes('./')) {

      // Get the path of the parent directory of this file

      const parentDirectory = sourceFile.getDirectoryPath();

      // Remove everything before and including "/app/administration/" from the parent directory

      const relativePath = parentDirectory.replace(/.*\/app\/administration\//, '');

      // Combine the relative path with the import path

      aliasPath = path.join(relativePath, importPath);

    } else {

      aliasPath = importPath;

    }

    componentImportMap[componentName] = {

      p: aliasPath,

      r: false,

      en: extendedComponentName,

      e: true,

    };

    return;

  }

  // If the thirdArgument is a ObjectLiteralExpression

  if (thirdArgument.getKind() === ts.SyntaxKind.ObjectLiteralExpression) {

    // Get the path of the parent directory of this file

    const path = sourceFile.getDirectoryPath().replace(/.*\/app\/administration\//, '');

    componentImportMap[componentName] = {

      p: path,

      r: false,

      en: extendedComponentName,

      e: false,

    };

  }

}

const componentImportMap: {[key: string]: componentInfo} = {};

for (const sourceFile of project.getSourceFiles()) {

  // print current file to console for progress tracking

  console.log(sourceFile.getFilePath());

  // collect all "Shopware.Component.register" or "Component.register" calls inside the file

  sourceFile.getDescendantsOfKind(

    ts.SyntaxKind.CallExpression,

  ).forEach(call => {

      if (isComponentCall(call, 'register')) {

        procsessComponentRegisterCall(sourceFile, call);

      }

      if (isComponentCall(call, 'extend')) {

        procsessComponentExtendCall(sourceFile, call);

      }

    }

  );

}

// Write output to file

const filestring = `/* eslint-disable */\n\nexport default ${JSON.stringify(componentImportMap)};`