milodb 1.0.6 → 1.0.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "milodb",
-  "version": "1.0.6",
+  "version": "1.0.8",
   "main": "src/milodb.js",
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1"

package/src/cli.js

@@ -77,7 +77,7 @@ async function initDatabase() {
 // Parse command line arguments
 const args = process.argv.slice(2);
 
-// Handle the `init` command
+// Handle the init command
 if (args[0] === 'init') {
   initDatabase();
 } else {

package/src/milodb.js

@@ -3,21 +3,26 @@ const path = require('path');
 const { encrypt, decrypt } = require('./encrypt');
 
 class MiloDB {
-  constructor(dbPath = path.join(__dirname, 'db'), encryptData = false) {
+  constructor(dbPath = path.join(process.cwd(), 'db'), encryptData = false) {
     // Use the root folder for dbPath if not provided
     this.dbPath = dbPath;
     this.encryptData = encryptData;
     fs.ensureDirSync(this.dbPath); // Ensure that the DB path exists
   }
 
-  // Create a new table (i.e., a new JSON file)
   async createTable(tableName) {
     const tablePath = path.join(this.dbPath, `${tableName}.json`);
-    if (fs.existsSync(tablePath)) {
-      throw new Error(`Table ${tableName} already exists.`);
+  
+    try {
+      if (fs.existsSync(tablePath)) {
+        throw new Error(`Table ${tableName} already exists.`);
+      }
+      await fs.writeJson(tablePath, [], { spaces: 2 });
+      console.log(`Table ${tableName} created successfully.`);
+    } catch (err) {
+      console.error(`Failed to create table: ${err.message}`);
+      // Additional logic, like suggesting to delete/rename the table, can go here
     }
-    await fs.writeJson(tablePath, [], { spaces: 2 });
-    console.log(`Table ${tableName} created successfully.`);
   }
 
   // Add a record to a table