@knocklabs/cli 0.2.4 → 0.3.0

package/dist/commands/layout/pull.js

@@ -109,7 +109,9 @@ class EmailLayoutPull extends _basecommand.default {
             });
             return this.apiV1.getEmailLayout(props);
         });
-        await _emaillayout.writeEmailLayoutDirFromData(dirContext, resp.data);
+        await _emaillayout.writeEmailLayoutDirFromData(dirContext, resp.data, {
+            withSchema: true
+        });
         const action = dirContext.exists ? "updated" : "created";
         const scope = (0, _command.formatCommandScope)(flags);
         this.log(`‣ Successfully ${action} \`${dirContext.key}\` at ${dirContext.abspath} using ${scope}`);
@@ -127,7 +129,9 @@ class EmailLayoutPull extends _basecommand.default {
         if (!input) return;
         _ux.spinner.start(`‣ Loading`);
         const emailLayouts = await this.listAllEmailLayouts();
-        await _emaillayout.writeEmailLayoutIndexDir(targetDirCtx, emailLayouts);
+        await _emaillayout.writeEmailLayoutIndexDir(targetDirCtx, emailLayouts, {
+            withSchema: true
+        });
         _ux.spinner.stop();
         const action = targetDirCtx.exists ? "updated" : "created";
         const scope = (0, _command.formatCommandScope)(flags);

package/dist/commands/layout/push.js

@@ -121,7 +121,9 @@ class EmailLayoutPush extends _basecommand.default {
                 // Update the layout directory with the successfully pushed layout
                 // payload from the server.
                 // eslint-disable-next-line no-await-in-loop
-                await _emaillayout.writeEmailLayoutDirFromData(layout, resp.data.email_layout);
+                await _emaillayout.writeEmailLayoutDirFromData(layout, resp.data.email_layout, {
+                    withSchema: true
+                });
                 continue;
             }
             const error = new _error.SourceError((0, _request.formatErrorRespMessage)(resp), _emaillayout.emailLayoutJsonPath(layout), "ApiError");

package/dist/commands/message-type/pull.js

@@ -112,7 +112,9 @@ class MessageTypePull extends _basecommand.default {
             });
             return this.apiV1.getMessageType(props);
         });
-        await _messagetype.writeMessageTypeDirFromData(dirContext, resp.data);
+        await _messagetype.writeMessageTypeDirFromData(dirContext, resp.data, {
+            withSchema: true
+        });
         const action = dirContext.exists ? "updated" : "created";
         const scope = (0, _command.formatCommandScope)(flags);
         this.log(`‣ Successfully ${action} \`${dirContext.key}\` at ${dirContext.abspath} using ${scope}`);
@@ -161,7 +163,9 @@ class MessageTypePull extends _basecommand.default {
         if (!input) return;
         _ux.spinner.start(`‣ Loading`);
         const messageTypes = await this.listAllMessageTypes();
-        await _messagetype.writeMessageTypesIndexDir(targetDirCtx, messageTypes);
+        await _messagetype.writeMessageTypesIndexDir(targetDirCtx, messageTypes, {
+            withSchema: true
+        });
         _ux.spinner.stop();
         const action = targetDirCtx.exists ? "updated" : "created";
         const scope = (0, _command.formatCommandScope)(flags);

package/dist/commands/message-type/push.js

@@ -121,7 +121,9 @@ class MessageTypePush extends _basecommand.default {
                 // Update the message type directory with the successfully pushed message
                 // type payload from the server.
                 // eslint-disable-next-line no-await-in-loop
-                await _messagetype.writeMessageTypeDirFromData(messageType, resp.data.message_type);
+                await _messagetype.writeMessageTypeDirFromData(messageType, resp.data.message_type, {
+                    withSchema: true
+                });
                 continue;
             }
             const error = new _error.SourceError((0, _request.formatErrorRespMessage)(resp), _messagetype.messageTypeJsonPath(messageType), "ApiError");

package/dist/commands/partial/pull.js

@@ -109,7 +109,9 @@ class PartialPull extends _basecommand.default {
             });
             return this.apiV1.getPartial(props);
         });
