Upload images (#5)

Author: Bhacaz

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## Unreleased
 
+* Upload images ([#5](https://github.com/Bhacaz/docs-as-code-confluence/pull/5))
 * Specified node version and updated dependencies ([#4](https://github.com/Bhacaz/docs-as-code-confluence/pull/4))
 
 ## v1 (2022-11-14)

confluence.js

@@ -70,6 +70,69 @@ class SyncConfluence {
       );
     });
   }
+
+  getAttachments(pageId) {
+    return new Promise((resolve) => {
+      this.confluenceApi.getAttachments(
+          this.spaceId,
+          pageId,
+          (err, data) => {
+            if (err) {
+              console.error(err);
+            } else {
+              if (data.results[0]) {
+                resolve(data.results);
+              } else {
+                resolve(undefined);
+              }
+            }
+          }
+      );
+    });
+  }
+
+  updateAttachment(pageId, attachmentId, source) {
+    return new Promise((resolve) => {
+      this.confluenceApi.updateAttachmentData(
+          this.spaceId,
+          pageId,
+          attachmentId,
+          source,
+          (err, data) => {
+            if (err) {
+              console.error(err);
+            } else {
+              if (data) {
+                resolve(data);
+              } else {
+                resolve(undefined);
+              }
+            }
+          }
+      );
+    });
+  }
+
+  uploadAttachment(pageId, source) {
+    return new Promise((resolve) => {
+      this.confluenceApi.createAttachment(
+          this.spaceId,
+          pageId,
+          source,
+          (err, data) => {
+            if (err) {
+              console.error(err);
+            } else {
+              if (data.results[0]) {
+                resolve(data.results[0]);
+              } else {
+                resolve(undefined);
+              }
+            }
+          }
+      );
+    });
+  }
 }
 
 module.exports = SyncConfluence;

index.js

@@ -3,6 +3,8 @@ const filesStructure = require("./files");
 const SyncConflence = require("./confluence");
 const markdownToHtml = require("./markdownToHtml");
 const core = require("@actions/core");
+const parser = require("node-html-parser")
+const path = require('path')
 
 const root = "./" + core.getInput("folder", { required: true }) + "/";
 const spaceId = core.getInput("space-id", { required: true });
@@ -41,6 +43,32 @@ async function findOrCreatePage(pageTitle, parentPageId) {
   return pageId;
 }
 
+async function uploadAttachment(attachmentSource, pageId) {
+  attachmentSource = root + attachmentSource;
+  const existingAttachments = await syncConfluence.getAttachments(pageId)
+  if (existingAttachments) {
+    for (let attachment of existingAttachments) {
+      if (attachment.title === path.basename(attachmentSource)) {
+        return await syncConfluence.updateAttachment(pageId, attachment.id, attachmentSource);
+      }
+    }
+  }
+  return await syncConfluence.uploadAttachment(pageId, attachmentSource);
+}
+
+async function handleAttachments(contentPageId, data) {
+  const html = parser.parse(data);
+  const images = html.querySelectorAll("img")
+  for (var image of images) {
+    const attachmentSource = image.getAttribute("src");
+    // TODO handle remote images
+    if (attachmentSource.includes("http")) { continue; }
+    var attachment = await uploadAttachment(attachmentSource.replace("..", "."), contentPageId);
+    image.replaceWith(parser.parse('<ac:image><ri:attachment ri:filename=' + attachment.title +' /></ac:image>'));
+  }
+  return html.toString()
+}
+
 async function main() {
   for (const f of filesStructure(root)) {
     let path = f.join("/");
@@ -52,8 +80,9 @@ async function main() {
           pageTitle,
           currentParentPageId
         );
-        markdownToHtml(root + path, (err, data) => {
-          syncConfluence.putContent(contentPageId, pageTitle, data);
+        markdownToHtml(root + path,  (err, data) => {
+          let htmlContent = handleAttachments(contentPageId, data);
+          syncConfluence.putContent(contentPageId, pageTitle, htmlContent);
         });
       } else {
         currentParentPageId = await findOrCreatePage(