[export] Add support for filtering on drafts and types

packages/@sanity/export/README.md

@@ -29,7 +29,16 @@ exportDataset({
   assets: false,
 
   // Exports documents only, without downloading or rewriting asset references
-  raw: true
+  // Default: `false`
+  raw: true,
+
+  // Whether or not to export drafts
+  // Default: `true`
+  drafts: true,
+
+  // Export only given document types (`_type`)
+  // Optional, default: all types
+  types: ['products', 'shops']
 })
 ```
 

packages/@sanity/export/src/export.js

@@ -9,7 +9,9 @@ const stringifyStream = require('./stringifyStream')
 const validateOptions = require('./validateOptions')
 const rejectOnApiError = require('./rejectOnApiError')
 const getDocumentsStream = require('./getDocumentsStream')
-const skipSystemDocuments = require('./skipSystemDocuments')
+const filterSystemDocuments = require('./filterSystemDocuments')
+const filterDocumentTypes = require('./filterDocumentTypes')
+const filterDrafts = require('./filterDrafts')
 
 const noop = () => null
 
@@ -57,8 +59,10 @@ function exportDataset(opts) {
       inputStream,
       split(JSON.parse),
       rejectOnApiError,
-      skipSystemDocuments,
+      filterSystemDocuments,
       assetStreamHandler,
+      filterDocumentTypes(options.types),
+      options.drafts ? miss.through.obj() : filterDrafts,
       stringifyStream
     )
 

packages/@sanity/export/src/filterDocumentTypes.js

@@ -0,0 +1,14 @@
+const miss = require('mississippi')
+
+module.exports = allowedTypes =>
+  allowedTypes
+    ? miss.through.obj((doc, enc, callback) => {
+        const type = doc && doc._type
+        if (allowedTypes.includes(type)) {
+          callback(null, doc)
+          return
+        }
+
+        callback()
+      })
+    : miss.through.obj()

packages/@sanity/export/src/filterDrafts.js

@@ -0,0 +1,11 @@
+const miss = require('mississippi')
+
+const isDraft = doc => doc && doc._id && doc._id.indexOf('drafts.') === 0
+
+module.exports = miss.through.obj((doc, enc, callback) => {
+  if (isDraft(doc)) {
+    return callback()
+  }
+
+  return callback(null, doc)
+})

packages/@sanity/export/src/validateOptions.js

@@ -1,9 +1,10 @@
 const defaults = require('lodash/defaults')
 
 const clientMethods = ['getUrl', 'config']
-const booleanFlags = ['assets', 'raw', 'compress']
+const booleanFlags = ['assets', 'raw', 'compress', 'drafts']
 const exportDefaults = {
   compress: true,
+  drafts: true,
   assets: true,
   raw: false
 }