unity-builder/src/model/cli/shared-workspace-locking.ts

import { CloudRunnerSystem } from '../cloud-runner/services/cloud-runner-system';
import * as fs from 'fs';
import CloudRunnerLogger from '../cloud-runner/services/cloud-runner-logger';
import CloudRunnerOptions from '../cloud-runner/cloud-runner-options';
export class SharedWorkspaceLocking {
  public static async GetLockedWorkspace(workspaceIfCreated: string, runId: string) {
    if (!CloudRunnerOptions.retainWorkspaces) {
      return;
    }

    const workspaces = await SharedWorkspaceLocking.GetFreeWorkspaces();
    for (const element of workspaces) {
      if (await SharedWorkspaceLocking.LockWorkspace(element, runId)) {
        return element;
      }
    }

    return await SharedWorkspaceLocking.CreateLockableWorkspace(workspaceIfCreated);
  }

  public static async GetFreeWorkspaces(): Promise<string[]> {
    const result: string[] = [];
    const workspaces = await SharedWorkspaceLocking.GetAllWorkspaces();
    for (const element of workspaces) {
      if (!(await SharedWorkspaceLocking.IsWorkspaceLocked(element))) {
        result.push(element);
      }
    }

    return result;
  }
  public static async GetAllWorkspaces(): Promise<string[]> {
    return (await SharedWorkspaceLocking.ReadLines(`aws s3 ls s3://game-ci-test-storage/locks/`)).map((x) =>
      x.replace(`/`, ``),
    );
  }
  public static async GetAllLocks(workspace: string): Promise<string[]> {
    return (await SharedWorkspaceLocking.ReadLines(`aws s3 ls s3://game-ci-test-storage/locks/${workspace}/`)).map(
      (x) => x.replace(`/`, ``),
    );
  }

  private static async ReadLines(command: string): Promise<string[]> {
    const result = await CloudRunnerSystem.Run(command, false, true);

    return result
      .split(`\n`)
      .map((x) => x.replace(`\r`, ``))
      .filter((x) => x !== ``)
      .map((x) => {
        const lineValues = x.split(` `);

        return lineValues[lineValues.length - 1];
      });
  }

  public static async LockWorkspace(workspace: string, runId: string): Promise<boolean> {
    const file = `${Date.now()}_${runId}_lock`;
    fs.writeFileSync(file, '');
    await CloudRunnerSystem.Run(
      `aws s3 cp ./${file} s3://game-ci-test-storage/locks/${workspace}/${file}`,
      false,
      true,
    );
    fs.rmSync(file);

    return SharedWorkspaceLocking.HasWorkspaceLock(workspace);
  }

  public static async ReleaseWorkspace(workspace: string, runId: string): Promise<boolean> {
    const file = (await SharedWorkspaceLocking.GetAllLocks(workspace)).filter((x) => x.includes(`_${runId}_lock`));
    CloudRunnerLogger.log(`${JSON.stringify(await SharedWorkspaceLocking.GetAllLocks(workspace))}`);
    CloudRunnerLogger.log(`Deleting file ${file}`);
    CloudRunnerLogger.log(`aws s3 rm s3://game-ci-test-storage/locks/${workspace}/${file}`);
    await CloudRunnerSystem.Run(`aws s3 rm s3://game-ci-test-storage/locks/${workspace}/${file}`, false, true);

    return !SharedWorkspaceLocking.HasWorkspaceLock(workspace);
  }
  public static async HasWorkspaceLock(workspace: string): Promise<boolean> {
    CloudRunnerLogger.log(
      (await CloudRunnerSystem.Run(`aws s3 ls s3://game-ci-test-storage/locks/${workspace}/`, false, true))
        .split('\n')
        .map((x) => {
          return x.split(' ');
        })
        .length.toString(),
    );

    return true;
  }

  public static async IsWorkspaceLocked(workspace: string): Promise<boolean> {
    const files = await SharedWorkspaceLocking.ReadLines(`aws s3 ls s3://game-ci-test-storage/locks/${workspace}/`);

    // 1 Because we expect 1 workspace file to exist in every workspace folder
    return files.length > 1;
  }

  public static async CreateLockableWorkspace(workspace: string) {
    const file = `${Date.now()}_workspace`;
    fs.writeFileSync(file, '');
    await CloudRunnerSystem.Run(
      `aws s3 cp ./${file} s3://game-ci-test-storage/locks/${workspace}/${file}`,
      false,
      true,
    );
    fs.rmSync(file);

    return workspace;
  }
  // eslint-disable-next-line no-unused-vars
  public static ReleaseLock(workspace: string) {}
}

export default SharedWorkspaceLocking;