Class ProgramProvider

client/src/program.ts:16–92

const ABI_FILE_SUFFIX = '-abi.json';

export class ProgramProvider implements TreeDataProvider<Function | Program> {
  private _onDidChangeTreeData: EventEmitter<Program | undefined | void> =
    new EventEmitter<Program | undefined | void>();
  readonly onDidChangeTreeData: Event<Program | undefined | void> =
    this._onDidChangeTreeData.event;

  constructor(
    private workspaceRoot: string | undefined,
    readonly type: ProgramType
  ) {}

  refresh(): void {
    this._onDidChangeTreeData.fire();
  }

  getTreeItem(element: TreeItem): TreeItem {
    return element;
  }

  getChildren(contract?: Function | Program): ProviderResult<Program[]> {
    if (!this.workspaceRoot) {
      window.showInformationMessage('No contract in empty workspace');
      return Promise.resolve([]);
    }

    return contract
      ? Promise.resolve(contract['children'])
      : this.getPrograms();
  }

  /**
   * Reads contracts from the ABIs.
   * @returns array of contracts
   */
  private async getPrograms(): Promise<Program[]> {
    const allFiles = getAllFiles(this.workspaceRoot, []);
    const swayFilePaths = allFiles.filter(file => file.endsWith('.sw'));
    const forcTomlFilePaths = allFiles.filter(file =>
      file.endsWith('Forc.toml')
    );
    const abiFilePaths = allFiles.filter(file =>
      file.endsWith(ABI_FILE_SUFFIX)
    );
    const programs = abiFilePaths.map(filepath => {
      const contractName = path
        .parse(filepath)
        .base.replace(ABI_FILE_SUFFIX, '');

      // This is assuming that there is only one sway contract in the same directory as a Forc.toml
      const forcFilePath = forcTomlFilePaths.find(forcFilePath =>
        fs.readFileSync(forcFilePath).toString().includes(contractName)
      );

      // Find out the program type of the sway file
      const swayFilePath = swayFilePaths.find(swayFilePath => {
        return swayFilePath.startsWith(path.parse(forcFilePath).dir);
      });
      const swayFileMatchesType = fs