Merge pull request #1 from ponsoc/introducing-some-improvements

Introducing some improvements

Author: ponsoc

README.md

@@ -71,6 +71,8 @@ main()
 
 `async connect()`: Connects to the database.
 
+`async enableForeignKeySupport`: Enables foreign keys on the connection.
+
 `async disconnect()`: Disconnects from the database.
 
 `async generate(tables: Array)`: Generates the database file based on the configuration. The tables parameter is an array of objects, each object represents a table in the database. The object has properties as described in the configuration section.
@@ -85,10 +87,9 @@ To enable debugging set the `DEBUG` environment variable to `sqlite-data-generat
 
 ## Roadmap
 
-- Fix foreign key constraints issue
 - Extend the example configuration to include more complex examples
 - Add the ability to generate multiple rows of data for a single row in the source table
 
 ## Known Issues
 
-- Foreign key constraints can be configured however they are not properly set up in the database.
+- N/A

package.json

@@ -1,6 +1,6 @@
 {
   "name": "sqlite-data-generator",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "A CLI for generating SQLite database files with example data",
   "main": "src/app.js",
   "scripts": {

src/app.js

@@ -9,6 +9,7 @@ const tables = require("./config/tables.js")({ faker, db });
 async function main() {
   try {
     await db.connect();
+    await db.enableForeignKeySupport();
     await db.generate(tables);
   } catch (error) {
     throw error;

src/config/tables.js

@@ -4,11 +4,11 @@ module.exports = function (dependencies) {
       name: "users",
       fields: [
         { name: "id", type: "INTEGER PRIMARY KEY" },
-        { name: "name", type: "TEXT", generator: () => dependencies.faker.name.findName() },
+        { name: "name", type: "TEXT", generator: () => dependencies.faker.person.fullName() },
         { name: "email", type: "TEXT", generator: () => dependencies.faker.internet.email() },
         { name: "age", type: "INTEGER", generator: () => dependencies.faker.number.int({ min: 18, max: 90 }) },
       ],
-      numRows: 10,
+      rows: 10,
     },
     {
       name: "posts",
@@ -19,13 +19,16 @@ module.exports = function (dependencies) {
         {
           name: "author_id",
           type: "INTEGER",
-          generator: async () => await dependencies.db.getRandomIdFromTable("users"),
+          generator: async () => {
+            const row = await dependencies.db.getRandomRowFromTable("users")
+            return row.id;
+          }
         },
         {
           type: "FOREIGN KEY(author_id) REFERENCES users(id)",
         },
       ],
-      numRows: 20,
+      rows: 20,
     },
   ];
 };

src/lib/SQLiteDataGenerator.js

@@ -23,12 +23,26 @@ class SQLiteDataGenerator {
     });
   }
 
+
+  async enableForeignKeySupport() {
+    return await new Promise((resolve, reject) => {
+      this.db.run('PRAGMA foreign_keys = ON', (err) => {
+        if (err) {
+          reject(new Error(`Error enabling foreign key support: ${err.message}`));
+        } else {
+          resolve();
+        }
+      });
+    })
+  }
+
+
   /**
    * Disconnects from the database.
    * @returns {Promise<void>} A promise that resolves when the disconnection is successful.
    */
   async disconnect() {
-    return await new Promise((resolve) => {
+    return await new Promise((resolve, reject) => {
       this.db.close((err) => {
         if (err) {
           reject(new Error(`Error closing the database connection: ${err.message}`));
@@ -46,6 +60,7 @@ class SQLiteDataGenerator {
    * @returns {Promise} - A promise that resolves when the data generation is complete.
    */
   async generate(tables) {
+    this.debug('Start generating...')
     for (const table of tables) {
       await this.#createTable(table);
       await this.#insertExampleData(table);
@@ -111,9 +126,14 @@ class SQLiteDataGenerator {
    * @returns {Promise<void>} - A promise that resolves when the example data is inserted.
    */
   async #insertExampleData(table) {
+    this.debug(`Generating data for table '${table.name}'`);
     const variables = await this.#resolveVariables(table.vars);
     const fieldsWithName = table.fields.filter((field) => field.name);
 
+    this.debug('table', table);
+    this.debug('variables', variables);
+    this.debug('fiedsWithName', fieldsWithName);
+
     if (typeof table.rows === "object") {
       await this.#eachRow(table.rows.sourceTable, table.name, fieldsWithName, variables, table.rows.filter);
     } else {