Merge pull request #177 from iterative/use-websites-server

gatsby-config.js

@@ -1,6 +1,9 @@
 require('dotenv').config()
 const path = require('path')
 
+const apiMiddleware = require('@dvcorg/websites-server/src/middleware/api')
+const redirectsMiddleware = require('@dvcorg/websites-server/src/middleware/redirects')
+
 const siteMetadata = {
   title: 'MLEM - Machine Learning Engineering Management',
   description: 'Open-source tool to simplify ML model deployment',
@@ -115,5 +118,9 @@ module.exports = {
       : true
   },
   plugins,
-  siteMetadata
+  siteMetadata,
+  developMiddleware: app => {
+    app.use(redirectsMiddleware)
+    app.use('/api', apiMiddleware)
+  }
 }

package.json

@@ -6,8 +6,8 @@
     "build": "gatsby build",
     "develop": "gatsby develop",
     "dev": "gatsby develop",
-    "start": "node ./src/server/index.js",
-    "heroku-postbuild": "./scripts/heroku-deploy.sh",
+    "start": "node ./src/server.js",
+    "heroku-postbuild": "heroku-deploy",
     "serve": "gatsby serve",
     "clean": "gatsby clean",
     "format": "prettier --write \"**/*.{js,json,css,md}\"",
@@ -22,6 +22,7 @@
   },
   "dependencies": {
     "@dvcorg/gatsby-theme-iterative": "0.1.16",
+    "@dvcorg/websites-server": "^0.0.10",
     "@octokit/graphql": "5.0.1",
     "@sentry/gatsby": "^7.13.0",
     "@svgr/webpack": "6.3.1",

src/server.js

@@ -0,0 +1,12 @@
+// eslint-disable-next-line @typescript-eslint/no-var-requires
+const server = require('@dvcorg/websites-server')
+
+const app = server.app
+
+// we can also extend to add further custom routes
+app.get('/api/status', (req, res) => {
+  res.send('ok')
+})
+
+// run the server
+server.run()