UPDATE: added check to schema attributes

Author: FridayCandour

db.ts

@@ -49,8 +49,9 @@ export const works = new Schema<{
     rating: {
       type: "INTEGER",
       default() {
-        return 1;
+        return "1";
       },
+      check: ["1", "1.5", "2", "2.5", "3", "3.5", "4", "4.5", "5"],
     },
     createdAt: {
       type: "DATETIME",

src/index.ts

@@ -18,6 +18,7 @@ interface SchemaColumnOptions {
   unique?: boolean;
   default?: () => string;
   target?: string;
+  check?: string[];
 }
 
 type Params = string | number | null | boolean;
@@ -674,6 +675,13 @@ export class Schema<Model extends Record<string, any> = {}> {
           (this.columns[col].target
             ? " REFERENCES " + this.columns[col].target + "(id)"
             : "") +
+          (this.columns[col].check?.length
+            ? "  CHECK (" +
+              col +
+              " IN (" +
+              this.columns[col].check.map((c) => "'" + c + "'").join(",") +
+              ")) "
+            : "") +
           (this.columns[col].default
             ? "  DEFAULT " + this.columns[col].default()
             : "") +