-        await _partial.writePartialDirFromData(dirContext, resp.data);
+        await _partial.writePartialDirFromData(dirContext, resp.data, {
+            withSchema: true
+        });
         const action = dirContext.exists ? "updated" : "created";
         const scope = (0, _command.formatCommandScope)(flags);
         this.log(`‣ Successfully ${action} \`${dirContext.key}\` at ${dirContext.abspath} using ${scope}`);
@@ -127,7 +129,9 @@ class PartialPull extends _basecommand.default {
         if (!input) return;
         _ux.spinner.start(`‣ Loading`);
         const partials = await this.listAllPartials();
-        await _partial.writePartialsIndexDir(targetDirCtx, partials);
+        await _partial.writePartialsIndexDir(targetDirCtx, partials, {
+            withSchema: true
+        });
         _ux.spinner.stop();
         const action = targetDirCtx.exists ? "updated" : "created";
         const scope = (0, _command.formatCommandScope)(flags);

package/dist/commands/partial/push.js

@@ -121,7 +121,9 @@ class PartialPush extends _basecommand.default {
                 // Update the partial directory with the successfully pushed partial
                 // payload from the server.
                 // eslint-disable-next-line no-await-in-loop
-                await _partial.writePartialDirFromData(partial, resp.data.partial);
+                await _partial.writePartialDirFromData(partial, resp.data.partial, {
+                    withSchema: true
+                });
                 continue;
             }
             const error = new _error.SourceError((0, _request.formatErrorRespMessage)(resp), _partial.partialJsonPath(partial), "ApiError");

package/dist/commands/translation/get.js

@@ -134,6 +134,7 @@ class TranslationGet extends _basecommand.default {
     }
 }
 _define_property(TranslationGet, "summary", "Display a single translation from an environment.");
