Bug fixes and app settings

config/default.ts

@@ -1,4 +1,8 @@
 export default {
+    app: {
+        name: 'Example App',
+        contact_email: 'noreply@example.net'
+    },
     log_level: "DEV",
     db_log_level: "ERROR",
     public_url: "http://localhost:4899",

package.json

@@ -1,6 +1,6 @@
 {
   "name": "wms-core",
-  "version": "0.10.2",
+  "version": "0.10.16",
   "description": "Node web framework",
   "repository": "git@gitlab.com:ArisuOngaku/wms-core.git",
   "author": "Alice Gaudon <alice@gaudon.pro>",

src/Application.ts

@@ -11,6 +11,7 @@ import {Type} from "./Utils";
 import LogRequestsComponent from "./components/LogRequestsComponent";
 import {ValidationBag} from "./db/Validator";
 import TemplateError = lib.TemplateError;
+import config from "config";
 
 export default abstract class Application {
     private readonly version: string;
@@ -42,7 +43,7 @@ export default abstract class Application {
     }
 
     public async start(): Promise<void> {
-        Logger.info(`${this.constructor.name} v${this.version} - hi`);
+        Logger.info(`${config.get('app.name')} v${this.version} - hi`);
         process.once('SIGINT', () => {
             this.stop().catch(console.error);
         });

src/auth/magic_link/MagicLinkAuthController.ts

@@ -11,13 +11,33 @@ import geoip from "geoip-lite";
 
 
 export default abstract class MagicLinkAuthController extends Controller {
-    public static async checkAndAuth(req: Request, magicLink: MagicLink): Promise<void> {
+    public static async checkAndAuth(req: Request, res: Response, magicLink: MagicLink): Promise<void> {
         if (magicLink.getSessionID() !== req.sessionID!) throw new BadOwnerMagicLink();
         if (!await magicLink.isAuthorized()) throw new UnauthorizedMagicLink();
         if (!await magicLink.isValid()) throw new InvalidMagicLink();
 
         // Auth
-        await req.authGuard.authenticateOrRegister(req.session!, magicLink);
+        try {
+            await req.authGuard.authenticateOrRegister(req.session!, magicLink);
+        } catch (e) {
+            if (e instanceof PendingApprovalAuthError) {
+                res.format({
+                    json: () => {
+                        res.json({
+                            'status': 'warning',
+                            'message': `Your account is pending review. You'll receive an email once you're approved.`
+                        });
+                    },
+                    html: () => {
+                        req.flash('warning', `Your account is pending review. You'll receive an email once you're approved.`);
+                        res.redirectBack('/');
+                    }
+                });
+                return;
+            } else {
+                throw e;
+            }
+        }
     }
 
     protected readonly loginMagicLinkActionType: string = 'Login';
@@ -114,27 +134,7 @@ export default abstract class MagicLinkAuthController extends Controller {
             return;
         }
 
-        try {
-            await MagicLinkAuthController.checkAndAuth(req, magicLink);
-        } catch (e) {
-            if (e instanceof PendingApprovalAuthError) {
-                res.format({
-                    json: () => {
-                        res.json({
-                            'status': 'warning',
-                            'message': `Your account is pending review. You'll receive an email once you're approved.`
-                        });
-                    },
-                    default: () => {
-                        req.flash('warning', `Your account is pending review. You'll receive an email once you're approved.`);
-                        res.redirectBack('/');
-                    }
-                });
-                return;
-            } else {
-                throw e;
-            }
-        }
+        await MagicLinkAuthController.checkAndAuth(req, res, magicLink);
 
         // Auth success
         const username = req.models.user?.name;

src/auth/models/User.ts

@@ -12,8 +12,8 @@ export default class User extends Model {
     }
 
     public name?: string;
-    public approved: boolean = false;
-    public is_admin: boolean = false;
+    public approved!: boolean;
+    public is_admin!: boolean;
     public created_at?: Date;
     public updated_at?: Date;
 
@@ -24,6 +24,12 @@ export default class User extends Model {
 
     public readonly mainEmail = this.emails.cloneReduceToOne().constraint(q => q.where('main', true));
 
+    constructor(data: any) {
+        super(data);
+        if (this.approved === undefined) this.approved = false;
+        if (this.is_admin === undefined) this.is_admin = false;
+    }
+
     protected init(): void {
         this.addProperty<string>('name', new Validator().acceptUndefined().between(3, 64));
         if (User.isApprovalMode()) this.addProperty<boolean>('approved', new Validator().defined());

src/components/NunjucksComponent.ts

@@ -29,6 +29,9 @@ export default class NunjucksComponent extends ApplicationComponent<void> {
             req.env = env;
             res.locals.url = req.url;
             res.locals.params = () => req.params;
+
+            res.locals.app = config.get('app');
+
             next();
         });
     }

src/db/Model.ts

@@ -224,7 +224,7 @@ export default abstract class Model {
         for (const indexField of this.getPrimaryKeyFields()) {
             query = query.where(indexField, this[indexField]);
         }
-        return typeof (await query.first()) !== 'undefined';
+        return (await query.limit(1).execute()).results.length > 0;
     }
 
     public async delete(): Promise<void> {

src/db/ModelQuery.ts

@@ -190,8 +190,10 @@ export default class ModelQuery<M extends Model> {
         // Eager loading execute
         for (const relationName of this.relations) {
             const relations = relationMap[relationName];
-            const allModels = await relations[0].eagerLoad(relations);
-            await Promise.all(relations.map(r => r.populate(allModels)));
+            if (relations.length > 0) {
+                const allModels = await relations[0].eagerLoad(relations);
+                await Promise.all(relations.map(r => r.populate(allModels)));
+            }
         }
 
         return models;

src/db/ModelRelation.ts

@@ -31,6 +31,11 @@ export default abstract class ModelRelation<S extends Model, O extends Model, R
         return this.cachedModels;
     }
 
+    public getOrFail(): R {
+        if (!this.cachedModels) throw new Error('Models were not fetched');
+        return this.cachedModels;
+    }
+
     public abstract async eagerLoad(relations: ModelRelation<S, O, R>[]): Promise<ModelQueryResult<O>>;
 
     public abstract async populate(models: ModelQueryResult<O>): Promise<void>;