Function useFiles

src/hooks/useFiles.ts:144–223

(options: CommandOptions)

}

export const useFiles = (options: CommandOptions) => {
  const [selectedFiles, setSelectedFiles] = useState<{ paths: string[]; csv: string }>();
  const [fileContents, setFileContents] = useState<{ [key: string]: string; contents: string }>();
  const [isLoading, setIsLoading] = useState<boolean>(true);
  const [error, setError] = useState<number>();
  const { models } = useModels();
  const { advancedSettings, revalidateAdvancedSettings } = useAdvancedSettings();

  const loadSelection = async () => {
    if (!options.minNumFiles)
      return {
        paths: [],
        csv: "",
      };

    const validExtensions = (options.acceptedFileExtensions || []).map((ext) => ext.toLowerCase());

    setIsLoading(true);
    const rawSelection = await ScriptRunner.SelectedFiles();

    // Remove invalid files and directories from the selection
    const selection = rawSelection.paths.reduce(
      (acc, filepath) => {
        if (
          filepath.trim().length > 0 &&
          (validExtensions.length == 0 || validExtensions.includes(path.extname(filepath).slice(1).toLowerCase()))
        ) {
          return { paths: [...acc.paths, filepath], csv: acc.csv + "," + filepath };
        }
        return acc;
      },
      { paths: [] as string[], csv: "" },
    );

    maxCharacters = maxCharacters / selection.paths.length;
    setSelectedFiles(selection);
    return selection;
  };

  const loadFileContents = async (selection: { paths: string[]; csv: string }) => {
    // Raise an error if too few files are selected
    if (selection.paths.length < (options.minNumFiles || 0)) {
      setError(ERRORTYPE.MIN_SELECTION_NOT_MET);
      setIsLoading(false);
      return;
    }

    await revalidateAdvancedSettings();

    let model = options.model ? models.find((m) => m.name == options.model) : undefined;
    if (!model) model = models.find((m) => m.isDefault) || models[0];
    const maxCharacters = model ? parseInt(model.lengthLimit) : 3000;

    const fileData: { [key: string]: string; contents: string } = {
      contents: "",
    };
    for (const [index, filepath] of selection.paths.entries()) {
      // Init. file contents with file name as header