@hippo-memo/cli 1.0.0

package/src/commands/init.ts

@@ -0,0 +1,158 @@
+import {
+  access,
+  constants,
+  mkdir,
+  readFile,
+  writeFile
+} from "node:fs/promises";
+import { EOL } from "node:os";
+import { join } from "node:path";
+import { createConfig, createStore } from "@hippo-memo/core";
+import { MEMORY_DIR_NAME } from "@hippo-memo/shared";
+
+const __dirname = import.meta.dirname;
+const TEMPLATE_DIR = join(__dirname, "../../template/system");
+const AGENTS_TEMPLATE_PATH = join(__dirname, "../../template/mcp/AGENTS.md");
+
+const TEMPLATE_URIS = {
+  agent: "system://agent",
+  soul: "system://soul",
+  user: "system://user"
+};
+
+async function readTemplateContent(filename: string): Promise<string> {
+  return readFile(join(TEMPLATE_DIR, filename), "utf-8");
+}
+
+/**
+ * 从内容中提取第一行非空内容作为注入标识符
+ */
+function getInjectionMarker(content: string): string {
+  const lines = content.split("\n");
+  const firstNonEmptyLine = lines.find((line) => line.trim() !== "");
+  return firstNonEmptyLine?.trim() || "";
+}
+
+/**
+ * 检查文件是否已注入（通过第一行判断）
+ */
+function isAlreadyInjected(content: string, marker: string): boolean {
+  const lines = content.split("\n");
+  // 跳过开头的空行
+  const firstNonEmptyLine = lines.find((line) => line.trim() !== "");
+  return firstNonEmptyLine?.trim() === marker;
+}
+
+/**
+ * 安全地将内容注入到文件顶部
+ * @param filePath 目标文件路径
+ * @param contentToInject 要注入的内容
+ */
+async function injectToTopOfFile(
+  filePath: string,
+  contentToInject: string
+): Promise<void> {
+  try {
+    let originalContent = "";
+
+    try {
+      originalContent = await readFile(filePath, "utf-8");
+    } catch {
+      originalContent = "";
+    }
+
+    // 从注入内容中提取第一行作为标识符
+    const injectionMarker = getInjectionMarker(contentToInject);
+
+    // 幂等性检查：通过第一行判断是否已注入
+    if (
+      originalContent &&
+      isAlreadyInjected(originalContent, injectionMarker)
+    ) {
+      console.log(`› Skipped: content already exists`);
+      return;
+    }
+
+    // 拼接内容，注意处理换行符
+    const separator = originalContent ? `${EOL}${EOL}` : "";
+    const updatedContent = `${contentToInject}${separator}${originalContent}`;
+
+    await writeFile(filePath, updatedContent, "utf-8");
+    console.log(`✔ Injected: ${filePath}`);
+  } catch (error) {
+    console.error(`✖ Failed: ${filePath}`, error);
+    throw error;
+  }
+}
+
+/**
+ * 检测并处理 AGENTS.md 和 CLAUDE.md 文件
+ */
+async function handleAgentsAndClaudeFiles(directory: string): Promise<void> {
+  const agentsPath = join(directory, "AGENTS.md");
+  const claudePath = join(directory, "CLAUDE.md");
+
+  let hasAgents = false;
+  let hasClaude = false;
+
+  try {
+    await access(agentsPath, constants.F_OK);
+    hasAgents = true;
+  } catch {
+    // AGENTS.md 不存在
+  }
+
+  try {
+    await access(claudePath, constants.F_OK);
+    hasClaude = true;
+  } catch {
+    // CLAUDE.md 不存在
+  }
+
+  // 读取 AGENTS.md 模板内容
+  const agentsContent = await readFile(AGENTS_TEMPLATE_PATH, "utf-8");
+
+  if (hasClaude) {
+    // CLAUDE.md 存在：优先注入到 CLAUDE.md 顶部
+    console.log("› Found: CLAUDE.md");
+    await injectToTopOfFile(claudePath, agentsContent);
+  } else if (hasAgents) {
+    // 只有 AGENTS.md 存在：注入到 AGENTS.md 顶部
+    console.log("› Found: AGENTS.md");
+    await injectToTopOfFile(agentsPath, agentsContent);
+  } else {
+    // 两者都不存在：创建 AGENTS.md
+    console.log("✔ Created: AGENTS.md");
+    await writeFile(agentsPath, agentsContent, "utf-8");
+  }
+}
+
+export async function init(directory: string): Promise<void> {
+  const memoryRoot = join(directory, MEMORY_DIR_NAME);
+
+  try {
+    await access(memoryRoot, constants.F_OK);
+    console.log(`Memory directory already exists at ${memoryRoot}`);
+  } catch {
+    // Directory doesn't exist, proceed with creation
+    await mkdir(memoryRoot, { recursive: true });
+
+    const store = createStore({
+      type: "filesystem",
+      config: createConfig(directory)
+    });
+
+    await Promise.all(
+      Object.entries(TEMPLATE_URIS).map(async ([key, uri]) => {
+        const content = await readTemplateContent(`${key}.md`);
+        await store.createMemory({ uri, content, priority: 10 });
+      })
+    );
+
+    console.log(`Initialized hippo memory at ${memoryRoot}`);
+  }
+
+  // 处理 AGENTS.md 和 CLAUDE.md 文件
+  console.log("◇ Checking AGENTS.md and CLAUDE.md...");
+  await handleAgentsAndClaudeFiles(directory);
+}

package/src/commands/web.ts

@@ -0,0 +1,13 @@
+import { createConfig, createStore } from "@hippo-memo/core";
+import { startServer } from "../api/server";
+
+export async function web(
+  directory: string,
+  port: number,
+  storeType: "filesystem" | "sqlite"
+): Promise<void> {
+  const config = createConfig(directory);
+  const store = createStore({ type: storeType, config });
+
+  await startServer(store, port);
+}

package/src/index.ts

@@ -0,0 +1,33 @@
+#!/usr/bin/env node
+import { Command } from "commander";
+import { init, web } from "./commands";
+
+const program = new Command();
+
+program.name("hippo").description("Hippo Memory Store CLI").version("1.0.0");
+
+program
+  .command("init")
+  .description("Initialize a new hippo memory directory")
+  .argument("[directory]", "Directory to initialize", ".")
+  .action(async (directory) => {
+    await init(directory);
+  });
+
+program
+  .command("web")
+  .description("Start the web server")
+  .argument("[directory]", "Project directory", ".")
+  .option("-p, --port <port>", "Port to listen on", "3000")
+  .option(
+    "-s, --store-type <type>",
+    "Store type (filesystem or sqlite)",
+    "filesystem"
+  )
+  .action(async (directory, options) => {
+    const port = Number.parseInt(options.port, 10);
+    const storeType = options.storeType;
+    await web(directory, port, storeType);
+  });
+
+program.parse();

package/tsconfig.json

@@ -0,0 +1,17 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "ES2022",
+    "moduleResolution": "bundler",
+    "allowSyntheticDefaultImports": true,
+    "esModuleInterop": true,
+    "resolveJsonModule": true,
+    "declaration": false,
+    "outDir": "./dist",
+    "rootDir": "./src",
+    "strict": true,
+    "skipLibCheck": true
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "dist"]
+}