genai-fs 1.0.2

package/src/components/FileBrowser.tsx

@@ -0,0 +1,173 @@
+import { useState, useEffect } from "react";
+import { Box, Text, useInput } from "ink";
+import { readdir, stat } from "fs/promises";
+import { join, dirname } from "path";
+import { homedir } from "os";
+
+interface FileEntry {
+  name: string;
+  path: string;
+  isDirectory: boolean;
+  size?: number;
+}
+
+interface FileBrowserProps {
+  onSelect: (filePath: string) => void;
+  onCancel: () => void;
+  initialPath?: string;
+}
+
+// ファイルサイズをフォーマット
+function formatSize(bytes: number | undefined): string {
+  if (bytes === undefined) return "";
+  if (bytes < 1024) return `${bytes} B`;
+  if (bytes < 1024 * 1024) return `${(bytes / 1024).toFixed(1)} KB`;
+  return `${(bytes / (1024 * 1024)).toFixed(1)} MB`;
+}
+
+export function FileBrowser({ onSelect, onCancel, initialPath }: FileBrowserProps) {
+  const [currentPath, setCurrentPath] = useState(initialPath || homedir());
+  const [entries, setEntries] = useState<FileEntry[]>([]);
+  const [selectedIndex, setSelectedIndex] = useState(0);
+  const [loading, setLoading] = useState(true);
+  const [error, setError] = useState<string | null>(null);
+
+  // ディレクトリの内容を読み込む
+  useEffect(() => {
+    const loadDirectory = async () => {
+      try {
+        setLoading(true);
+        setError(null);
+
+        const items = await readdir(currentPath);
+        const fileEntries: FileEntry[] = [];
+
+        // 親ディレクトリへのエントリ
+        const parent = dirname(currentPath);
+        if (parent !== currentPath) {
+          fileEntries.push({
+            name: "..",
+            path: parent,
+            isDirectory: true,
+          });
+        }
+
+        // 各アイテムの情報を取得
+        for (const name of items) {
+          // 隠しファイルをスキップ
+          if (name.startsWith(".")) continue;
+
+          try {
+            const fullPath = join(currentPath, name);
+            const stats = await stat(fullPath);
+            fileEntries.push({
+              name,
+              path: fullPath,
+              isDirectory: stats.isDirectory(),
+              size: stats.isFile() ? stats.size : undefined,
+            });
+          } catch {
+            // アクセスできないファイルはスキップ
+          }
+        }
+
+        // ディレクトリを先に、その後ファイルをソート
+        fileEntries.sort((a, b) => {
+          if (a.name === "..") return -1;
+          if (b.name === "..") return 1;
+          if (a.isDirectory && !b.isDirectory) return -1;
+          if (!a.isDirectory && b.isDirectory) return 1;
+          return a.name.localeCompare(b.name);
+        });
+
+        setEntries(fileEntries);
+        setSelectedIndex(0);
+      } catch (e) {
+        setError(e instanceof Error ? e.message : "Failed to load directory");
+      } finally {
+        setLoading(false);
+      }
+    };
+
+    loadDirectory();
+  }, [currentPath]);
+
+  useInput((input, key) => {
+    if (loading) return;
+
+    if (key.upArrow) {
+      setSelectedIndex((prev) => (prev > 0 ? prev - 1 : entries.length - 1));
+    }
+    if (key.downArrow) {
+      setSelectedIndex((prev) => (prev < entries.length - 1 ? prev + 1 : 0));
+    }
+    if (key.return) {
+      const selected = entries[selectedIndex];
+      if (selected) {
+        if (selected.isDirectory) {
+          setCurrentPath(selected.path);
+        } else {
+          onSelect(selected.path);
+        }
+      }
+    }
+    if (key.escape || input === "q") {
+      onCancel();
+    }
+  });
+
+  // 表示するエントリ数を制限（スクロール対応）
+  const maxVisible = 15;
+  const startIndex = Math.max(0, selectedIndex - Math.floor(maxVisible / 2));
+  const visibleEntries = entries.slice(startIndex, startIndex + maxVisible);
+
+  return (
+    <Box flexDirection="column">
+      <Box marginBottom={1}>
+        <Text color="yellow">Select a file:</Text>
+      </Box>
+
+      <Box marginBottom={1}>
+        <Text dimColor>Current path: </Text>
+        <Text>{currentPath}</Text>
+      </Box>
+
+      {loading && <Text color="gray">Loading...</Text>}
+      {error && <Text color="red">Error: {error}</Text>}
+
+      {!loading && !error && entries.length === 0 && (
+        <Text color="gray">No files found</Text>
+      )}
+
+      {!loading && !error && entries.length > 0 && (
+        <Box flexDirection="column">
+          {visibleEntries.map((entry, idx) => {
+            const actualIndex = startIndex + idx;
+            const isSelected = actualIndex === selectedIndex;
+            return (
+              <Box key={entry.path} gap={1}>
+                <Text color={isSelected ? "green" : "white"}>
+                  {isSelected ? "❯ " : "  "}
+                  {entry.isDirectory ? "📁 " : "📄 "}
+                  {entry.name}
+                </Text>
+                {!entry.isDirectory && entry.size !== undefined && (
+                  <Text dimColor>{formatSize(entry.size)}</Text>
+                )}
+              </Box>
+            );
+          })}
+          {entries.length > maxVisible && (
+            <Text dimColor>
+              ({selectedIndex + 1}/{entries.length})
+            </Text>
+          )}
+        </Box>
+      )}
+
+      <Box marginTop={1}>
+        <Text dimColor>Up/Down: Select  Enter: Confirm/Open  Esc: Cancel</Text>
+      </Box>
+    </Box>
+  );
+}

