refactor(controller): share download operation errors

seroxdesign committed 3 months ago

Commit 9c735a31 parent f7ad3903 files · +17 −25Browse files

Showing the changes to controller/src/modules (3 of 3 files). Show all

Unified · Split

3 files

 import { EngineOperationError } from "../engine-spec";
 import type { DownloadFileInfo, DownloadStatus, ModelDownload } from "../types";
 import type { DownloadStore } from "./download-store";
+import { attempt, operationError } from "./download-operation";
 import {
   buildHuggingFaceFileList,
   fetchEffect,
 const toTimestamp = (): string => new Date().toISOString();
-const operationError = (operation: string, cause: unknown): EngineOperationError =>
-  new EngineOperationError({
-    operation,
-    message: cause instanceof Error ? cause.message : String(cause),
-  });
-const attempt = <A>(operation: string, evaluate: () => A): Effect.Effect<A, EngineOperationError> =>
-  Effect.try({
-    try: evaluate,
-    catch: (cause) => operationError(operation, cause),
-  });
 const closeWriter = (
   writer: ReturnType<typeof createWriteStream>,
 ): Effect.Effect<void, EngineOperationError> =>

@@ -0,0 +1,14 @@
	1	import { Effect } from "effect";
	2	import { EngineOperationError } from "../engine-spec";
	3
	4	export const operationError = (operation: string, cause: unknown): EngineOperationError =>
	5	new EngineOperationError({
	6	operation,
	7	message: cause instanceof Error ? cause.message : String(cause),
	8	});
	9
	10	export const attempt = <A>(operation: string, evaluate: () => A): Effect.Effect<A, EngineOperationError> =>
	11	Effect.try({
	12	try: evaluate,
	13	catch: (cause) => operationError(operation, cause),
	14	});

 import { Effect, Option, Schema } from "effect";
 import { openSqliteDatabase } from "../../../stores/sqlite";
-import { EngineOperationError } from "../engine-spec";
+import type { EngineOperationError } from "../engine-spec";
 import type { ModelDownload } from "../types";
+import { attempt, operationError } from "./download-operation";
 const DownloadFileSchema = Schema.Struct({
   path: Schema.String,
   error: Schema.NullOr(Schema.String),
 });
-const operationError = (operation: string, cause: unknown): EngineOperationError =>
-  new EngineOperationError({
-    operation,
-    message: cause instanceof Error ? cause.message : String(cause),
-  });
-const attempt = <A>(operation: string, evaluate: () => A): Effect.Effect<A, EngineOperationError> =>
-  Effect.try({
-    try: evaluate,
-    catch: (cause) => operationError(operation, cause),
-  });
 const decodeDownload = (value: unknown): Effect.Effect<ModelDownload, EngineOperationError> =>
   attempt("parse-download-record", () =>
     typeof value === "string" ? JSON.parse(value) : value,