openAgent

1import { test, expect } from "bun:test"
2import {
3  formatImportFileError,
4  parseShareUrl,
5  shouldAttachShareAuthHeaders,
6  transformShareData,
7  type ShareData,
8} from "../../src/cli/cmd/import"
9import { FSUtil } from "@opencode-ai/core/fs-util"
10import { PlatformError } from "effect"
11 
12test("formats import file errors", () => {
13  expect(
14    formatImportFileError(
15      "test.json",
16      new PlatformError.PlatformError(
17        new PlatformError.SystemError({
18          _tag: "NotFound",
19          module: "FileSystem",
20          method: "readFileString",
21        }),
22      ),
23    ),
24  ).toBe("File not found: test.json")
25  expect(
26    formatImportFileError(
27      "test.json",
28      new PlatformError.PlatformError(
29        new PlatformError.SystemError({
30          _tag: "PermissionDenied",
31          module: "FileSystem",
32          method: "readFileString",
33        }),
34      ),
35    ),
36  ).toBe("Failed to read file: Permission denied")
37  expect(
38    formatImportFileError(
39      "test.json",
40      new FSUtil.FileSystemError({ method: "readJson", cause: new SyntaxError("Unexpected token") }),
41    ),
42  ).toBe("Invalid JSON in test.json: Unexpected token")
43})
44 
45// parseShareUrl tests
46test("parses valid share URLs", () => {
47  expect(parseShareUrl("https://opncd.ai/share/Jsj3hNIW")).toBe("Jsj3hNIW")
48  expect(parseShareUrl("https://custom.example.com/share/abc123")).toBe("abc123")
49  expect(parseShareUrl("http://localhost:3000/share/test_id-123")).toBe("test_id-123")
50})
51 
52test("rejects invalid URLs", () => {
53  expect(parseShareUrl("https://opncd.ai/s/Jsj3hNIW")).toBeNull() // legacy format
54  expect(parseShareUrl("https://opncd.ai/share/")).toBeNull()
55  expect(parseShareUrl("https://opncd.ai/share/id/extra")).toBeNull()
56  expect(parseShareUrl("not-a-url")).toBeNull()
57})
58 
59test("only attaches share auth headers for same-origin URLs", () => {
60  expect(shouldAttachShareAuthHeaders("https://control.example.com/share/abc", "https://control.example.com")).toBe(
61    true,
62  )
63  expect(shouldAttachShareAuthHeaders("https://other.example.com/share/abc", "https://control.example.com")).toBe(false)
64  expect(shouldAttachShareAuthHeaders("https://control.example.com:443/share/abc", "https://control.example.com")).toBe(
65    true,
66  )
67  expect(shouldAttachShareAuthHeaders("not-a-url", "https://control.example.com")).toBe(false)
68})
69 
70// transformShareData tests
71test("transforms share data to storage format", () => {
72  const data: ShareData[] = [
73    { type: "session", data: { id: "sess-1", title: "Test" } as any },
74    { type: "message", data: { id: "msg-1", sessionID: "sess-1" } as any },
75    { type: "part", data: { id: "part-1", messageID: "msg-1" } as any },
76    { type: "part", data: { id: "part-2", messageID: "msg-1" } as any },
77  ]
78 
79  const result = transformShareData(data)!
80 
81  expect(result.info.id).toBe("sess-1")
82  expect(result.messages).toHaveLength(1)
83  expect(result.messages[0].parts).toHaveLength(2)
84})
85 
86test("returns null for invalid share data", () => {
87  expect(transformShareData([])).toBeNull()
88  expect(transformShareData([{ type: "message", data: {} as any }])).toBeNull()
89  expect(transformShareData([{ type: "session", data: { id: "s" } as any }])).toBeNull() // no messages
90})
91