feat: seeds (#25)

Author: Stefen Alper

packages/smooth-backend-wordpress/package.json

@@ -26,8 +26,10 @@
   },
   "dependencies": {
     "axios": "^0.18.0",
+    "mime-types": "^2.1.24",
     "mkdirp": "^0.5.1",
     "ncp": "^2.0.0",
+    "p-limit": "^2.2.0",
     "qs": "^6.7.0"
   }
 }

packages/smooth-backend-wordpress/src/acf/api.js

@@ -1,5 +1,9 @@
+import fs from 'fs'
 import axios from 'axios'
 import qs from 'qs'
+import path from 'path'
+import mime from 'mime-types'
+import pLimit from 'p-limit'
 
 function getParams({ lang, type, query }) {
   const params = {}
@@ -13,7 +17,19 @@ function paramsSerializer(params) {
   return qs.stringify(params, { arrayFormat: 'indices' })
 }
 
-export function createClient({ baseUrl, defaultLanguage }) {
+export const onCreateResponse = ({ id, acf }) => ({ id, ...acf })
+
+export function createClient({
+  baseUrl,
+  concurrency,
+  password,
+  user,
+  defaultLanguage,
+}) {
+  const limit = pLimit(concurrency)
+  const token = Buffer.from(`${user}:${password}`).toString('base64')
+  const headers = { Authorization: `Basic ${token}` }
+
   return {
     async getContents({ type, slug, lang, query }) {
       type = type.toLowerCase()
@@ -40,10 +56,12 @@ export function createClient({ baseUrl, defaultLanguage }) {
         data,
       }
     },
+
     async getContent(request) {
       const { data } = await this.getContents(request)
       return data[0] || null
     },
+
     async getContentPreview({ lang, id }) {
       const params = getParams({ lang: lang || defaultLanguage })
       const { data } = await axios.get(
@@ -52,5 +70,52 @@ export function createClient({ baseUrl, defaultLanguage }) {
       )
       return data
     },
+
+    async create({ type, data }) {
+      const url = `${baseUrl}/wp-json/wp/v2/${type}`
+
+      if (type === 'media') {
+        const mediaPath = data
+        const filename = path.basename(mediaPath)
+        const { data: content } = await axios.post(url, fs.readFileSync(data), {
+          headers: {
+            'cache-control': 'no-cache',
+            'Content-Disposition': `attachment; filename=${filename}`,
+            'Content-Type': mime.contentType(filename),
+            ...headers,
+          },
+        })
+        return onCreateResponse(content)
+      }
+
+      const { data: content } = await axios.post(url, data, { headers })
+      return onCreateResponse(content)
+    },
+
+    async createMany({ type, data }) {
+      return Promise.all(
+        data.map(d => limit(() => this.create({ type, data: d }))),
+      )
+    },
+
+    async truncate({ type }) {
+      const { data: results } = await axios.get(
+        `${baseUrl}/wp-json/wp/v2/${type}`,
+        {
+          params: { per_page: 100 },
+        },
+      )
+
+      return Promise.all(
+        results.map(async ({ id }) =>
+          limit(() =>
+            axios.delete(`${baseUrl}/wp-json/wp/v2/${type}/${id}`, {
+              headers,
+              params: { force: 1 },
+            }),
+          ),
+        ),
+      )
+    },
   }
 }

packages/smooth-backend-wordpress/src/acf/index.js

@@ -35,3 +35,13 @@ export async function getContents({ request }, options) {
   const apiClient = createClient(options)
   return apiClient.getContents(request)
 }
+
+export async function createMany({ request }, options) {
+  const apiClient = createClient(options)
+  return apiClient.createMany(request)
+}
+
+export async function truncate({ request }, options) {
+  const apiClient = createClient(options)
+  return apiClient.truncate(request)
+}

packages/smooth-backend-wordpress/src/smooth-node.js

@@ -15,6 +15,11 @@ export function resolveOptions(options) {
     options.homeUrl = process.env.WP_HOME
   }
 
+  options.user = options.user || process.env.WP_USER || 'admin'
+  options.password = options.password || process.env.WP_PASSWORD || 'admin'
+  options.concurrency =
+    Number(options.concurrency || process.env.SEED_CONCURRENCY) || 8
+
   return options
 }
 
@@ -37,3 +42,20 @@ export async function getContent(params, options) {
 export async function getContents(params, options) {
   return acf.getContents(params, options)
 }
+
+export async function createMany(params, options) {
+  return acf.createMany(params, options)
+}
+
+export async function truncate(params, options) {
+  return acf.truncate(params, options)
+}
+
+export function createBackendPayload({ data, name, type }) {
+  if (type === 'media') {
+    return data
+  }
+
+  const { slug, ...fields } = data
+  return { status: 'publish', slug, title: name, fields }
+}

packages/smooth/seeds.js

@@ -0,0 +1 @@
+module.exports = require('./lib/seeds')

packages/smooth/src/api/index.js

@@ -15,5 +15,17 @@ export function createAPIClient({ config, lang }) {
         ...options,
       })
     },
+    async createMany(options) {
+      return callApi(config.plugins, 'createMany', {
+        ...defaultOptions,
+        ...options,
+      })
+    },
+    async truncate(options) {
+      return callApi(config.plugins, 'truncate', {
+        ...defaultOptions,
+        ...options,
+      })
+    },
   }
 }

