Fix for WordPress Tag and Image

Author: thandaanda

gatsby-wordpress-theme-libre/fix-source-nodes.js

@@ -1,4 +1,6 @@
 const { createRemoteFileNode } = require(`gatsby-source-filesystem`);
+const fetch = require("node-fetch");
+const FormData = require("form-data");
 const currentTimeTag = new Date().getTime();
 const createTag = {
   wordpress_id: currentTimeTag,
@@ -122,28 +124,52 @@ const createImage = {
   path: "/wordpress-balsa/",
 };
 
-module.exports = async function sourceNodes({
-  actions,
-  getNodesByType,
-  getNodes,
-  store,
-  cache,
-  createNodeId,
-  createContentDigest,
-  getCache,
-  reporter,
-}) {
-  const { createNode } = actions;
-
-  const wordPressTag = getNodesByType("wordpress__TAG");
-
-  const wordPressTagExists =
-    wordPressTag &&
-    wordPressTag.length > 0 &&
-    wordPressTag[0].internal &&
-    wordPressTag[0].internal.owner === "gatsby-source-wordpress";
-
-  if (!wordPressTagExists) {
+module.exports = async function sourceNodes(
+  {
+    actions,
+    getNodesByType,
+    getNodes,
+    store,
+    cache,
+    createNodeId,
+    createContentDigest,
+    getCache,
+    reporter,
+  },
+  { wordpressConfig }
+) {
+  const { createNode, deleteNode } = actions;
+
+  let url;
+
+  let accessToken;
+
+  let remoteWordPressTagCount;
+  let remoteWordPressMediaCount;
+
+  if (wordpressConfig.hostingWPCOM) {
+    url = `https://public-api.wordpress.com/wp/v2/sites/${wordpressConfig.baseUrl}`;
+    accessToken = await getWPCOMAccessToken(wordpressConfig.auth);
+  } else {
+    url = `https://${wordpressConfig.baseUrl}/wp-json/wp/v2`;
+  }
+
+  remoteWordPressMediaCount = await getWordPressEntityCount(
+    `${url}/media`,
+    accessToken
+  );
+  remoteWordPressTagCount = await getWordPressEntityCount(
+    `${url}/tags`,
+    accessToken
+  );
+
+  const wordPressTags = getNodesByType("wordpress__TAG");
+
+  const originalTagSourceExists = wordPressTags.filter(
+    (wordPressTag) => wordPressTag.internal.owner === "gatsby-source-wordpress"
+  );
+
+  if (remoteWordPressTagCount == 0 || !originalTagSourceExists) {
     let node = {
       ...createTag,
       children: [],
@@ -156,15 +182,14 @@ module.exports = async function sourceNodes({
     createNode(node);
   }
 
-  const wordpressMedia = getNodesByType("wordpress__wp_media");
+  const wordpressMedias = getNodesByType("wordpress__wp_media");
 
-  const mediaExists =
-    wordpressMedia &&
-    wordpressMedia.length > 0 &&
-    wordpressMedia[0].internal &&
-    wordpressMedia[0].internal.owner === "gatsby-source-wordpress";
+  const originalMediaSourceExists = wordpressMedias.filter(
+    (wordpressMedia) =>
+      wordpressMedia.internal.owner === "gatsby-source-wordpress"
+  );
 
-  if (!mediaExists) {
+  if (remoteWordPressMediaCount == 0 || !originalMediaSourceExists) {
     const encodedSourceUrl = encodeURI(createImage.source_url);
     const mediaDataCacheKey = `wordpress-media-placeholder`;
 
@@ -212,3 +237,46 @@ module.exports = async function sourceNodes({
     }
   }
 };
+
+async function getWPCOMAccessToken(_auth) {
+  let result;
+  const oauthUrl = `https://public-api.wordpress.com/oauth2/token`;
+
+  try {
+    const form = new FormData();
+    form.append("client_id", _auth.wpcom_app_clientId);
+    form.append("client_secret", _auth.wpcom_app_clientSecret);
+    form.append("username", _auth.wpcom_user);
+    form.append("password", _auth.wpcom_pass);
+    form.append("grant_type", `password`);
+
+    const response = await fetch(oauthUrl, { method: "POST", body: form });
+    result = await response.json();
+    result = result.access_token;
+  } catch (e) {
+    console.log(e);
+  }
+
+  return result;
+}
+
+async function getWordPressEntityCount(url, accessToken) {
+  const getOptions = () => {
+    let o = {
+      method: `get`,
+    };
+
+    if (accessToken) {
+      o.headers = {
+        Authorization: `Bearer ${accessToken}`,
+      };
+    }
+
+    return o;
+  };
+  const options = getOptions();
+  const response = await fetch(url, options);
+  const data = await response.json();
+
+  return data.length;
+}

gatsby-wordpress-theme-libre/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@draftbox-co/gatsby-wordpress-theme-libre",
-  "version": "1.0.38",
+  "version": "1.0.39",
   "license": "MIT",
   "resolutions": {
     "sharp": "0.23.4"

yarn.lock

@@ -1738,6 +1738,11 @@
   resolved "https://registry.yarnpkg.com/@types/debug/-/debug-0.0.30.tgz#dc1e40f7af3b9c815013a7860e6252f6352a84df"
   integrity sha512-orGL5LXERPYsLov6CWs3Fh6203+dXzJkR7OnddIr2514Hsecwc8xRpzCapshBbKFImCsvS/mk6+FWiN5LyZJAQ==
 
+"@types/eslint-visitor-keys@^1.0.0":
+  version "1.0.0"
+  resolved "https://registry.yarnpkg.com/@types/eslint-visitor-keys/-/eslint-visitor-keys-1.0.0.tgz#1ee30d79544ca84d68d4b3cdb0af4f205663dd2d"
+  integrity sha512-OCutwjDZ4aFS6PB1UZ988C4YgwlBHJd6wCeQqaLdmadZ/7e+w79+hbMUFC1QXDNCmdyoRfAFdm0RypzwR+Qpag==
+
 "@types/events@*":
   version "3.0.0"
   resolved "https://registry.yarnpkg.com/@types/events/-/events-3.0.0.tgz#2862f3f58a9a7f7c3e78d79f130dd4d71c25c2a7"
@@ -1913,11 +1918,6 @@
   version "2.30.0"
   resolved "https://registry.yarnpkg.com/@typescript-eslint/parser/-/parser-2.30.0.tgz#7681c305a6f4341ae2579f5e3a75846c29eee9ce"
   integrity sha512-9kDOxzp0K85UnpmPJqUzdWaCNorYYgk1yZmf4IKzpeTlSAclnFsrLjfwD9mQExctLoLoGAUXq1co+fbr+3HeFw==
-  dependencies:
-    "@types/eslint-visitor-keys" "^1.0.0"
-    "@typescript-eslint/experimental-utils" "2.30.0"
-    "@typescript-eslint/typescript-estree" "2.30.0"
-    eslint-visitor-keys "^1.1.0"
 
 "@typescript-eslint/typescript-estree@2.30.0":
   version "2.30.0"