package/src/components/Spinner.tsx

@@ -0,0 +1,26 @@
+import { useState, useEffect } from "react";
+import { Text } from "ink";
+
+interface SpinnerProps {
+  message?: string;
+}
+
+const frames = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];
+
+export function Spinner({ message = "Processing..." }: SpinnerProps) {
+  const [frameIndex, setFrameIndex] = useState(0);
+
+  useEffect(() => {
+    const timer = setInterval(() => {
+      setFrameIndex((prev) => (prev + 1) % frames.length);
+    }, 80);
+
+    return () => clearInterval(timer);
+  }, []);
+
+  return (
+    <Text color="cyan">
+      {frames[frameIndex]} {message}
+    </Text>
+  );
+}

package/src/components/StatusScreen.tsx

@@ -0,0 +1,22 @@
+import { Box, Text } from "ink";
+import { Spinner } from "./Spinner.tsx";
+
+interface StatusScreenProps {
+  isProcessing: boolean;
+  message: string;
+}
+
+// 処理中/完了/エラー表示
+export function StatusScreen({ isProcessing, message }: StatusScreenProps) {
+  const isError = message.startsWith("Error");
+
+  return (
+    <Box flexDirection="column">
+      {isProcessing ? (
+        <Spinner message={message} />
+      ) : (
+        <Text color={isError ? "red" : "green"}>{message}</Text>
+      )}
+    </Box>
+  );
+}

package/src/components/StoreDetail.tsx

@@ -0,0 +1,80 @@
+import { Box, Text } from "ink";
+import type { FileSearchStore } from "../lib/api.ts";
+import { formatBytes, formatDate } from "../utils/formatters.ts";
+
+interface StoreDetailProps {
+  store: FileSearchStore;
+}
+
+// Store詳細表示
+export function StoreDetail({ store }: StoreDetailProps) {
+  const active = store.activeDocumentsCount ?? "0";
+  const pending = store.pendingDocumentsCount ?? "0";
+  const failed = store.failedDocumentsCount ?? "0";
+
+  return (
+    <Box flexDirection="column" gap={0}>
+      <Box>
+        <Box width={14}>
+          <Text bold dimColor>
+            DisplayName
+          </Text>
+        </Box>
+        <Text>{store.displayName}</Text>
+      </Box>
+      <Box>
+        <Box width={14}>
+          <Text bold dimColor>
+            ID
+          </Text>
+        </Box>
+        <Text>{store.name.split("/").pop()}</Text>
+      </Box>
+      <Box>
+        <Box width={14}>
+          <Text bold dimColor>
+            FullID
+          </Text>
+        </Box>
+        <Text>{store.name}</Text>
+      </Box>
+      <Box>
+        <Box width={14}>
+          <Text bold dimColor>
+            Size
+          </Text>
+        </Box>
+        <Text>{formatBytes(store.sizeBytes)}</Text>
+      </Box>
+      <Box>
+        <Box width={14}>
+          <Text bold dimColor>
+            Documents
+          </Text>
+        </Box>
+        <Text color="green">{active} active</Text>
+        {pending !== "0" && <Text color="yellow"> / {pending} pending</Text>}
+        {failed !== "0" && <Text color="red"> / {failed} failed</Text>}
+      </Box>
+      <Box>
+        <Box width={14}>
+          <Text bold dimColor>
+            Created
+          </Text>
+        </Box>
+        <Text>{formatDate(store.createTime)}</Text>
+      </Box>
+      <Box>
+        <Box width={14}>
+          <Text bold dimColor>
+            Updated
+          </Text>
+        </Box>
+        <Text>{formatDate(store.updateTime)}</Text>
+      </Box>
+      <Box marginTop={1}>
+        <Text dimColor>Esc/b: Back  q: Quit</Text>
+      </Box>
+    </Box>
+  );
+}