packages/smooth/src/cli/index.js

@@ -12,6 +12,9 @@ import {
   buildSchemaDefinition,
   buildWebpack,
 } from '../build'
+import babelRequire from '../babel/require'
+import { createAPIClient } from '../api'
+import processSeeds from '../seeds/process'
 
 const logger = createLogger()
 
@@ -99,6 +102,18 @@ async function startCommand() {
   console.log('Started 🚀')
 }
 
+async function seedCommand() {
+  const config = await getConfig({ dev: false })
+  const api = createAPIClient({ config })
+  await processSeeds({
+    api,
+    config,
+    createSeeds: babelRequire(config.seedsPath).default,
+    logger,
+  })
+  console.log('Seeded 🍙')
+}
+
 function runCmd(command) {
   command().catch(error => {
     // eslint-disable-next-line no-console
@@ -122,5 +137,9 @@ program
     'Starts the application in production mode.\nThe application should be compiled with `smooth build` first.',
   )
   .action(() => runCmd(startCommand))
+program
+  .command('seed')
+  .description('Seeds the application using ./seeds directory.')
+  .action(() => runCmd(seedCommand))
 
 program.parse(process.argv)

packages/smooth/src/config/index.js

@@ -38,6 +38,7 @@ export async function getConfig({ dev }) {
     blocksPath: path.join(localCwd, 'src/blocks'),
     pagesPath: path.join(localCwd, 'src/pages'),
     schemasPath: path.join(localCwd, 'src/schemas'),
+    seedsPath: path.join(localCwd, process.env.SEEDS_PATH || './seeds'),
     server: {
       port: process.env.PORT || 3000,
     },

packages/smooth/src/plugin/nodeHooks/createBackendPayload.js

@@ -0,0 +1,4 @@
+import { applyHook } from '../node'
+
+export const createBackendPayload = config => ({ data, name, type }) =>
+  applyHook(config, 'createBackendPayload', { data, name, type }, 'data')

packages/smooth/src/plugin/nodeHooks/index.js

@@ -1,7 +1,8 @@
+export * from './createBackendPayload'
 export * from './onBuild'
 export * from './onCreateApolloServerConfig'
-export * from './onCreateServer'
 export * from './onCreateBabelConfig'
+export * from './onCreateServer'
 export * from './onCreateWebpackConfig'
 export * from './onRenderBody'
 export * from './wrapRootElement'