feat: core improvements + local fs provider + UI fixes

2018-07-29 22:23:33 -04:00
parent 803d86ff63
commit 2817c72ec3
65 changed files with 482 additions and 264 deletions
--- a/server/models/authentication.js
+++ b/server/models/authentication.js
@@ -0,0 +1,99 @@
+const Model = require('objection').Model
+const fs = require('fs-extra')
+const path = require('path')
+const _ = require('lodash')
+const yaml = require('js-yaml')
+const commonHelper = require('../helpers/common')
+
+/* global WIKI */
+
+/**
+ * Authentication model
+ */
+module.exports = class Authentication extends Model {
+  static get tableName() { return 'authentication' }
+
+  static get jsonSchema () {
+    return {
+      type: 'object',
+      required: ['key', 'title', 'isEnabled', 'useForm'],
+
+      properties: {
+        id: {type: 'integer'},
+        key: {type: 'string'},
+        title: {type: 'string'},
+        isEnabled: {type: 'boolean'},
+        useForm: {type: 'boolean'},
+        config: {type: 'object'},
+        selfRegistration: {type: 'boolean'},
+        domainWhitelist: {type: 'object'},
+        autoEnrollGroups: {type: 'object'}
+      }
+    }
+  }
+
+  static async getStrategies() {
+    const strategies = await WIKI.models.authentication.query()
+    return strategies.map(str => ({
+      ...str,
+      domainWhitelist: _.get(str.domainWhitelist, 'v', []),
+      autoEnrollGroups: _.get(str.autoEnrollGroups, 'v', [])
+    }))
+  }
+
+  static async refreshStrategiesFromDisk() {
+    try {
+      const dbStrategies = await WIKI.models.authentication.query()
+
+      // -> Fetch definitions from disk
+      const authDirs = await fs.readdir(path.join(WIKI.SERVERPATH, 'modules/authentication'))
+      let diskStrategies = []
+      for (let dir of authDirs) {
+        const def = await fs.readFile(path.join(WIKI.SERVERPATH, 'modules/authentication', dir, 'definition.yml'), 'utf8')
+        diskStrategies.push(yaml.safeLoad(def))
+      }
+
+      let newStrategies = []
+      _.forEach(diskStrategies, strategy => {
+        if (!_.some(dbStrategies, ['key', strategy.key])) {
+          newStrategies.push({
+            key: strategy.key,
+            title: strategy.title,
+            isEnabled: false,
+            useForm: strategy.useForm,
+            config: _.transform(strategy.props, (result, value, key) => {
+              if (_.isPlainObject(value)) {
+                let cfgValue = {
+                  type: value.type.toLowerCase(),
+                  value: !_.isNil(value.default) ? value.default : commonHelper.getTypeDefaultValue(value.type)
+                }
+                if (_.isArray(value.enum)) {
+                  cfgValue.enum = value.enum
+                }
+                _.set(result, key, cfgValue)
+              } else {
+                _.set(result, key, {
+                  type: value.toLowerCase(),
+                  value: commonHelper.getTypeDefaultValue(value)
+                })
+              }
+              return result
+            }, {}),
+            selfRegistration: false,
+            domainWhitelist: { v: [] },
+            autoEnrollGroups: { v: [] }
+          })
+        }
+      })
+      if (newStrategies.length > 0) {
+        await WIKI.models.authentication.query().insert(newStrategies)
+        WIKI.logger.info(`Loaded ${newStrategies.length} new authentication strategies: [ OK ]`)
+      } else {
+        WIKI.logger.info(`No new authentication strategies found: [ SKIPPED ]`)
+      }
+    } catch (err) {
+      WIKI.logger.error(`Failed to scan or load new authentication providers: [ FAILED ]`)
+      WIKI.logger.error(err)
+    }
+  }
+}