wiki/server/models/assetFolders.js

const Model = require('objection').Model
const _ = require('lodash')

/* global WIKI */

/**
 * Users model
 */
module.exports = class AssetFolder extends Model {
  static get tableName() { return 'assetFolders' }

  static get jsonSchema () {
    return {
      type: 'object',

      properties: {
        id: {type: 'integer'},
        name: {type: 'string'},
        slug: {type: 'string'}
      }
    }
  }

  static get relationMappings() {
    return {
      parent: {
        relation: Model.BelongsToOneRelation,
        modelClass: AssetFolder,
        join: {
          from: 'assetFolders.folderId',
          to: 'assetFolders.id'
        }
      }
    }
  }

  /**
   * Get full folder hierarchy starting from specified folder to root
   *
   * @param {Number} folderId Id of the folder
   */
  static async getHierarchy (folderId) {
    let hier
    if (WIKI.config.db.type === 'mssql') {
      hier = await WIKI.models.knex.with('ancestors', qb => {
        qb.select('id', 'name', 'slug', 'parentId').from('assetFolders').where('id', folderId).unionAll(sqb => {
          sqb.select('a.id', 'a.name', 'a.slug', 'a.parentId').from('assetFolders AS a').join('ancestors', 'ancestors.parentId', 'a.id')
        })
      }).select('*').from('ancestors')
    } else {
      hier = await WIKI.models.knex.withRecursive('ancestors', qb => {
        qb.select('id', 'name', 'slug', 'parentId').from('assetFolders').where('id', folderId).union(sqb => {
          sqb.select('a.id', 'a.name', 'a.slug', 'a.parentId').from('assetFolders AS a').join('ancestors', 'ancestors.parentId', 'a.id')
        })
      }).select('*').from('ancestors')
    }
    // The ancestors are from children to grandparents, must reverse for correct path order.
    return _.reverse(hier)
  }

  /**
   * Get full folder paths
   */
  static async getAllPaths () {
    const all = await WIKI.models.assetFolders.query()
    let folders = {}
    all.forEach(fld => {
      _.set(folders, fld.id, fld.slug)
      let parentId = fld.parentId
      while (parentId !== null || parentId > 0) {
        const parent = _.find(all, ['id', parentId])
        _.set(folders, fld.id, `${parent.slug}/${_.get(folders, fld.id)}`)
        parentId = parent.parentId
      }
    })
    return folders
  }

  /**
   * Get asset folder by resolving its path from root
   * @param {string} path Folder path, e.g. 'abc/def/ghi'
   */
  static async getFolderByPath(path) {
    const segments = String(path).split('/').filter(Boolean)
    if (segments.length === 0) return null
    const db = this.knex()
    const targetPath = segments.join('/')
    let results
    if (WIKI.config.db.type === 'mssql') {
      results = await db
        .with('folder_path', (qb) => {
          qb.select('af.*').select(db.raw('CAST(af.slug AS VARCHAR(MAX)) AS current_path')).from('assetFolders as af').whereNull('af.parentId').unionAll((uqb) => {
            uqb.select('af.*').select(db.raw("CAST(fp.current_path + '/' + af.slug AS VARCHAR(MAX)) AS current_path")).from('assetFolders as af').join('folder_path as fp', 'af.parentId', 'fp.id')
          })
        }).select('*').from('folder_path').where('current_path', targetPath).limit(1)
    } else {
      results = await db
        .withRecursive('folder_path', (qb) => {
          qb.select('af.*').select(db.raw('af.slug::TEXT AS current_path')).from('assetFolders as af').whereNull('af.parentId').unionAll((uqb) => {
            uqb.select('af.*').select(db.raw("fp.current_path || '/' || af.slug AS current_path")).from('assetFolders as af').join('folder_path as fp', 'af.parentId', 'fp.id')
          }, true)
        }).select('*').from('folder_path').where('current_path', targetPath).limit(1)
    }
    return results[0] || null
  }
}