refactor: expand home dir for absolute path resolution

buzsh · buzsh · commit 40f5c7d2be29 · 2024-05-20T14:36:50.000-06:00
diff --git a/swiftgen/src/utils/fileUtils.ts b/swiftgen/src/utils/fileUtils.ts
@@ -1,5 +1,6 @@
 import fs from 'fs';
 import path from 'path';
+import os from 'os';
 
 export function readConfig(configPath: string) {
   return JSON.parse(fs.readFileSync(configPath, 'utf8'));
@@ -12,7 +13,13 @@ export function ensureOutputDirExists(outputDir: string) {
 }
 
 export function writeSwiftCodeToFile(swiftCode: string, outputDir: string, outputFileName: string, outputSuffix: string) {
-  const resolvedOutputDir = path.isAbsolute(outputDir) ? outputDir : path.join(__dirname, outputDir);
+  // Expand `~` to the home directory
+  if (outputDir.startsWith('~')) {
+    outputDir = path.join(os.homedir(), outputDir.slice(1));
+  }
+
+  // Resolve the output directory
+  const resolvedOutputDir = path.isAbsolute(outputDir) ? outputDir : path.join(process.cwd(), 'dist', outputDir);
   ensureOutputDirExists(resolvedOutputDir);
 
   const outputFilePath = path.join(resolvedOutputDir, `${outputFileName}${outputSuffix}`);
