Initial commit: ACE-Step UI - Open source music generation interface

server/src/services/storage/local.ts

@@ -0,0 +1,60 @@
+import { writeFile, unlink, stat, mkdir, copyFile } from 'fs/promises';
+import path from 'path';
+import { fileURLToPath } from 'url';
+import type { StorageProvider } from './index.js';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+const AUDIO_DIR = path.join(__dirname, '../../../public/audio');
+
+export class LocalStorageProvider implements StorageProvider {
+  private audioDir: string;
+
+  constructor() {
+    this.audioDir = AUDIO_DIR;
+  }
+
+  async upload(key: string, data: Buffer, _contentType: string): Promise<string> {
+    const filepath = path.join(this.audioDir, key);
+    await mkdir(path.dirname(filepath), { recursive: true });
+    await writeFile(filepath, data);
+    return `/audio/${key}`;
+  }
+
+  async getUrl(key: string, _expiresIn?: number): Promise<string> {
+    return `/audio/${key}`;
+  }
+
+  getPublicUrl(key: string): string {
+    return `/audio/${key}`;
+  }
+
+  async delete(key: string): Promise<void> {
+    const filepath = path.join(this.audioDir, key);
+    try {
+      await unlink(filepath);
+    } catch (err) {
+      const error = err as NodeJS.ErrnoException;
+      if (error.code !== 'ENOENT') {
+        throw error;
+      }
+    }
+  }
+
+  async exists(key: string): Promise<boolean> {
+    const filepath = path.join(this.audioDir, key);
+    try {
+      await stat(filepath);
+      return true;
+    } catch {
+      return false;
+    }
+  }
+
+  async copy(sourceKey: string, destKey: string): Promise<void> {
+    const sourcePath = path.join(this.audioDir, sourceKey);
+    const destPath = path.join(this.audioDir, destKey);
+    await mkdir(path.dirname(destPath), { recursive: true });
+    await copyFile(sourcePath, destPath);
+  }
+}