package/src/components/StoreForm.tsx

@@ -0,0 +1,38 @@
+import { Box, Text } from "ink";
+
+interface StoreFormProps {
+  mode: "create" | "delete";
+  input: string;
+  storeName?: string; // 削除時の確認用
+}
+
+// Store作成/削除確認フォーム
+export function StoreForm({ mode, input, storeName }: StoreFormProps) {
+  const isValid =
+    mode === "create" ? input.trim().length > 0 : input === storeName;
+
+  return (
+    <Box flexDirection="column">
+      <Box marginBottom={1}>
+        <Text color={mode === "create" ? "green" : "red"} bold>
+          {mode === "create" ? "Create new store" : "Delete this store?"}
+        </Text>
+      </Box>
+      <Text>
+        {mode === "create"
+          ? "Enter the store name:"
+          : `To delete "${storeName}", type the store name to confirm:`}
+      </Text>
+      <Box marginTop={1}>
+        <Text dimColor>{">"} </Text>
+        <Text color={isValid ? "green" : "white"}>{input}</Text>
+        <Text color="gray">|</Text>
+      </Box>
+      <Box marginTop={1}>
+        <Text dimColor>
+          {mode === "create" ? "Enter: Create" : "Enter: Confirm"}  Esc: Cancel
+        </Text>
+      </Box>
+    </Box>
+  );
+}

package/src/components/StoreList.tsx

@@ -0,0 +1,80 @@
+import { Box, Text } from "ink";
+import type { FileSearchStore } from "../lib/api.ts";
+import { formatDate } from "../utils/formatters.ts";
+
+interface StoreListProps {
+  stores: FileSearchStore[];
+  selectedIndex: number;
+  loading: boolean;
+  error: string | null;
+}
+
+// Store一覧テーブル
+export function StoreList({
+  stores,
+  selectedIndex,
+  loading,
+  error,
+}: StoreListProps) {
+  if (loading) {
+    return <Text color="gray">Loading...</Text>;
+  }
+
+  if (error) {
+    return <Text color="red">Error: {error}</Text>;
+  }
+
+  if (stores.length === 0) {
+    return (
+      <Box flexDirection="column">
+        <Text color="gray">No stores found</Text>
+        <Box marginTop={1}>
+          <Text dimColor>n: New  q: Quit</Text>
+        </Box>
+      </Box>
+    );
+  }
+
+  return (
+    <Box flexDirection="column">
+      <Box gap={1} marginBottom={1}>
+        <Box width={24}>
+          <Text bold dimColor>
+            Name
+          </Text>
+        </Box>
+        <Box width={22}>
+          <Text bold dimColor>
+            Created
+          </Text>
+        </Box>
+        <Box width={22}>
+          <Text bold dimColor>
+            Updated
+          </Text>
+        </Box>
+      </Box>
+      {stores.map((store, index) => (
+        <Box key={store.name} gap={1}>
+          <Box width={24}>
+            <Text color={index === selectedIndex ? "green" : "white"}>
+              {index === selectedIndex ? "❯ " : "  "}
+              {store.displayName.slice(0, 20)}
+            </Text>
+          </Box>
+          <Box width={22}>
+            <Text dimColor>{formatDate(store.createTime)}</Text>
+          </Box>
+          <Box width={22}>
+            <Text dimColor>{formatDate(store.updateTime)}</Text>
+          </Box>
+        </Box>
+      ))}
+      <Box marginTop={1}>
+        <Text dimColor>
+          Up/Down: Select  Enter: Open  i: Info  n: New  d: Delete  q: Quit
+        </Text>
+      </Box>
+    </Box>
+  );
+}

package/src/index.tsx

