feat: import assets from disk module

client/components/admin.vue

@@ -26,7 +26,7 @@
             v-list-item(to='/pages', v-if='hasPermission([`manage:system`, `write:pages`, `manage:pages`, `delete:pages`])')
               v-list-item-avatar(size='24'): v-icon mdi-file-document-outline
               v-list-item-title {{ $t('admin:pages.title') }}
-              v-list-item-action.pr-3
+              v-list-item-action(style='min-width:auto;')
                 v-chip(x-small, :color='darkMode ? `grey darken-3-d4` : `grey lighten-5`')
                   .caption.grey--text {{ info.pagesTotal }}
             v-list-item(to='/theme', v-if='hasPermission([`manage:system`, `manage:theme`])')
@@ -38,13 +38,13 @@
             v-list-item(to='/groups', v-if='hasPermission([`manage:system`, `manage:groups`, `write:groups`])')
               v-list-item-avatar(size='24'): v-icon mdi-account-group
               v-list-item-title {{ $t('admin:groups.title') }}
-              v-list-item-action.pr-3
+              v-list-item-action(style='min-width:auto;')
                 v-chip(x-small, :color='darkMode ? `grey darken-3-d4` : `grey lighten-4`')
                   .caption.grey--text {{ info.groupsTotal }}
             v-list-item(to='/users', v-if='hasPermission([`manage:system`, `manage:groups`, `write:groups`, `manage:users`, `write:users`])')
               v-list-item-avatar(size='24'): v-icon mdi-account-box
               v-list-item-title {{ $t('admin:users.title') }}
-              v-list-item-action.pr-3
+              v-list-item-action(style='min-width:auto;')
                 v-chip(x-small, :color='darkMode ? `grey darken-3-d4` : `grey lighten-4`')
                   .caption.grey--text {{ info.usersTotal }}
           template(v-if='hasPermission(`manage:system`)')

server/controllers/upload.js

@@ -88,6 +88,7 @@ router.post('/u', multer({
   // Process upload file
   await WIKI.models.assets.upload({
     ...fileMeta,
+    mode: 'upload',
     folderId: folderId,
     assetPath,
     user: req.user

server/models/assets.js

@@ -95,8 +95,7 @@ module.exports = class Asset extends Model {
       kind: _.startsWith(opts.mimetype, 'image/') ? 'image' : 'binary',
       mime: opts.mimetype,
       fileSize: opts.size,
-      folderId: opts.folderId,
+      folderId: opts.folderId
-      authorId: opts.user.id
     }
 
     // Save asset data
@@ -105,6 +104,9 @@ module.exports = class Asset extends Model {
 
       if (asset) {
         // Patch existing asset
+        if (opts.mode === 'upload') {
+          assetRow.authorId = opts.user.id
+        }
         await WIKI.models.assets.query().patch(assetRow).findById(asset.id)
         await WIKI.models.knex('assetData').where({
           id: asset.id
@@ -113,6 +115,7 @@ module.exports = class Asset extends Model {
         })
       } else {
         // Create asset entry
+        assetRow.authorId = opts.user.id
         asset = await WIKI.models.assets.query().insert(assetRow)
         await WIKI.models.knex('assetData').insert({
           id: asset.id,
@@ -121,7 +124,11 @@ module.exports = class Asset extends Model {
       }
 
       // Move temp upload to cache
+      if (opts.mode === 'upload') {
         await fs.move(opts.path, path.join(process.cwd(), `data/cache/${fileHash}.dat`), { overwrite: true })
+      } else {
+        await fs.copy(opts.path, path.join(process.cwd(), `data/cache/${fileHash}.dat`), { overwrite: true })
+      }
 
       // Add to Storage
       if (!opts.skipStorage) {

server/modules/storage/disk/storage.js

@@ -9,6 +9,7 @@ const pipeline = Promise.promisify(stream.pipeline)
 const klaw = require('klaw')
 const pageHelper = require('../../../helpers/page.js')
 const moment = require('moment')
+const mime = require('mime-types').lookup
 
 /* global WIKI */
 
@@ -167,6 +168,7 @@ module.exports = {
     WIKI.logger.info(`(STORAGE/DISK) Importing all content from local disk folder to the DB...`)
 
     const rootUser = await WIKI.models.users.getRootUser()
+    let assetFolders = await WIKI.models.assetFolders.getAllPaths()
 
     await pipeline(
       klaw(this.config.path, {
@@ -178,12 +180,14 @@ module.exports = {
         objectMode: true,
         transform: async (file, enc, cb) => {
           const relPath = file.path.substr(this.config.path.length + 1)
-          if (relPath && relPath.length > 3) {
+          if (file.stats.size < 1) {
+            // Skip directories and zero-byte files
+            return cb()
+          } else if (relPath && relPath.length > 3) {
             WIKI.logger.info(`(STORAGE/DISK) Processing ${relPath}...`)
             const contentType = pageHelper.getContentType(relPath)
-            if (!contentType) {
+            if (contentType) {
-              return cb()
+              // -> Page
-            }
               const contentPath = pageHelper.getPagePath(relPath)
 
               let itemContents = ''
@@ -228,6 +232,51 @@ module.exports = {
                 WIKI.logger.warn(`(STORAGE/DISK) Failed to process ${relPath}`)
                 WIKI.logger.warn(err)
               }
+            } else {
+              // -> Asset
+
+              // -> Find existing folder
+              const filePathInfo = path.parse(file.path)
+              const folderPath = path.dirname(relPath).replace(/\\/g, '/')
+              let folderId = _.toInteger(_.findKey(assetFolders, fld => { return fld === folderPath })) || null
+
+              // -> Create missing folder structure
+              if (!folderId && folderPath !== '.') {
+                const folderParts = folderPath.split('/')
+                let currentFolderPath = []
+                let currentFolderParentId = null
+                for (const folderPart of folderParts) {
+                  currentFolderPath.push(folderPart)
+                  const existingFolderId = _.findKey(assetFolders, fld => { return fld === currentFolderPath.join('/') })
+                  if (!existingFolderId) {
+                    const newFolderObj = await WIKI.models.assetFolders.query().insert({
+                      slug: folderPart,
+                      name: folderPart,
+                      parentId: currentFolderParentId
+                    })
+                    _.set(assetFolders, newFolderObj.id, currentFolderPath.join('/'))
+                    currentFolderParentId = newFolderObj.id
+                  } else {
+                    currentFolderParentId = _.toInteger(existingFolderId)
+                  }
+                }
+                folderId = currentFolderParentId
+              }
+
+              // -> Import asset
+              await WIKI.models.assets.upload({
+                mode: 'import',
+                originalname: filePathInfo.base,
+                ext: filePathInfo.ext,
+                mimetype: mime(filePathInfo.base) || 'application/octet-stream',
+                size: file.stats.size,
+                folderId: folderId,
+                path: file.path,
+                assetPath: relPath,
+                user: rootUser,
+                skipStorage: true
+              })
+            }
           }
           cb()
         }