feat: show number of images imported

hexojs · Jul 25, 2020 · ae9c3da · ae9c3da
1 parent c3b2ec0
commit ae9c3da
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/lib/migrator.js b/lib/migrator.js
@@ -22,6 +22,7 @@ module.exports = async function(args) {
   let untitledPostCounter = 0;
   let errNum = 0;
   let skipNum = 0;
+  let imgNum = 0;
   let input, feed;
   const rExcerpt = /<!--+\s*more\s*--+>/i;
   const postExcerpt = '\n<!-- more -->\n';
@@ -106,6 +107,7 @@ module.exports = async function(args) {
           }
 
           images[image_url] = imagePath;
+          imgNum++;
         } catch (err) {
           log.e(err);
         }
@@ -213,6 +215,7 @@ module.exports = async function(args) {
       log.w('%d posts did not have titles and were prefixed with "Untitled Post".', untitledPostCounter);
     }
     if (postsNum) log.i('%d posts migrated.', postsNum);
+    if (imgNum) log.i('%d images migrated.', imgNum);
     if (errNum) log.error('%d posts failed to migrate.', posts.length);
     if (skipNum) log.i('%d posts skipped.', skipNum);
   }

diff --git a/test/index.js b/test/index.js
@@ -183,6 +183,11 @@ describe('migrator', function() {
 
       log.i.calledWith('Image found: %s', imagePath).should.eql(true);
 
+      const { items } = await parseFeed(xml);
+      const imgNum = items.filter(({type}) => type === 'attachment').length;
+      const { lastCall } = log.i;
+      lastCall.calledWith('%d images migrated.', imgNum).should.eql(true);
+
       const image = await readFile(join(hexo.source_dir, imagePath), { encoding: 'binary' });
       const header = Buffer.from(image, 'binary').toString('hex').substring(0, 14);