123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150
import { join } from "https://deno.land/std@0.175.0/path/mod.ts";import { getVendorDir } from "../cli/pb/config.ts";import expandEntryPaths from "../cli/pb/cmds/gen/expandEntryPaths.ts";import { loadPollapoYml } from "../cli/pollapo/pollapoYml.ts";import { parseBufWorkYaml } from "../compat/buf/workspace.ts";import { Loader } from "../core/loader/index.ts";import combineLoader from "../core/loader/combineLoader.ts";import memoizeLoader, { MemoizedLoader } from "../core/loader/memoizeLoader.ts";import {  createLoader as createDenoFsLoader,  fromFileUrl,  resolve,} from "../core/loader/deno-fs.ts";import { OpenProtoManager } from "./open-proto-manager.ts";
export interface ProjectManager {  addProjectPath(projectPath: string): Promise<void>;  getProjectPath(filePath: string): string | undefined;  getProjectEntryPaths(projectPath: string): Promise<string[]>;  getProjectDependencyPaths(projectPath: string): Promise<string[]>;  getProjectProtoFiles(filePath: string): Promise<string[]>;  getProjectLoader(projectPath?: string): Promise<MemoizedLoader>;  createProjectLoader(projectPath?: string): Promise<MemoizedLoader>;}
export function createProjectManager(  openProtoManager: OpenProtoManager,): ProjectManager {  // sorted string array in descending order  // ex) ['file:///foo/baz', 'file:///foo/bar/baz', 'file:///foo/bar', 'file:///foo']  const projectPaths: string[] = [];  const loaders: { [projectPath: string]: MemoizedLoader } = {};  return {    addProjectPath,    getProjectPath,    getProjectEntryPaths,    getProjectDependencyPaths,    getProjectProtoFiles,    async getProjectLoader(projectPath) {      if (!projectPath) return await createProjectLoader(projectPath);      return loaders[projectPath] ??= await createProjectLoader(projectPath);    },    createProjectLoader,  };  async function addProjectPath(projectPath: string): Promise<void> {    projectPaths.push(projectPath);    projectPaths.sort().reverse();  }  function getProjectPath(filePath: string): string | undefined {    return projectPaths.find((p) => filePath.startsWith(p));  }  async function getProjectEntryPaths(projectPath: string): Promise<string[]> {    return [projectPath, ...await getBufWorkDirs(projectPath)];  }  async function getProjectDependencyPaths(    projectPath: string,  ): Promise<string[]> {    const result: string[] = [];    const pollapoPath = join(projectPath, ".pollapo");    try {      if ((await Deno.lstat(pollapoPath)).isDirectory) result.push(pollapoPath);    } catch {}    return result;  }  async function getProjectProtoFiles(filePath: string): Promise<string[]> {    const projectPath = getProjectPath(filePath);    if (projectPath) {      const entryPaths = await getProjectEntryPaths(projectPath);      return await expandEntryPaths(entryPaths);    } else {      return [filePath];    }  }  async function createProjectLoader(    projectPath?: string,  ): Promise<MemoizedLoader> {    const roots = projectPath      ? [        ...await getProjectEntryPaths(projectPath),        ...await getProjectDependencyPaths(projectPath),        ...getVendorDirs(),      ]      : getVendorDirs();    const repo = projectPath && await getPollapoRepo(projectPath);    return memoizeLoader(combineLoader(      createOpenProtoLoader(openProtoManager),      repo && createPollapoRepoLoader(projectPath, repo),      createDenoFsLoader({ roots }),    ));  }}
function getVendorDirs() {  if (Deno.build.os === "windows") {    return [getVendorDir()];  } else {    return [getVendorDir(), "/usr/local/include"];  }}
async function getBufWorkDirs(  projectPath: string,): Promise<string[]> {  try {    const bufWorkYamlPath = resolve(projectPath, "buf.work.yaml");    const bufWorkYaml = parseBufWorkYaml(      await Deno.readTextFile(fromFileUrl(bufWorkYamlPath)),    );    return bufWorkYaml.directories.map(      (directory) => resolve(projectPath, String(directory)),    );  } catch {}  return [];}
async function getPollapoRepo(  projectPath: string,): Promise<string | undefined> {  try {    const pollapoYmlPath = resolve(projectPath, "pollapo.yml");    const pollapoYml = await loadPollapoYml(fromFileUrl(pollapoYmlPath));    if (pollapoYml?.repo) return String(pollapoYml.repo);  } catch {}}
function createOpenProtoLoader(openProtoManager: OpenProtoManager): Loader {  return {    async load(absolutePath) {      const data = openProtoManager.getLastParsed(absolutePath);      if (data == null) return null;      return { absolutePath, data };    },  };}
function createPollapoRepoLoader(projectPath: string, repo: string): Loader {  return {    async load(path) {      if (!path.startsWith(repo + "/")) return null;      const absolutePath = resolve(projectPath, path.slice(repo.length + 1));      const filePath = fromFileUrl(absolutePath);      try {        return { absolutePath, data: await Deno.readTextFile(filePath) };      } catch {        return null;      }    },  };}