|
| 1 | +import { uint8ToHex } from '@siastorage/core' |
| 2 | +import type { DatabaseAdapter, SdkAdapter, SdkAuthAdapters } from '@siastorage/core/adapters' |
| 3 | +// oxlint-disable-next-line no-restricted-imports -- CLI daemon needs internal API to set SDK |
| 4 | +import type { AppService, AppServiceInternal } from '@siastorage/core/app' |
| 5 | +import { createAppService } from '@siastorage/core/app' |
| 6 | +import { APP_META } from '@siastorage/core/config' |
| 7 | +import { runMigrations } from '@siastorage/core/db' |
| 8 | +import { coreMigrations, sortMigrations } from '@siastorage/core/db/migrations' |
| 9 | +import type { FsIOAdapter } from '@siastorage/core/services/fsFileUri' |
| 10 | +import type { UploadManager } from '@siastorage/core/services/uploader' |
| 11 | +import { |
| 12 | + createJsonFileStorage, |
| 13 | + createNodeCryptoAdapter, |
| 14 | + createNodeDetectMimeType, |
| 15 | + createNodeDownloadAdapter, |
| 16 | + createNodeFsIO, |
| 17 | + createNodeSdkAdapter, |
| 18 | + createNodeSdkAuthAdapter, |
| 19 | + createNodeUploaderAdapters, |
| 20 | + createSharpThumbnailAdapter, |
| 21 | + ensureDataDir, |
| 22 | + getDataDir, |
| 23 | + getPaths, |
| 24 | + type NodeSdkAuthResult, |
| 25 | +} from '@siastorage/node-adapters' |
| 26 | +import { createBunDatabase } from '@siastorage/node-adapters/bunDatabase' |
| 27 | + |
| 28 | +export const isTestMode = process.env.SIA_TEST_MODE === '1' |
| 29 | + |
| 30 | +export type CliApp = { |
| 31 | + service: AppService |
| 32 | + internal: AppServiceInternal |
| 33 | + uploadManager: UploadManager |
| 34 | + db: DatabaseAdapter |
| 35 | + fsIO: FsIOAdapter |
| 36 | + paths: ReturnType<typeof getPaths> |
| 37 | + bootstrap: Bootstrap |
| 38 | +} |
| 39 | + |
| 40 | +export type CreateDatabaseFn = (path: string) => DatabaseAdapter |
| 41 | + |
| 42 | +/** |
| 43 | + * Bundle of auth + SDK adapters chosen at startup. Holds the strategy for |
| 44 | + * connecting the SDK so callers don't need to know whether they're running |
| 45 | + * against a real indexer or a MockSdk. |
| 46 | + */ |
| 47 | +export type Bootstrap = { |
| 48 | + authAdapters: SdkAuthAdapters |
| 49 | + sdkAuth: NodeSdkAuthResult |
| 50 | + /** Set only when running with `SIA_TEST_MODE=1` — pre-attached during bootstrap. */ |
| 51 | + testSdkAdapter?: SdkAdapter |
| 52 | + /** Performs the SDK connection (real handshake or test-mode short-circuit). */ |
| 53 | + connect: (app: CliApp) => Promise<boolean> |
| 54 | +} |
| 55 | + |
| 56 | +async function buildBootstrap(): Promise<Bootstrap> { |
| 57 | + if (isTestMode) { |
| 58 | + const { createTestBootstrap } = await import('./testMode') |
| 59 | + return createTestBootstrap() |
| 60 | + } |
| 61 | + return createRealBootstrap() |
| 62 | +} |
| 63 | + |
| 64 | +function createRealBootstrap(): Bootstrap { |
| 65 | + const sdkAuth = createNodeSdkAuthAdapter() |
| 66 | + return { |
| 67 | + authAdapters: sdkAuth.adapters, |
| 68 | + sdkAuth, |
| 69 | + async connect(app) { |
| 70 | + const indexerURL = await app.service.settings.getIndexerURL() |
| 71 | + const keyBytes = await app.service.auth.getAppKey(indexerURL) |
| 72 | + if (!keyBytes) return false |
| 73 | + |
| 74 | + const keyHex = uint8ToHex(new Uint8Array(keyBytes)) |
| 75 | + await sdkAuth.adapters.createBuilder(indexerURL, JSON.stringify(APP_META)) |
| 76 | + |
| 77 | + const connected = await sdkAuth.adapters.connectWithKey(keyHex) |
| 78 | + if (!connected) return false |
| 79 | + |
| 80 | + const sdk = sdkAuth.getLastSdk() |
| 81 | + if (!sdk) return false |
| 82 | + |
| 83 | + app.internal.setSdk(createNodeSdkAdapter(sdk)) |
| 84 | + app.service.connection.setState({ isConnected: true }) |
| 85 | + app.internal.initUploader() |
| 86 | + return true |
| 87 | + }, |
| 88 | + } |
| 89 | +} |
| 90 | + |
| 91 | +export async function createCliAppService( |
| 92 | + dataDir?: string, |
| 93 | + opts?: { createDatabase?: CreateDatabaseFn }, |
| 94 | +): Promise<CliApp> { |
| 95 | + const dir = dataDir ?? getDataDir() |
| 96 | + const p = getPaths(dir) |
| 97 | + ensureDataDir(dir) |
| 98 | + |
| 99 | + const createDb = opts?.createDatabase ?? createBunDatabase |
| 100 | + const db = createDb(p.dbPath) |
| 101 | + await runMigrations(db, sortMigrations(coreMigrations)) |
| 102 | + |
| 103 | + const storage = createJsonFileStorage(p.storagePath) |
| 104 | + const secrets = createJsonFileStorage(p.secretsPath, { mode: 0o600 }) |
| 105 | + const crypto = createNodeCryptoAdapter() |
| 106 | + const fsIO = createNodeFsIO(p.filesDir) |
| 107 | + const uploaderAdapters = createNodeUploaderAdapters() |
| 108 | + const detectMimeType = createNodeDetectMimeType() |
| 109 | + const thumbnail = createSharpThumbnailAdapter() |
| 110 | + |
| 111 | + const bootstrap = await buildBootstrap() |
| 112 | + |
| 113 | + const { service, internal, uploadManager } = createAppService({ |
| 114 | + db, |
| 115 | + storage, |
| 116 | + secrets, |
| 117 | + crypto, |
| 118 | + fsIO, |
| 119 | + downloadObject: createNodeDownloadAdapter({ |
| 120 | + fsIO, |
| 121 | + getAppKey: async (indexerURL: string) => service.auth.getAppKey(indexerURL), |
| 122 | + }), |
| 123 | + uploader: uploaderAdapters, |
| 124 | + sdkAuth: bootstrap.authAdapters, |
| 125 | + thumbnail, |
| 126 | + detectMimeType, |
| 127 | + }) |
| 128 | + |
| 129 | + // In test mode the MockSdk attaches before connectSdk runs, so the daemon |
| 130 | + // comes up with an SDK already wired and connectSdk just flips the flag. |
| 131 | + if (bootstrap.testSdkAdapter) { |
| 132 | + internal.setSdk(bootstrap.testSdkAdapter) |
| 133 | + } |
| 134 | + |
| 135 | + return { service, internal, uploadManager, db, fsIO, paths: p, bootstrap } |
| 136 | +} |
| 137 | + |
| 138 | +/** Performs the SDK connection chosen at bootstrap (real or test-mode). */ |
| 139 | +export function connectSdk(app: CliApp): Promise<boolean> { |
| 140 | + return app.bootstrap.connect(app) |
| 141 | +} |
0 commit comments