@@ -0,0 +1,5 @@
+#!/usr/bin/env bun
+import { render } from "ink";
+import { App } from "./App.tsx";
+
+render(<App />);

package/src/lib/api.ts

@@ -0,0 +1,191 @@
+// Gemini FileStore API クライアント
+import { GoogleGenAI } from "@google/genai";
+
+const BASE_URL = "https://generativelanguage.googleapis.com/v1beta";
+
+function getApiKey(): string {
+  const apiKey = process.env.GEMINI_API_KEY;
+  if (!apiKey) {
+    throw new Error("GEMINI_API_KEY environment variable is not set");
+  }
+  return apiKey;
+}
+
+// Google GenAI クライアント
+let _client: GoogleGenAI | null = null;
+function getClient(): GoogleGenAI {
+  if (!_client) {
+    _client = new GoogleGenAI({ apiKey: getApiKey() });
+  }
+  return _client;
+}
+
+// ストアの型定義
+export interface FileSearchStore {
+  name: string; // 例: "fileSearchStores/xxx"
+  displayName: string;
+  createTime?: string;
+  updateTime?: string;
+  activeDocumentsCount?: string; // アクティブなドキュメント数
+  pendingDocumentsCount?: string; // 処理中のドキュメント数
+  failedDocumentsCount?: string; // 失敗したドキュメント数
+  sizeBytes?: string; // ストア全体のサイズ
+}
+
+// ドキュメントの状態
+export type DocumentState = "STATE_UNSPECIFIED" | "PENDING" | "ACTIVE" | "FAILED";
+
+// カスタムメタデータ
+export interface CustomMetadata {
+  key: string;
+  stringValue?: string;
+  numericValue?: number;
+}
+
+// ドキュメントの型定義
+export interface Document {
+  name: string; // 例: "fileSearchStores/xxx/documents/yyy"
+  displayName?: string;
+  mimeType?: string;
+  sizeBytes?: string;
+  createTime?: string;
+  updateTime?: string;
+  state?: DocumentState;
+  customMetadata?: CustomMetadata[];
+}
+
+interface ListStoresResponse {
+  fileSearchStores?: FileSearchStore[];
+  nextPageToken?: string;
+}
+
+interface ListDocumentsResponse {
+  documents?: Document[];
+  nextPageToken?: string;
+}
+
+// ストア一覧を取得
+export async function listStores(pageSize = 20): Promise<FileSearchStore[]> {
+  const apiKey = getApiKey();
+  const url = `${BASE_URL}/fileSearchStores?key=${apiKey}&pageSize=${pageSize}`;
+
+  const response = await fetch(url);
+  if (!response.ok) {
+    const error = await response.text();
+    throw new Error(`Failed to fetch stores: ${error}`);
+  }
+
+  const data = (await response.json()) as ListStoresResponse;
+  return data.fileSearchStores ?? [];
+}
+
+// ストアを作成
+export async function createStore(displayName: string): Promise<FileSearchStore> {
+  const apiKey = getApiKey();
+  const url = `${BASE_URL}/fileSearchStores?key=${apiKey}`;
+
+  const response = await fetch(url, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({ displayName }),
+  });
+
+  if (!response.ok) {
+    const error = await response.text();
+    throw new Error(`Failed to create store: ${error}`);
+  }
+
+  // 長時間実行オペレーションが返される
+  const operation = await response.json();
+  // 完了を待つ場合はポーリングが必要だが、一旦オペレーション結果を返す
+  return operation as FileSearchStore;
+}
+
+// ストアを削除
+export async function deleteStore(storeName: string): Promise<void> {
+  const apiKey = getApiKey();
+  const url = `${BASE_URL}/${storeName}?key=${apiKey}`;
+
+  const response = await fetch(url, { method: "DELETE" });
+  if (!response.ok) {
+    const error = await response.text();
+    throw new Error(`Failed to delete store: ${error}`);
+  }
+}
+
+// ドキュメント一覧を取得
+export async function listDocuments(
+  storeName: string,
+  pageSize = 20
+): Promise<Document[]> {
+  const apiKey = getApiKey();
+  const url = `${BASE_URL}/${storeName}/documents?key=${apiKey}&pageSize=${pageSize}`;
+
+  const response = await fetch(url);
+  if (!response.ok) {
+    const error = await response.text();
+    throw new Error(`Failed to fetch documents: ${error}`);
+  }
+
+  const data = (await response.json()) as ListDocumentsResponse;
+  return data.documents ?? [];
+}
+
+// ドキュメント詳細を取得
+export async function getDocument(documentName: string): Promise<Document> {
+  const apiKey = getApiKey();
+  const url = `${BASE_URL}/${documentName}?key=${apiKey}`;
+
+  const response = await fetch(url);
+  if (!response.ok) {
+    const error = await response.text();
+    throw new Error(`Failed to fetch document: ${error}`);
+  }
+
+  return (await response.json()) as Document;
+}
+
+// アップロードオペレーションのレスポンス
+export interface UploadOperation {
+  name: string;
+  done?: boolean;
+  error?: { code: number; message: string };
+  response?: Document;
+}
+
+// ドキュメントを削除
+export async function deleteDocument(documentName: string): Promise<void> {
+  const apiKey = getApiKey();
+  // force=trueで強制削除
+  const url = `${BASE_URL}/${documentName}?key=${apiKey}&force=true`;
+
+  const response = await fetch(url, { method: "DELETE" });
+  if (!response.ok) {
+    const error = await response.text();
+    throw new Error(`Failed to delete document: ${error}`);
+  }
+}
+
+// ファイルをアップロード（Google GenAI SDKを使用）
+export async function uploadFile(
+  storeName: string,
+  filePath: string
+): Promise<void> {
+  const client = getClient();
+
+  // アップロード開始
+  let operation = await client.fileSearchStores.uploadToFileSearchStore({
+    fileSearchStoreName: storeName,
+    file: filePath,
+  });
+
+  // 完了まで待機
+  while (!operation.done) {
+    await new Promise((resolve) => setTimeout(resolve, 2000));
+    operation = await client.operations.get({ operation });
+  }
+
+  if (operation.error) {
+    throw new Error(`Upload failed: ${operation.error.message}`);
+  }
+}

