wrapture 0.0.1

package/src/wrapture.ts

@@ -0,0 +1,51 @@
+/* eslint-disable no-console */
+/* global process */
+import chalk from 'chalk';
+import { Command } from 'commander';
+
+import { existsSync } from 'node:fs';
+import path from 'node:path';
+
+import { convert } from './utils/convert.js';
+import { generateWrapper } from './utils/generate-wrapper.js';
+
+const program = new Command();
+
+program
+  .name('wrapture')
+  .description('🌀 One-click model exporter: from PyTorch to Web-ready JS/TS')
+  .version('0.1.0')
+  .requiredOption('-i, --input <file>', 'Path to the PyTorch model (.pt)')
+  .requiredOption(
+    '-o, --output <dir>',
+    'Output directory for the wrapped model'
+  )
+  .option('--quantize', 'Apply quantization to reduce model size')
+  .option('--format <type>', 'Export format: onnx (default)', 'onnx')
+  .option(
+    '--backend <backend>',
+    'Inference backend: webgpu | wasm | cpu',
+    'webgpu'
+  )
+  .action(async (opts) => {
+    const input = path.resolve(opts.input);
+    const output = path.resolve(opts.output);
+
+    if (!existsSync(input)) {
+      console.error(chalk.red(`Input file not found: ${input}`));
+      process.exit(1);
+    }
+
+    console.log(chalk.cyan('✨ Wrapture: Exporting model...'));
+
+    try {
+      await convert(input, output, opts);
+      await generateWrapper(output, opts);
+      console.log(chalk.green('✅ Done! Your model is wrapped and ready.'));
+    } catch (err) {
+      console.error(chalk.red('❌ Failed to export model:'), err);
+      process.exit(1);
+    }
+  });
+
+program.parse(process.argv);

package/tsconfig.json

@@ -0,0 +1,27 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "outDir": "bin",
+    "module": "Node16",
+    "lib": ["ES2022", "dom", "dom.iterable"],
+    "sourceMap": true,
+    "declaration": true,
+    "declarationMap": true,
+    "declarationDir": "./dts",
+    "noImplicitReturns": true,
+    "noImplicitThis": true,
+    "noImplicitAny": false,
+    "strictNullChecks": true,
+    "skipLibCheck": true,
+    "esModuleInterop": true,
+    "resolveJsonModule": true
+  },
+  "exclude": [
+    "node_modules",
+    "dist",
+    "bin",
+    "scripts",
+    "dts",
+    "**/__tests__/**"
+  ]
+}

package/tsup.config.ts

@@ -0,0 +1,14 @@
+/* eslint-disable import/no-unused-modules */
+import { defineConfig } from 'tsup';
+
+export default defineConfig({
+  entry: ['src/wrapture.ts'],
+  format: ['esm'],
+  target: 'node16',
+  outDir: 'bin',
+  clean: true,
+  tsconfig: './tsconfig.json',
+  banner: {
+    js: '#!/usr/bin/env node'
+  }
+});

package/typedoc.json

@@ -0,0 +1,33 @@
+{
+  "$schema": "https://typedoc-plugin-markdown.org/schema.json",
+  "plugin": [
+    "typedoc-plugin-markdown",
+    "typedoc-plugin-remark",
+    "typedoc-plugin-mdn-links",
+    "typedoc-plugin-rename-defaults",
+    "./custom-typedoc-plugin.js"
+  ],
+  "out": "./api",
+
+  "outputFileStrategy": "modules",
+  "interfacePropertiesFormat": "table",
+  "classPropertiesFormat": "table",
+  "enumMembersFormat": "table",
+  "typeDeclarationFormat": "htmlTable",
+  "propertyMembersFormat": "table",
+  "theme": "custom-markdown-theme",
+  "parametersFormat": "table",
+  "entryPoints": ["src"],
+  "entryPointStrategy": "expand",
+  "readme": "./public/docs/README.md",
+  "gitRevision": "main",
+  "githubPages": false,
+  "mergeReadme": true,
+  "useCodeBlocks": true,
+  "expandObjects": true,
+  "remarkPlugins": [
+    "unified-prettier",
+    ["remark-github", { "repository": "https://github.com/phun-ky/wrapture" }],
+    ["remark-toc", { "heading": "Table of Contents", "maxDepth": 3 }]
+  ]
+}