增加自动下载

src/cv/addon.ts

@@ -1,18 +1,21 @@
+import path from "path";
 import { LINE_8, MARKER_CROSS } from "./consts";
 
 export const config = {
-	ADDON_PATH: "../../build/cv.node",
+	ADDON_PATH: path.join(__dirname, "../../build/cv.node"),
 	DEFAULT_LINE_TYPE: LINE_8,
 	DEFAULT_THICKNESS: 1,
 	DEFAULT_MARKER: MARKER_CROSS,
 };
 
+export type AddonConfig = typeof config;
+
 export function getAddon() {
 	return require(config.ADDON_PATH);
 }
 
-export function getConfig<N extends keyof typeof config>(name: N): typeof config[N] { return config[name]; }
-export function setConfig<N extends keyof typeof config>(name: N, value: typeof config[N]) { config[name] = value; }
+export function getConfig<N extends keyof AddonConfig>(name: N): AddonConfig[N] { return config[name]; }
+export function setConfig<N extends keyof AddonConfig>(name: N, value: AddonConfig[N]) { config[name] = value; }
 
 export function CVMat() { return getAddon().Mat; }
 

src/cv/common.ts

@@ -1,11 +1,10 @@
-import { getAddon, getConfig } from "./addon";
+export { getConfig as C } from "./addon";
 import { Mat } from "./mat";
 
 export type TypedArray = Uint8Array | Int8Array | Uint16Array | Int16Array | Uint32Array | Int32Array | BigUint64Array | BigInt64Array | Float32Array | Float64Array
 
 export function FromCV(mat: any): Mat { return new (Mat as any)(mat); }
 export function M(mat: Mat) { return (mat as any).__mat__; }
-export function C(name: Parameters<typeof getConfig>[0]) { return getConfig(name); }
 
 export function resolveArgs<R extends Array<any>>(args: any[], checker: boolean | (() => boolean)): R {
 	if (typeof checker === "function") checker = checker();

src/cv/download.ts

@@ -0,0 +1,77 @@
+import os from "os";
+import fs from "fs";
+import path from "path";
+import { C } from "./common";
+
+const URLS = {
+	GITHUB: `https://github.com/kangkang520/node-addons/releases/download/cv{{version}}/cv_{{platform}}_{{arch}}.node`,
+	URNAS: `http://git.urnas.cn:5200/yizhi-js-lib/opencv/releases/download/{{version}}/cv_{{platform}}_{{arch}}.node`,
+}
+
+function getURL(template: string) {
+	const version = require("../../package.json").version;
+	let platform = "";
+	let arch = "";
+	switch (os.platform()) {
+		case "win32":
+			platform = "windows";
+			break;
+		case "linux":
+			platform = "linux";
+			break;
+		default:
+			throw new Error(`Unsupported platform: ${os.platform()}, Please compile the addon yourself.`);
+	}
+	switch (os.arch()) {
+		case "x64":
+			arch = "x64";
+			break;
+		default:
+			throw new Error(`Unsupported architecture: ${os.arch()}, Please compile the addon yourself.`);
+	}
+
+	return template.replaceAll("{{version}}", version).replaceAll("{{platform}}", platform).replaceAll("{{arch}}", arch);
+}
+
+async function getStream() {
+	for (const [name, url] of Object.entries(URLS)) {
+		try {
+			return await fetch(getURL(url)).then(res => {
+				if (res.status != 200) throw new Error("Failed to download addon.");
+				return res.blob().then(b => b.stream());
+			})
+		} catch (e) { }
+	}
+	throw new Error("Failed to download addon.");
+}
+
+export async function downloadAddon(savename?: string) {
+	const defaultAddon = path.resolve(process.cwd(), C("ADDON_PATH"));
+	const saveName = savename ? path.resolve(path.dirname(defaultAddon), savename) : defaultAddon;
+	if (fs.existsSync(saveName)) return saveName;
+	
+	await fs.promises.mkdir(path.dirname(saveName), { recursive: true });
+
+	const stream = await getStream();
+	const cacheFile = await new Promise<string>((resolve, reject) => {
+		const cacheFile = path.join(os.tmpdir(), Date.now() + ".cv.node");
+		let fsStream!: ReturnType<typeof fs.createWriteStream>;
+		stream.pipeTo(new WritableStream({
+			start(controller) {
+				fsStream = fs.createWriteStream(cacheFile);
+			},
+			async write(chunk, controller) {
+				await new Promise<void>((resolve, reject) => fsStream.write(chunk, err => err ? reject(err) : resolve()));
+			},
+			close() {
+				fsStream.end();
+				resolve(cacheFile);
+			},
+			abort() { }
+		})).catch(reject);
+	});
+
+	await fs.promises.cp(cacheFile, saveName);
+	await fs.promises.rm(cacheFile);
+	return saveName;
+}

src/cv/index.ts

@@ -1,4 +1,5 @@
 export { setConfig as config } from "./addon";
+export { downloadAddon } from "./download";
 export * from "./mat";
 export * from "./consts";
 export * from "./proc";

src/test.ts

@@ -2,6 +2,7 @@ import cv from ".";
 
 
 async function test() {
+	await cv.downloadAddon();
 	const fs = await import("fs");
 	// const buffer = fs.readFileSync("data/im1.jpeg")
 	const res = await cv.imread("test_data/im1.jpeg");