package/src/utils/formatters.ts

@@ -0,0 +1,44 @@
+// フォーマット関数
+
+// ファイルサイズをフォーマット
+export function formatBytes(bytes: string | undefined): string {
+  if (!bytes) return "-";
+  const size = parseInt(bytes, 10);
+  if (size < 1024) return `${size} B`;
+  if (size < 1024 * 1024) return `${(size / 1024).toFixed(1)} KB`;
+  return `${(size / (1024 * 1024)).toFixed(1)} MB`;
+}
+
+// 日時をフォーマット
+export function formatDate(dateStr: string | undefined): string {
+  if (!dateStr) return "-";
+  const date = new Date(dateStr);
+  return date.toLocaleString("ja-JP");
+}
+
+// APIキーをマスク
+export function maskApiKey(): string {
+  const apiKey = process.env.GEMINI_API_KEY;
+  if (!apiKey) return "Not set";
+  if (apiKey.length <= 8) return "****";
+  return `${apiKey.slice(0, 4)}...${apiKey.slice(-4)}`;
+}
+
+// ステータス表示をフォーマット
+export function formatState(
+  state: string | undefined
+): { text: string; color: string } {
+  switch (state) {
+    case "ACTIVE":
+    case "STATE_ACTIVE":
+      return { text: "Active", color: "green" };
+    case "PENDING":
+    case "STATE_PENDING":
+      return { text: "Pending", color: "yellow" };
+    case "FAILED":
+    case "STATE_FAILED":
+      return { text: "Failed", color: "red" };
+    default:
+      return { text: state ?? "-", color: "gray" };
+  }
+}

package/tsconfig.json

@@ -0,0 +1,29 @@
+{
+  "compilerOptions": {
+    // Environment setup & latest features
+    "lib": ["ESNext"],
+    "target": "ESNext",
+    "module": "Preserve",
+    "moduleDetection": "force",
+    "jsx": "react-jsx",
+    "allowJs": true,
+
+    // Bundler mode
+    "moduleResolution": "bundler",
+    "allowImportingTsExtensions": true,
+    "verbatimModuleSyntax": true,
+    "noEmit": true,
+
+    // Best practices
+    "strict": true,
+    "skipLibCheck": true,
+    "noFallthroughCasesInSwitch": true,
+    "noUncheckedIndexedAccess": true,
+    "noImplicitOverride": true,
+
+    // Some stricter flags (disabled by default)
+    "noUnusedLocals": false,
+    "noUnusedParameters": false,
+    "noPropertyAccessFromIndexSignature": false
+  }
+}