+_define_property(TranslationGet, "verifyFeatureEnabled", "translations");
 _define_property(TranslationGet, "flags", {
     environment: _core.Flags.string({
         default: "development",

package/dist/commands/translation/list.js

@@ -137,6 +137,7 @@ class TranslationList extends _basecommand.default {
     }
 }
 _define_property(TranslationList, "summary", "Display all translations for an environment.");
+_define_property(TranslationList, "verifyFeatureEnabled", "translations");
 _define_property(TranslationList, "flags", {
     environment: _core.Flags.string({
         default: "development",

package/dist/commands/translation/pull.js

@@ -173,6 +173,7 @@ class TranslationPull extends _basecommand.default {
     }
 }
 _define_property(TranslationPull, "summary", "Pull one or more translations from an environment into a local file system.");
+_define_property(TranslationPull, "verifyFeatureEnabled", "translations");
 _define_property(TranslationPull, "flags", {
     environment: _core.Flags.string({
         default: "development",

package/dist/commands/translation/push.js

@@ -125,6 +125,7 @@ class TranslationPush extends _basecommand.default {
     }
 }
 _define_property(TranslationPush, "summary", "Push one or more translations from a local file system to Knock.");
+_define_property(TranslationPush, "verifyFeatureEnabled", "translations");
 _define_property(TranslationPush, "flags", {
     environment: _core.Flags.string({
         summary: "Pushing a translation is only allowed in the development environment",

package/dist/commands/translation/validate.js

@@ -120,6 +120,7 @@ class TranslationValidate extends _basecommand.default {
     }
 }
 _define_property(TranslationValidate, "summary", "Validate one or more translations from a local file system.");
+_define_property(TranslationValidate, "verifyFeatureEnabled", "translations");
 _define_property(TranslationValidate, "flags", {
     environment: _core.Flags.string({
         summary: "Validating a translation is only done in the development environment",

package/dist/commands/workflow/new.js

@@ -12,7 +12,7 @@ const _nodepath = /*#__PURE__*/ _interop_require_wildcard(require("node:path"));
 const _core = require("@oclif/core");
 const _fsextra = /*#__PURE__*/ _interop_require_wildcard(require("fs-extra"));
 const _basecommand = /*#__PURE__*/ _interop_require_default(require("../../lib/base-command"));
-const _objectisomorphic = require("../../lib/helpers/object.isomorphic");
+const _const = require("../../lib/helpers/const");
 const _ux = require("../../lib/helpers/ux");
 const _workflow = /*#__PURE__*/ _interop_require_wildcard(require("../../lib/marshal/workflow"));
 function _define_property(obj, key, value) {
@@ -119,17 +119,12 @@ class WorkflowNew extends _basecommand.default {
         const isExistingWorkflow = await this.checkExistingWorkflow();
         if (isExistingWorkflow) {
             this.log("");
-            this.warn(`Workflow \`${args.workflowKey}\` already exists in \`development\` environment`);
+            this.warn(`Workflow \`${args.workflowKey}\` already exists in \`${flags.environment}\` environment`);
         }
     }
     async checkExistingWorkflow() {
-        const props = (0, _objectisomorphic.merge)(this.props, {
-            flags: {
-                environment: "development"
-            }
-        });
         try {
-            const resp = await this.apiV1.getWorkflow(props);
+            const resp = await this.apiV1.getWorkflow(this.props);
             return resp.status === 200;
         } catch  {}
     }
@@ -140,7 +135,11 @@ _define_property(WorkflowNew, "flags", {
             "step"
         ]
     }),
-    force: _core.Flags.boolean()
+    force: _core.Flags.boolean(),
+    environment: _core.Flags.string({
+        hidden: true,
+        default: _const.KnockEnv.Development
+    })
 });
 _define_property(WorkflowNew, "args", {
     workflowKey: _core.Args.string({

package/dist/commands/workflow/pull.js

@@ -113,7 +113,9 @@ class WorkflowPull extends _basecommand.default {
             return this.apiV1.getWorkflow(props);
         });
         // 3. Write the fetched workflow to create or update the workflow directory.
-        await _workflow.writeWorkflowDirFromData(dirContext, resp.data);
+        await _workflow.writeWorkflowDirFromData(dirContext, resp.data, {
+            withSchema: true
+        });
         const action = dirContext.exists ? "updated" : "created";
         const scope = (0, _command.formatCommandScope)(flags);
         this.log(`‣ Successfully ${action} \`${dirContext.key}\` at ${dirContext.abspath} using ${scope}`);
@@ -162,7 +164,9 @@ class WorkflowPull extends _basecommand.default {
         // Fetch all workflows then write them to the local file system.
         _ux.spinner.start(`‣ Loading`);
         const workflows = await this.listAllWorkflows();
-        await _workflow.writeWorkflowsIndexDir(targetDirCtx, workflows);
+        await _workflow.writeWorkflowsIndexDir(targetDirCtx, workflows, {
+            withSchema: true
+        });
         _ux.spinner.stop();
         const action = targetDirCtx.exists ? "updated" : "created";
         const scope = (0, _command.formatCommandScope)(flags);

package/dist/commands/workflow/push.js

@@ -121,7 +121,9 @@ class WorkflowPush extends _basecommand.default {
                 // Update the workflow directory with the successfully pushed workflow
                 // payload from the server.
                 // eslint-disable-next-line no-await-in-loop
-                await _workflow.writeWorkflowDirFromData(workflow, resp.data.workflow);
+                await _workflow.writeWorkflowDirFromData(workflow, resp.data.workflow, {
+                    withSchema: true
+                });
                 continue;
             }
             const error = new _error.SourceError((0, _request.formatErrorRespMessage)(resp), _workflow.workflowJsonPath(workflow), "ApiError");
@@ -138,11 +140,8 @@ class WorkflowPush extends _basecommand.default {
 _define_property(WorkflowPush, "summary", "Push one or more workflows from a local file system to Knock.");
 _define_property(WorkflowPush, "flags", {
     environment: _core.Flags.string({
-        summary: "Pushing a workflow is only allowed in the development environment",
-        default: _const.KnockEnv.Development,
-        options: [
-            _const.KnockEnv.Development
-        ]
+        summary: "The environment to push the workflow to. Defaults to development.",
+        default: _const.KnockEnv.Development
     }),
     branch: _flag.branch,
     all: _core.Flags.boolean({

package/dist/commands/workflow/validate.js

@@ -125,11 +125,8 @@ class WorkflowValidate extends _basecommand.default {
 _define_property(WorkflowValidate, "summary", "Validate one or more workflows from a local file system.");
 _define_property(WorkflowValidate, "flags", {
     environment: _core.Flags.string({
-        summary: "Validating a workflow is only done in the development environment",
-        default: _const.KnockEnv.Development,
-        options: [
-            _const.KnockEnv.Development
-        ]
+        summary: "The environment to validate the workflow in. Defaults to development.",
+        default: _const.KnockEnv.Development
     }),
     branch: _flag.branch,
     all: _core.Flags.boolean({

package/dist/lib/base-command.js

@@ -11,6 +11,7 @@ Object.defineProperty(exports, "default", {
 const _core = require("@oclif/core");
 const _apiv1 = /*#__PURE__*/ _interop_require_default(require("./api-v1"));
 const _auth = /*#__PURE__*/ _interop_require_default(require("./auth"));
+const _accountfeatures = require("./helpers/account-features");
 const _runcontext = /*#__PURE__*/ _interop_require_wildcard(require("./run-context"));
 const _urls = require("./urls");
 const _userconfig = require("./user-config");
@@ -106,7 +107,12 @@ class BaseCommand extends _core.Command {
         }
         // 6. Instantiate a knock api client.
         this.apiV1 = new _apiv1.default(this.sessionContext, this.config);
-        // 7. Load the run context of the invoked command.
+        // 7. Verify that required features are enabled for the account.
+        const ctor = this.ctor;
+        if (ctor.verifyFeatureEnabled) {
+            await this.verifyFeatureEnabled(ctor.verifyFeatureEnabled);
+        }
+        // 8. Load the run context of the invoked command.
         this.runContext = await _runcontext.load(this.id);
     }
     buildSessionContext() {
@@ -138,6 +144,15 @@ class BaseCommand extends _core.Command {
             this.error("No token found. Refusing to run command.");
         }
     }
+    async verifyFeatureEnabled(feature) {
+        if (feature === "translations") {
+            const featureCheck = await (0, _accountfeatures.checkTranslationsFeature)(this.apiV1);
+            if (!featureCheck.enabled) {
+                this.log(featureCheck.message);
+                this.exit(0);
+            }
+        }
+    }
     async refreshAccessTokenForSession() {
         // Maybe refresh the access token?
         try {
@@ -166,6 +181,7 @@ class BaseCommand extends _core.Command {
         super(...args), _define_property(this, "props", void 0), _define_property(this, "apiV1", void 0), _define_property(this, "runContext", void 0), _define_property(this, "sessionContext", void 0), _define_property(this, "configStore", void 0), _define_property(this, "requiresAuth", true);
     }
 }
+_define_property(BaseCommand, "verifyFeatureEnabled", void 0);
 // Base flags are inherited by any command that extends BaseCommand.
 _define_property(BaseCommand, "baseFlags", {
     // Evaluated in the following precedence:

package/dist/lib/helpers/account-features.js

@@ -0,0 +1,53 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", {
+    value: true
+});
+function _export(target, all) {
+    for(var name in all)Object.defineProperty(target, name, {
+        enumerable: true,
+        get: Object.getOwnPropertyDescriptor(all, name).get
+    });
+}
+_export(exports, {
+    get checkAccountFeature () {
+        return checkAccountFeature;
+    },
+    get checkTranslationsFeature () {
+        return checkTranslationsFeature;
+    }
+});
+const _request = require("./request");
+async function checkAccountFeature(apiV1, featureName, options = {}) {
+    try {
+        const whoamiResp = await apiV1.whoami();
+        if (!(0, _request.isSuccessResp)(whoamiResp)) {
+            return {
+                enabled: false,
+                message: `Unable to retrieve account information: ${whoamiResp.status} ${whoamiResp.statusText}`
+            };
+        }
+        const accountFeatures = whoamiResp.data.account_features;
+        const isFeatureEnabled = (accountFeatures === null || accountFeatures === void 0 ? void 0 : accountFeatures[featureName]) === true;
+        if (!isFeatureEnabled) {
+            const defaultMessage = `The ${featureName} feature is not enabled for your account. Please contact support to enable this feature.`;
+            return {
+                enabled: false,
+                message: options.errorMessage || defaultMessage
+            };
+        }
+        return {
+            enabled: true
+        };
+    } catch (error) {
+        return {
+            enabled: false,
+            message: error instanceof Error ? error.message : "Unknown error occurred while checking translations feature"
+        };
+    }
+}
+async function checkTranslationsFeature(apiV1, options = {}) {
+    return checkAccountFeature(apiV1, "translations_allowed", {
+        errorMessage: options.errorMessage || "Translations are not enabled for your account. Please contact support to enable the translations feature.",
+        ...options
+    });
+}

package/dist/lib/marshal/email-layout/processor.isomorphic.js

@@ -38,8 +38,9 @@ const compileExtractionSettings = (emailLayout)=>{
     }
     return map;
 };
-const buildEmailLayoutDirBundle = (remoteEmailLayout, localEmailLayout = {})=>{
+const buildEmailLayoutDirBundle = (remoteEmailLayout, localEmailLayout, $schema)=>{
     const bundle = {};
+    localEmailLayout = localEmailLayout || {};
     const mutRemoteEmailLayout = (0, _lodash.cloneDeep)(remoteEmailLayout);
     // A map of extraction settings of every field in the email layout
     const compiledExtractionSettings = compileExtractionSettings(mutRemoteEmailLayout);
@@ -75,5 +76,5 @@ const buildEmailLayoutDirBundle = (remoteEmailLayout, localEmailLayout = {})=>{
     // the layout JSON realtive path + the file content.
     return (0, _lodash.set)(bundle, [
         LAYOUT_JSON
-    ], (0, _helpersisomorphic.prepareResourceJson)(mutRemoteEmailLayout));
+    ], (0, _helpersisomorphic.prepareResourceJson)(mutRemoteEmailLayout, $schema));
 };

package/dist/lib/marshal/email-layout/writer.js

@@ -73,13 +73,15 @@ function _interop_require_wildcard(obj, nodeInterop) {
     }
     return newObj;
 }
-const writeEmailLayoutDirFromData = async (emailLayoutDirCtx, remoteEmailLayout)=>{
+const EMAIL_LAYOUT_SCHEMA = "https://schemas.knock.app/cli/email-layout.json";
+const writeEmailLayoutDirFromData = async (emailLayoutDirCtx, remoteEmailLayout, options)=>{
+    const { withSchema = false } = options || {};
     // If the layout directory exists on the file system (i.e. previously
     // pulled before), then read the layout file to use as a reference.
     const [localEmailLayout] = emailLayoutDirCtx.exists ? await (0, _reader.readEmailLayoutDir)(emailLayoutDirCtx, {
         withExtractedFiles: true
     }) : [];
-    const bundle = (0, _processorisomorphic.buildEmailLayoutDirBundle)(remoteEmailLayout, localEmailLayout);
+    const bundle = (0, _processorisomorphic.buildEmailLayoutDirBundle)(remoteEmailLayout, localEmailLayout, withSchema ? EMAIL_LAYOUT_SCHEMA : undefined);
     const backupDirPath = _nodepath.default.resolve(_const.sandboxDir, (0, _lodash.uniqueId)("backup"));
     try {
         // We store a backup in case there's an error.
@@ -109,7 +111,7 @@ const writeEmailLayoutDirFromData = async (emailLayoutDirCtx, remoteEmailLayout)
         await _fsextra.remove(backupDirPath);
     }
 };
-const writeEmailLayoutIndexDir = async (indexDirCtx, remoteEmailLayouts)=>{
+const writeEmailLayoutIndexDir = async (indexDirCtx, remoteEmailLayouts, options)=>{
     const backupDirPath = _nodepath.default.resolve(_const.sandboxDir, (0, _lodash.uniqueId)("backup"));
     try {
         if (indexDirCtx.exists) {
@@ -124,7 +126,7 @@ const writeEmailLayoutIndexDir = async (indexDirCtx, remoteEmailLayouts)=>{
                 abspath: emailLayoutDirPath,
                 exists: indexDirCtx.exists ? await (0, _helpers.isEmailLayoutDir)(emailLayoutDirPath) : false
             };
-            return writeEmailLayoutDirFromData(emailLayoutDirCtx, remoteEmailLayout);
+            return writeEmailLayoutDirFromData(emailLayoutDirCtx, remoteEmailLayout, options);
         });
         await Promise.all(writeEmailLayoutDirPromises);
     } catch (error) {

package/dist/lib/marshal/guide/processor.isomorphic.js

@@ -45,8 +45,9 @@ const compileExtractionSettings = (guide)=>{
     }
     return map;
 };
-const buildGuideDirBundle = (remoteGuide, localGuide = {})=>{
+const buildGuideDirBundle = (remoteGuide, localGuide, $schema)=>{
     const bundle = {};
+    localGuide = localGuide || {};
     const mutRemoteGuide = (0, _lodash.cloneDeep)(remoteGuide);
     // A map of extraction settings of every field in the guide.
     const compiledExtractionSettings = compileExtractionSettings(mutRemoteGuide);
@@ -81,5 +82,5 @@ const buildGuideDirBundle = (remoteGuide, localGuide = {})=>{
     // the guide JSON relative path + the file content.
     return (0, _lodash.set)(bundle, [
         GUIDE_JSON
-    ], (0, _helpersisomorphic.prepareResourceJson)(mutRemoteGuide));
+    ], (0, _helpersisomorphic.prepareResourceJson)(mutRemoteGuide, $schema));
 };

package/dist/lib/marshal/guide/writer.js

@@ -68,13 +68,15 @@ function _interop_require_wildcard(obj, nodeInterop) {
     }
     return newObj;
 }
-const writeGuideDirFromData = async (guideDirCtx, remoteGuide)=>{
+const GUIDE_SCHEMA = "https://schemas.knock.app/cli/guide.json";
+const writeGuideDirFromData = async (guideDirCtx, remoteGuide, options)=>{
+    const { withSchema = false } = options || {};
     // If the guide directory exists on the file system (i.e. previously
     // pulled before), then read the guide file to use as a reference.
     const [localGuide] = guideDirCtx.exists ? await (0, _reader.readGuideDir)(guideDirCtx, {
         withExtractedFiles: true
     }) : [];
-    const bundle = (0, _processorisomorphic.buildGuideDirBundle)(remoteGuide, localGuide);
+    const bundle = (0, _processorisomorphic.buildGuideDirBundle)(remoteGuide, localGuide, withSchema ? GUIDE_SCHEMA : undefined);
     return writeGuideDirFromBundle(guideDirCtx, bundle);
 };
 /*
@@ -136,7 +138,7 @@ const writeGuideDirFromData = async (guideDirCtx, remoteGuide)=>{
     });
     await Promise.all(promises);
 };
-const writeGuidesIndexDir = async (indexDirCtx, remoteGuides)=>{
+const writeGuidesIndexDir = async (indexDirCtx, remoteGuides, options)=>{
     const backupDirPath = _nodepath.resolve(_const.sandboxDir, (0, _lodash.uniqueId)("backup"));
     try {
         // If the index directory already exists, back it up in the temp sandbox
@@ -154,7 +156,7 @@ const writeGuidesIndexDir = async (indexDirCtx, remoteGuides)=>{
                 abspath: guideDirPath,
                 exists: indexDirCtx.exists ? await (0, _helpers.isGuideDir)(guideDirPath) : false
             };
-            return writeGuideDirFromData(guideDirCtx, guide);
+            return writeGuideDirFromData(guideDirCtx, guide, options);
         });
         await Promise.all(writeGuideDirPromises);
     } catch (error) {

package/dist/lib/marshal/message-type/processor.isomorphic.js

@@ -38,9 +38,10 @@ const compileExtractionSettings = (messageType)=>{
     }
     return map;
 };
-const buildMessageTypeDirBundle = (remoteMessageType, localMessageType = {})=>{
+const buildMessageTypeDirBundle = (remoteMessageType, localMessageType, $schema)=>{
     const bundle = {};
     const mutRemoteMessageType = (0, _lodash.cloneDeep)(remoteMessageType);
+    localMessageType = localMessageType || {};
     // A map of extraction settings of every field in the message type
     const compiledExtractionSettings = compileExtractionSettings(mutRemoteMessageType);
     // Iterate through each extractable field, determine whether we need to
@@ -74,5 +75,5 @@ const buildMessageTypeDirBundle = (remoteMessageType, localMessageType = {})=>{
     // the message type JSON relative path + the file content.
     return (0, _lodash.set)(bundle, [
         MESSAGE_TYPE_JSON
-    ], (0, _helpersisomorphic.prepareResourceJson)(mutRemoteMessageType));
+    ], (0, _helpersisomorphic.prepareResourceJson)(mutRemoteMessageType, $schema));
 };

package/dist/lib/marshal/message-type/writer.js

@@ -68,13 +68,15 @@ function _interop_require_wildcard(obj, nodeInterop) {
     }
     return newObj;
 }
-const writeMessageTypeDirFromData = async (messageTypeDirCtx, remoteMessageType)=>{
+const MESSAGE_TYPE_SCHEMA = "https://schemas.knock.app/cli/message-type.json";
+const writeMessageTypeDirFromData = async (messageTypeDirCtx, remoteMessageType, options)=>{
+    const { withSchema = false } = options || {};
     // If the message type directory exists on the file system (i.e. previously
     // pulled before), then read the message type file to use as a reference.
     const [localMessageType] = messageTypeDirCtx.exists ? await (0, _reader.readMessageTypeDir)(messageTypeDirCtx, {
         withExtractedFiles: true
     }) : [];
-    const bundle = (0, _processorisomorphic.buildMessageTypeDirBundle)(remoteMessageType, localMessageType);
+    const bundle = (0, _processorisomorphic.buildMessageTypeDirBundle)(remoteMessageType, localMessageType, withSchema ? MESSAGE_TYPE_SCHEMA : undefined);
     return writeMessageTypeDirFromBundle(messageTypeDirCtx, bundle);
 };
 /*
@@ -136,7 +138,7 @@ const writeMessageTypeDirFromData = async (messageTypeDirCtx, remoteMessageType)
     });
     await Promise.all(promises);
 };
-const writeMessageTypesIndexDir = async (indexDirCtx, remoteMessageTypes)=>{
+const writeMessageTypesIndexDir = async (indexDirCtx, remoteMessageTypes, options)=>{
     const backupDirPath = _nodepath.resolve(_const.sandboxDir, (0, _lodash.uniqueId)("backup"));
     try {
         // If the index directory already exists, back it up in the temp sandbox
@@ -154,7 +156,7 @@ const writeMessageTypesIndexDir = async (indexDirCtx, remoteMessageTypes)=>{
                 abspath: messageTypeDirPath,
                 exists: indexDirCtx.exists ? await (0, _helpers.isMessageTypeDir)(messageTypeDirPath) : false
             };
-            return writeMessageTypeDirFromData(messageTypeDirCtx, messageType);
+            return writeMessageTypeDirFromData(messageTypeDirCtx, messageType, options);
         });
         await Promise.all(promises);
     } catch (error) {

package/dist/lib/marshal/partial/processor.isomorphic.js

@@ -61,8 +61,9 @@ const compileExtractionSettings = (partial)=>{
     }
     return map;
 };
-const buildPartialDirBundle = (remotePartial, localPartial = {})=>{
+const buildPartialDirBundle = (remotePartial, localPartial, $schema)=>{
     const bundle = {};
+    localPartial = localPartial || {};
     const mutRemotePartial = (0, _lodash.cloneDeep)(remotePartial);
     // A map of extraction settings of every field in the partial
     const compiledExtractionSettings = compileExtractionSettings(mutRemotePartial);
@@ -97,5 +98,5 @@ const buildPartialDirBundle = (remotePartial, localPartial = {})=>{
     // the partial JSON relative path + the file content.
     return (0, _lodash.set)(bundle, [
         PARTIAL_JSON
-    ], (0, _helpersisomorphic.prepareResourceJson)(mutRemotePartial));
+    ], (0, _helpersisomorphic.prepareResourceJson)(mutRemotePartial, $schema));
 };

package/dist/lib/marshal/partial/writer.js

@@ -68,13 +68,15 @@ function _interop_require_wildcard(obj, nodeInterop) {
     }
     return newObj;
 }
-const writePartialDirFromData = async (partialDirCtx, remotePartial)=>{
+const PARTIAL_SCHEMA = "https://schemas.knock.app/cli/partial.json";
+const writePartialDirFromData = async (partialDirCtx, remotePartial, options)=>{
+    const { withSchema = false } = options || {};
     // If the partial directory exists on the file system (i.e. previously
     // pulled before), then read the partial file to use as a reference.
     const [localPartial] = partialDirCtx.exists ? await (0, _reader.readPartialDir)(partialDirCtx, {
         withExtractedFiles: true
     }) : [];
-    const bundle = (0, _processorisomorphic.buildPartialDirBundle)(remotePartial, localPartial);
+    const bundle = (0, _processorisomorphic.buildPartialDirBundle)(remotePartial, localPartial, withSchema ? PARTIAL_SCHEMA : undefined);
     return writePartialDirFromBundle(partialDirCtx, bundle);
 };
 /*
@@ -136,7 +138,7 @@ const writePartialDirFromData = async (partialDirCtx, remotePartial)=>{
     });
     await Promise.all(promises);
 };
-const writePartialsIndexDir = async (indexDirCtx, remotePartials)=>{
+const writePartialsIndexDir = async (indexDirCtx, remotePartials, options)=>{
     const backupDirPath = _nodepath.resolve(_const.sandboxDir, (0, _lodash.uniqueId)("backup"));
     try {
         // If the index directory already exists, back it up in the temp sandbox
@@ -154,7 +156,7 @@ const writePartialsIndexDir = async (indexDirCtx, remotePartials)=>{
                 abspath: partialDirPath,
                 exists: indexDirCtx.exists ? await (0, _helpers.isPartialDir)(partialDirPath) : false
             };
-            return writePartialDirFromData(partialDirCtx, partial);
+            return writePartialDirFromData(partialDirCtx, partial, options);
         });
         await Promise.all(writePartialDirPromises);
     } catch (error) {

package/dist/lib/marshal/shared/helpers.isomorphic.js

@@ -15,7 +15,7 @@ const REMOVED_READONLY_FIELDS = [
     "sha",
     "updated_at"
 ];
-const prepareResourceJson = (resource)=>{
+const prepareResourceJson = (resource, $schema)=>{
     var _resource___annotation;
     const readonlyFields = ((_resource___annotation = resource.__annotation) === null || _resource___annotation === void 0 ? void 0 : _resource___annotation.readonly_fields) || [];
     const [readonly, remainder] = (0, _objectisomorphic.split)(resource, readonlyFields);
@@ -25,6 +25,12 @@ const prepareResourceJson = (resource)=>{
         ...remainder,
         __readonly: filteredReadonlyFields
     };
+    // Append the $schema property to the resource JSON if it is provided.
+    if ($schema) {
+        Object.assign(resourceJson, {
+            $schema
+        });
+    }
     // Strip out all schema annotations, so not to expose them to end users.
     return (0, _objectisomorphic.omitDeep)(resourceJson, [
         "__annotation"

package/dist/lib/marshal/workflow/processor.isomorphic.js

@@ -250,8 +250,9 @@ const recursivelyBuildWorkflowDirBundle = (bundle, steps, localWorkflowStepsByRe
         }
     }
 };
-const buildWorkflowDirBundle = (remoteWorkflow, localWorkflow = {})=>{
+const buildWorkflowDirBundle = (remoteWorkflow, localWorkflow, $schema)=>{
     const bundle = {};
+    localWorkflow = localWorkflow || {};
     const mutWorkflow = (0, _lodash.cloneDeep)(remoteWorkflow);
     const localWorkflowStepsByRef = keyLocalWorkflowStepsByRef(localWorkflow.steps);
     // Recursively traverse the workflow step tree, mutating it and the bundle
@@ -260,5 +261,5 @@ const buildWorkflowDirBundle = (remoteWorkflow, localWorkflow = {})=>{
     // Then, prepare the workflow data to be written into a workflow json file.
     return (0, _lodash.set)(bundle, [
         WORKFLOW_JSON
-    ], (0, _helpersisomorphic.prepareResourceJson)(mutWorkflow));
+    ], (0, _helpersisomorphic.prepareResourceJson)(mutWorkflow, $schema));
 };