Initial commit

Author: makinori

.gitignore

@@ -0,0 +1,4 @@
+dist
+node_modules
+.vscode
+test-output.webp

.prettierrc

@@ -0,0 +1,11 @@
+{
+	"arrowParens": "avoid",
+	"bracketSpacing": true,
+	"htmlWhitespaceSensitivity": "css",
+	"printWidth": 80,
+	"semi": true,
+	"singleQuote": false,
+	"tabWidth": 4,
+	"trailingComma": "all",
+	"useTabs": true
+}

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2022 [fullname]
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -0,0 +1,29 @@
+# Rembg Node
+
+Rembg lets you **easily remove backgrounds** from images using the [U2-Net ai model](https://github.com/xuebinqin/U-2-Net)
+
+This is a loose port of the original [Rembg for Python](https://github.com/danielgatis/rembg), big thanks to [@danielgatis](https://github.com/danielgatis)
+
+It uses [sharp](https://github.com/lovell/sharp) for input and output so you can easily integrate it
+
+The masking algorithm isn't fully complete yet, but the results are pretty awesome already!
+
+## Example (TypeScript)
+
+```ts
+import { RemBg } from "rembg-node";
+import sharp from "sharp";
+
+(async () => {
+	const input = sharp("test-input.jpg");
+
+	// optional arguments
+	const rembg = new RemBg({
+		logging: true,
+	});
+
+	const output = await rembg.remove(input);
+
+	await output.webp().toFile("test-output.webp");
+})();
+```

package.json

@@ -0,0 +1,28 @@
+{
+	"name": "rembg-sharp",
+	"version": "1.0.0",
+	"description": "Rembg is a tool to remove images background.",
+	"repository": "https://github.com/makifoxgirl/rembg-sharp",
+	"author": "Maki",
+	"license": "MIT",
+	"main": "dist/index.js",
+	"types": "dist/index.d.ts",
+	"files": [
+		"/dist"
+	],
+	"scripts": {
+		"start": "tsc -w",
+		"build": "tsc",
+		"prepublishOnly": "npm run build"
+	},
+	"dependencies": {
+		"axios": "^0.27.2",
+		"onnxruntime-node": "^1.11.0",
+		"sharp": "^0.30.7"
+	},
+	"devDependencies": {
+		"@types/node": "^18.0.3",
+		"@types/sharp": "^0.30.4",
+		"typescript": "^4.7.4"
+	}
+}

src/gdown.ts

@@ -0,0 +1,62 @@
+import axios from "axios";
+import * as fs from "fs/promises";
+import * as stream from "stream";
+import * as util from "util";
+
+const pipeline = util.promisify(stream.pipeline);
+
+function streamToString(stream: NodeJS.ReadableStream): Promise<string> {
+	return new Promise((resolve, reject) => {
+		const data = [];
+		stream.on("data", chunk => {
+			data.push(chunk);
+		});
+		stream.on("end", () => {
+			resolve(Buffer.concat(data).toString("utf8"));
+		});
+		stream.on("error", error => {
+			reject(error);
+		});
+	});
+}
+
+// super primitive lol but works (for now)
+export async function gdown(id: string, savePath: string) {
+	const url = "https://drive.google.com/uc?id=" + id;
+
+	const res = await axios({ url, responseType: "stream" });
+
+	const downloadable = res.headers["content-disposition"] != null;
+
+	let fh: fs.FileHandle;
+	try {
+		if (downloadable) {
+			fh = await fs.open(savePath, "w+");
+			await pipeline(res.data, fh.createWriteStream());
+		} else {
+			// aa fuck intermediary page
+			const html = await streamToString(res.data);
+
+			const urlMatches = html.match(/action="([^]+?\/uc\?id=[^]+?)"/i);
+			if (urlMatches == null)
+				throw new Error("Failed to download gdrive link");
+
+			const actualRes = await axios({
+				method: "post",
+				url: urlMatches[1].replace(/&amp;/g, "&"),
+				responseType: "stream",
+			});
+
+			const actualDownloadable =
+				actualRes.headers["content-disposition"] != null;
+
+			if (!actualDownloadable)
+				throw new Error("Failed to download gdrive link");
+
+			fh = await fs.open(savePath, "w+");
+			await pipeline(actualRes.data, fh.createWriteStream());
+		}
+	} finally {
+		if (fh) fh.close();
+	}
+}