feat: add graph:init command

README.md

@@ -162,6 +162,7 @@ Manage netlify functions
 | [`graph:config:write`](/docs/commands/graph.md#graphconfigwrite) | Write a .graphqlrc.json file to the current directory for use with local tooling (e.g. the graphql extension for vscode)  |
 | [`graph:edit`](/docs/commands/graph.md#graphedit) | Launch the browser to edit your local graph functions from Netlify  |
 | [`graph:handler`](/docs/commands/graph.md#graphhandler) | Generate a handler for a Graph operation given its name. See `graph:operations` for a list of operations.  |
+| [`graph:init`](/docs/commands/graph.md#graphinit) | Initialize all the resources for Netlify Graph  |
 | [`graph:library`](/docs/commands/graph.md#graphlibrary) | Generate the Graph function library  |
 | [`graph:operations`](/docs/commands/graph.md#graphoperations) | List all of the locally available operations  |
 | [`graph:pull`](/docs/commands/graph.md#graphpull) | Pull down your local Netlify Graph schema, and process pending Graph edit events  |

docs/README.md

@@ -118,6 +118,7 @@ Manage netlify functions
 | [`graph:config:write`](/docs/commands/graph.md#graphconfigwrite) | Write a .graphqlrc.json file to the current directory for use with local tooling (e.g. the graphql extension for vscode)  |
 | [`graph:edit`](/docs/commands/graph.md#graphedit) | Launch the browser to edit your local graph functions from Netlify  |
 | [`graph:handler`](/docs/commands/graph.md#graphhandler) | Generate a handler for a Graph operation given its name. See `graph:operations` for a list of operations.  |
+| [`graph:init`](/docs/commands/graph.md#graphinit) | Initialize all the resources for Netlify Graph  |
 | [`graph:library`](/docs/commands/graph.md#graphlibrary) | Generate the Graph function library  |
 | [`graph:operations`](/docs/commands/graph.md#graphoperations) | List all of the locally available operations  |
 | [`graph:pull`](/docs/commands/graph.md#graphpull) | Pull down your local Netlify Graph schema, and process pending Graph edit events  |

docs/commands/graph.md

@@ -25,6 +25,7 @@ netlify graph
 | [`graph:config:write`](/docs/commands/graph.md#graphconfigwrite) | Write a .graphqlrc.json file to the current directory for use with local tooling (e.g. the graphql extension for vscode)  |
 | [`graph:edit`](/docs/commands/graph.md#graphedit) | Launch the browser to edit your local graph functions from Netlify  |
 | [`graph:handler`](/docs/commands/graph.md#graphhandler) | Generate a handler for a Graph operation given its name. See `graph:operations` for a list of operations.  |
+| [`graph:init`](/docs/commands/graph.md#graphinit) | Initialize all the resources for Netlify Graph  |
 | [`graph:library`](/docs/commands/graph.md#graphlibrary) | Generate the Graph function library  |
 | [`graph:operations`](/docs/commands/graph.md#graphoperations) | List all of the locally available operations  |
 | [`graph:pull`](/docs/commands/graph.md#graphpull) | Pull down your local Netlify Graph schema, and process pending Graph edit events  |
@@ -92,6 +93,23 @@ netlify graph:handler
 - `httpProxy` (*string*) - Proxy server address to route requests through.
 - `httpProxyCertificateFilename` (*string*) - Certificate file to use when connecting using a proxy server
 
+---
+## `graph:init`
+
+Initialize all the resources for Netlify Graph
+
+**Usage**
+
+```bash
+netlify graph:init
+```
+
+**Flags**
+
+- `debug` (*boolean*) - Print debugging information
+- `httpProxy` (*string*) - Proxy server address to route requests through.
+- `httpProxyCertificateFilename` (*string*) - Certificate file to use when connecting using a proxy server
+
 ---
 ## `graph:library`
 

docs/commands/index.md

@@ -99,6 +99,7 @@ Manage netlify functions
 | [`graph:config:write`](/docs/commands/graph.md#graphconfigwrite) | Write a .graphqlrc.json file to the current directory for use with local tooling (e.g. the graphql extension for vscode)  |
 | [`graph:edit`](/docs/commands/graph.md#graphedit) | Launch the browser to edit your local graph functions from Netlify  |
 | [`graph:handler`](/docs/commands/graph.md#graphhandler) | Generate a handler for a Graph operation given its name. See `graph:operations` for a list of operations.  |
+| [`graph:init`](/docs/commands/graph.md#graphinit) | Initialize all the resources for Netlify Graph  |
 | [`graph:library`](/docs/commands/graph.md#graphlibrary) | Generate the Graph function library  |
 | [`graph:operations`](/docs/commands/graph.md#graphoperations) | List all of the locally available operations  |
 | [`graph:pull`](/docs/commands/graph.md#graphpull) | Pull down your local Netlify Graph schema, and process pending Graph edit events  |

src/commands/graph/graph-init.js

@@ -0,0 +1,116 @@
+// @ts-check
+const { Buffer } = require('buffer')
+
+const { v4: uuidv4 } = require('uuid')
+
+const { OneGraphCliClient, ensureCLISession } = require('../../lib/one-graph/cli-client')
+const { chalk, error, log } = require('../../utils')
+
+const { ensureAppForSite, executeCreateApiTokenMutation } = OneGraphCliClient
+
+/**
+ * Creates the `netlify graph:init` command
+ * @param {import('commander').OptionValues} options
+ * @param {import('../base-command').BaseCommand} command
+ * @returns
+ */
+const graphInit = async (options, command) => {
+  const { api, site, state } = command.netlify
+  const siteId = site.id
+
+  if (!siteId) {
+    error(
+      `No siteId defined, unable to start Netlify Graph. To enable, run ${chalk.yellow(
+        'netlify init',
+      )} or ${chalk.yellow('netlify link')}`,
+    )
+  }
+
+  const netlifyToken = await command.authenticate()
+  // @ts-ignore: we need better types for our api object
+  const siteData = await api.getSite({ siteId })
+
+  await ensureAppForSite(netlifyToken, siteId)
+
+  await ensureCLISession({
+    metadata: {},
+    netlifyToken,
+    site,
+    state,
+  })
+
+  let envChanged = false
+
+  // Get current environment variables set in the UI
+  const {
+    build_settings: { env = {} },
+  } = siteData
+
+  const newEnv = {
+    ...env,
+  }
+
+  if (!env.NETLIFY_GRAPH_WEBHOOK_SECRET) {
+    envChanged = true
+    const secret = Buffer.from(uuidv4()).toString('base64')
+    newEnv.NETLIFY_GRAPH_WEBHOOK_SECRET = secret
+  }
+
+  if (!env.NETLIFY_GRAPH_PERSIST_QUERY_TOKEN) {
+    const variables = {
+      nfToken: netlifyToken,
+      input: {
+        appId: siteId,
+        scopes: ['MODIFY_SCHEMA', 'PERSIST_QUERY'],
+      },
+    }
+
+    const result = await executeCreateApiTokenMutation(variables, {
+      siteId,
+    })
+
+    const token =
+      result.data &&
+      result.data.oneGraph &&
+      result.data.oneGraph.createApiToken &&
+      result.data.oneGraph.createApiToken.accessToken &&
+      result.data.oneGraph.createApiToken.accessToken.token
+
+    if (token) {
+      envChanged = true
+      newEnv.NETLIFY_GRAPH_PERSIST_QUERY_TOKEN = token
+    } else {
+      error(`Unable to create Netlify Graph persist query token: ${JSON.stringify(result.errors, null, 2)}`)
+    }
+  }
+
+  if (envChanged) {
+    // Apply environment variable updates
+    // @ts-ignore
+    await api.updateSite({
+      siteId,
+      body: {
+        build_settings: {
+          env: newEnv,
+        },
+      },
+    })
+
+    log(`Finished updating Graph-related environment variables for site ${siteData.name}`)
+  }
+}
+
+/**
+ * Creates the `netlify graph:init` command
+ * @param {import('../base-command').BaseCommand} program
+ * @returns
+ */
+const createGraphInitCommand = (program) =>
+  program
+    .command('graph:init')
+    .description('Initialize all the resources for Netlify Graph')
+    .action(async (options, command) => {
+      await graphInit(options, command)
+    })
+
+module.exports = { createGraphInitCommand }

src/commands/graph/graph.js

@@ -2,6 +2,7 @@
 const { createGraphConfigWriteCommand } = require('./graph-config-write')
 const { createGraphEditCommand } = require('./graph-edit')
 const { createGraphHandlerCommand } = require('./graph-handler')
+const { createGraphInitCommand } = require('./graph-init')
 const { createGraphLibraryCommand } = require('./graph-library')
 const { createGraphOperationCommand } = require('./graph-operations')
 const { createGraphPullCommand } = require('./graph-pull')
@@ -24,6 +25,7 @@ const createGraphCommand = (program) => {
   createGraphConfigWriteCommand(program)
   createGraphEditCommand(program)
   createGraphHandlerCommand(program)
+  createGraphInitCommand(program)
   createGraphLibraryCommand(program)
   createGraphOperationCommand(program)
   createGraphPullCommand(program)

tests/integration/snapshots/220.command.graph.test.js.md

@@ -27,6 +27,7 @@ Generated by [AVA](https://avajs.dev).
       $ graph:config:write                      Write a .graphqlrc.json file to the current directory for use with local tooling (e.g. the graphql extension for vscode)␊
       $ graph:edit                              Launch the browser to edit your local graph functions from Netlify␊
       $ graph:handler                           Generate a handler for a Graph operation given its name. See \`graph:operations\` for a list of operations.␊
+      $ graph:init                              Initialize all the resources for Netlify Graph␊
       $ graph:library                           Generate the Graph function library␊
       $ graph:operations                        List all of the locally available operations␊
       $ graph:pull                              Pull down your local Netlify Graph schema, and process pending Graph edit events␊
@@ -55,6 +56,7 @@ Generated by [AVA](https://avajs.dev).
       $ graph:config:write                      Write a .graphqlrc.json file to the current directory for use with local tooling (e.g. the graphql extension for vscode)␊
       $ graph:edit                              Launch the browser to edit your local graph functions from Netlify␊
       $ graph:handler                           Generate a handler for a Graph operation given its name. See \`graph:operations\` for a list of operations.␊
+      $ graph:init                              Initialize all the resources for Netlify Graph␊
       $ graph:library                           Generate the Graph function library␊
       $ graph:operations                        List all of the locally available operations␊
       $ graph:pull                              Pull down your local Netlify Graph schema, and process pending Graph edit events␊