Merge pull request #11 from SecJS/refactor/len-primary-key-insert-update

feat: add timestamps property to table builder

Author: jlenon7

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@secjs/database",
-  "version": "1.0.6",
+  "version": "1.0.7",
   "description": "Handle your application database with factories, seeders and query builder in Node.js",
   "license": "MIT",
   "author": "João Lenon <lenon@secjs.com.br>",

src/Builders/TableBuilder.ts

@@ -8,6 +8,7 @@
  */
 
 import { ColumnBuilder } from './ColumnBuilder'
+import { String } from '@secjs/utils'
 
 export class TableBuilder {
   private columns: ColumnBuilder[] = []
@@ -63,4 +64,35 @@ export class TableBuilder {
 
     return column
   }
+
+  timestamps(
+    useTimestamps = true,
+    defaultToNow = true,
+    useCamelCase = false,
+  ): void {
+    let createdAtName = 'created_at'
+    let updatedAtName = 'updated_at'
+
+    if (useCamelCase) {
+      createdAtName = String.toCamelCase(createdAtName)
+      updatedAtName = String.toCamelCase(updatedAtName)
+    }
+
+    const createdAtColumn = new ColumnBuilder(createdAtName, 'date')
+    const updatedAtColumn = new ColumnBuilder(updatedAtName, 'date')
+
+    if (defaultToNow) {
+      let now: number | Date = new Date()
+
+      if (useTimestamps) {
+        now = Date.now()
+      }
+
+      createdAtColumn.defaultTo(now)
+      updatedAtColumn.defaultTo(now)
+    }
+
+    this.columns.push(createdAtColumn)
+    this.columns.push(updatedAtColumn)
+  }
 }