From bb4a463c367057219757d9a63acbca8145483a9d Mon Sep 17 00:00:00 2001
From: antekpapieski <maciejpapiewski9@gmail.com>
Date: Thu, 28 May 2020 22:29:46 +0200
Subject: [PATCH] add ability to keep the messages in a .js file or in a module

---
 src/gatsby-node.js | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/gatsby-node.js b/src/gatsby-node.js
index 4f7af36..c21a632 100644
--- a/src/gatsby-node.js
+++ b/src/gatsby-node.js
@@ -1,4 +1,5 @@
 const webpack = require("webpack")
+const fs = require('fs')
 
 function flattenMessages(nestedMessages, prefix = "") {
   return Object.keys(nestedMessages).reduce((messages, key) => {
@@ -53,8 +54,17 @@ exports.onCreatePage = async ({ page, actions }, pluginOptions) => {
 
   const getMessages = (path, language) => {
     try {
-      // TODO load yaml here
-      const messages = require(`${path}/${language}.json`)
+
+      let messages;
+
+
+      if(fs.existsSync(`${path}/${language}.json`))
+        messages = require(`${path}/${language}.json`)
+      else if(fs.existsSync(`${path}/${language}/index.js`))
+        messages = require(`${path}/${language}/index.js`)
+      else
+        messages = require(`${path}/${language}.js`)
+      
 
       return flattenMessages(messages)
     } catch (error) {