Add loading the list of co-located media

Author: techxplorer

src/lib/blog-post.js

@@ -3,7 +3,7 @@
  */
 
 import { lstatSync } from "node:fs";
-import { readFile } from "node:fs/promises";
+import { readFile, readdir } from "node:fs/promises";
 import path from "node:path";
 
 import slug from "slug";
@@ -28,6 +28,12 @@ class BlogPost {
    */
   postContent = {};
 
+  /**
+   * A list of media co-located with the post.
+   * @type {Array}
+   */
+  postMedia = [];
+
   /**
    * The content of the new post including front matter and content.
    * @type {object}
@@ -95,6 +101,20 @@ class BlogPost {
     } else {
       this.postContent = TomlMatter.parse( postFileContent );
     }
+
+    this.postMedia = await readdir(
+      this.postPath
+    );
+
+    this.postMedia = this.postMedia.filter( mediaFile => {
+      if ( path.extname( mediaFile ) === ".jpeg" ) {
+        return true;
+      } else if ( path.extname( mediaFile ) === ".jpg" ) {
+        return true;
+      } else {
+        return false;
+      }
+    } );
   }
 
   /**
@@ -127,6 +147,7 @@ class BlogPost {
     this.newPostContent.data = frontMatter;
 
   }
+
 }
 
 export default BlogPost;

tests/lib/blog-post.test.js

@@ -28,6 +28,9 @@ const testPassNewFrontMatter = [
   "taxonomies",
   "extra"
 ];
+const testMediaCountOne = 1;
+const testMediaCountTwo = 1;
+const testMediaCountThree = 4;
 
 describe( "BlogPost", () => {
 
@@ -189,6 +192,34 @@ describe( "BlogPost", () => {
 
     } );
 
+    it( "should load the correct number of media files", async() => {
+
+      let blogPost = new BlogPost( testPassPathOne );
+      await blogPost.loadPost();
+
+      assert.ok(
+        blogPost.postMedia.length,
+        testMediaCountOne
+      );
+
+      blogPost = new BlogPost( testPassPathTwo );
+      await blogPost.loadPost();
+
+      assert.ok(
+        blogPost.postMedia.length,
+        testMediaCountTwo
+      );
+
+      blogPost = new BlogPost( testPassPathThree );
+      await blogPost.loadPost();
+
+      assert.ok(
+        blogPost.postMedia.length,
+        testMediaCountThree
+      );
+
+    } );
+
   } );
 
   describe( "convertPost", async() => {