@dronedeploy/rocos-js-sdk 3.0.10 → 3.0.12

package/cjs/helpers/kscript/nodes/statements/index.js

@@ -0,0 +1,14 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SwitchStatement = exports.ExpressionStatement = exports.ReturnStatement = exports.BlockStatement = void 0;
+var BlockStatement_1 = require("./BlockStatement");
+Object.defineProperty(exports, "BlockStatement", { enumerable: true, get: function () { return __importDefault(BlockStatement_1).default; } });
+var ReturnStatement_1 = require("./ReturnStatement");
+Object.defineProperty(exports, "ReturnStatement", { enumerable: true, get: function () { return __importDefault(ReturnStatement_1).default; } });
+var ExpressionStatement_1 = require("./ExpressionStatement");
+Object.defineProperty(exports, "ExpressionStatement", { enumerable: true, get: function () { return __importDefault(ExpressionStatement_1).default; } });
+var SwitchStatement_1 = require("./SwitchStatement");
+Object.defineProperty(exports, "SwitchStatement", { enumerable: true, get: function () { return __importDefault(SwitchStatement_1).default; } });

package/cjs/models/asset-storage/SyncIntegrations.d.ts

@@ -12,9 +12,9 @@ export interface AssetSyncDefinitionModel {
 export interface AssetSyncDefinitionOverrideModel extends AssetSyncDefinitionModel {
     _isOverridden: boolean;
 }
-export interface AssetSyncDefinitionsModel {
-    items: AssetSyncDefinitionModel[];
-    profileOverriddenItems?: AssetSyncDefinitionOverrideModel[];
+export interface AssetSyncDefinitionsModel<I extends AssetSyncDefinitionModel = AssetSyncDefinitionOverrideModel> {
+    items: I[];
+    profileOverriddenItems?: AssetSyncDefinitionModel[];
 }
 export interface AssetSyncIntegrationProviderModel {
     provider: string;

package/cjs/services/AssetStorageService.d.ts

@@ -1,4 +1,4 @@
-import { AssetItemModel, AssetItemWithObservationsModel, AssetSyncDefinitionsModel, AssetSyncIntegrationModel, AssetSyncIntegrationProviderModel, CreateAssetSyncIntegrationModel, IBaseService, IRocosSDKConfig, RocosError } from '../models';
+import { AssetItemModel, AssetItemWithObservationsModel, AssetSyncDefinitionModel, AssetSyncDefinitionsModel, AssetSyncIntegrationModel, AssetSyncIntegrationProviderModel, CreateAssetSyncIntegrationModel, IBaseService, IRocosSDKConfig, RocosError } from '../models';
 import { BaseServiceAbstract } from './BaseServiceAbstract';
 export declare class AssetStorageService extends BaseServiceAbstract implements IBaseService {
     constructor(config: IRocosSDKConfig);
@@ -92,7 +92,7 @@ export declare class AssetStorageService extends BaseServiceAbstract implements
      * @param definition
      * @param syncNow - Sync configured paths immediately
      */
-    setRobotSyncDefinitions(projectId: string, callsign: string, definition: AssetSyncDefinitionsModel, syncNow?: boolean): Promise<AssetSyncDefinitionsModel>;
+    setRobotSyncDefinitions(projectId: string, callsign: string, definition: AssetSyncDefinitionsModel<AssetSyncDefinitionModel>, syncNow?: boolean): Promise<AssetSyncDefinitionsModel>;
     /**
      * Get Robot Sync Definitions
      * @param projectId
@@ -105,7 +105,7 @@ export declare class AssetStorageService extends BaseServiceAbstract implements
      * @param profileId
      * @param definition
      */
-    setProfileSyncDefinitions(projectId: string, profileId: string, definition: AssetSyncDefinitionsModel): Promise<AssetSyncDefinitionsModel>;
+    setProfileSyncDefinitions(projectId: string, profileId: string, definition: Omit<AssetSyncDefinitionsModel<AssetSyncDefinitionModel>, 'profileOverriddenItems'>): Promise<AssetSyncDefinitionsModel>;
     /**
      * Get Profile Sync Definitions
      * @param projectId

package/cjs/services/BaseStreamService.js

@@ -44,7 +44,7 @@ class BaseStreamService {
                 this.status = msg === models_1.SubscriberStatusEnum.STOPPED || msg === models_1.SubscriberStatusEnum.ALIVE;
                 this.status$.next(msg);
             });
-            await streamRegister.addStream(stream);
+            streamRegister.addStream(stream);
             await this.initStream(stream);
         }
         return { stream, isNew };

package/cjs/services/WebRTCSignallingService.js

@@ -53,8 +53,7 @@ class WebRTCSignallingService {
                 this.status = msg === SubscriberStatusEnum_1.SubscriberStatusEnum.STOPPED || msg === SubscriberStatusEnum_1.SubscriberStatusEnum.ALIVE;
                 this.status$.next(msg);
             });
-            // don't wait for these promises, they can be init after startup
-            await streamRegister.addStream(stream);
+            streamRegister.addStream(stream);
             await this.initStream(stream);
         }
         return stream;

package/esm/api/StreamRegister.d.ts

@@ -6,8 +6,8 @@ export declare class StreamRegister {
     private constructor();
     static getInstance(): StreamRegister;
     static getIdentifier($identifier: string, scope?: string): string;
-    addStream(stream: IBaseStream): Promise<void>;
+    addStream(stream: IBaseStream): void;
     getStream(identifier: string): IBaseStream | undefined;
-    removeStream(stream: IBaseStream): Promise<void>;
-    removeAllStreams(): Promise<boolean>;
+    removeStream(stream: IBaseStream): void;
+    removeAllStreams(): boolean;
 }

package/esm/api/StreamRegister.js

@@ -14,10 +14,10 @@ export class StreamRegister {
     static getIdentifier($identifier, scope) {
         return `${$identifier}-${scope ?? ''}`;
     }
-    async addStream(stream) {
+    addStream(stream) {
         try {
             if (this.teleStreams.has(stream.identifier)) {
-                await this.removeStream(stream);
+                this.removeStream(stream);
             }
             this.teleStreams.set(stream.identifier, stream);
         }
@@ -34,9 +34,9 @@ export class StreamRegister {
     getStream(identifier) {
         return this.teleStreams.get(identifier);
     }
-    async removeStream(stream) {
+    removeStream(stream) {
         try {
-            await stream.stopStream();
+            stream.stopStream();
             this.teleStreams.delete(stream.identifier);
         }
         catch (e) {
@@ -49,10 +49,10 @@ export class StreamRegister {
             }
         }
     }
-    async removeAllStreams() {
+    removeAllStreams() {
         try {
             for (const [, stream] of this.teleStreams.entries()) {
-                await this.removeStream(stream);
+                this.removeStream(stream);
             }
         }
         catch (e) {

package/esm/api/streams/telemetry/TelemetryStreamAbstract.js

@@ -100,6 +100,14 @@ export class TelemetryStreamAbstract {
         const before = this.getSubscriptions();
         this.subscriptions.delete(params.uniqueId);
         const after = this.getSubscriptions();
+        // Do this before we send the telemetry request so that new subscriptions don't get this stream from the register while it is closing
+        if (!this.subscriptions.size) {
+            this.logger.info('No subscriptions remaining closing stream', params.uniqueId);
+            this.stopStream();
+            // self remove when no subscriptions are left
+            StreamRegister.getInstance().removeStream(this);
+            this.logger.info('Stream closed');
+        }
         // assign the current values from state after change
         this.callsignsLookup = new CallsignsLookup(Object.keys(after));
         this.sources = arrayUnique(Object.values(after).reduce((a, v) => a.concat(v), []));
@@ -113,13 +121,6 @@ export class TelemetryStreamAbstract {
                 this.logger.error(`Failed to unsubscribe: ${err}`);
             }
         }
-        if (!this.subscriptions.size) {
-            this.logger.info('No subscriptions remaining closing stream', params.uniqueId);
-            this.stopStream();
-            // self remove when no subscriptions a re left
-            void StreamRegister.getInstance().removeStream(this);
-            this.logger.info('Stream closed');
-        }
     }
     sendAcknowledgment(uid, status, noRetry) {
         return this.sendAcknowledgmentInternal(uid, status, noRetry);

package/esm/helpers/kscript/Context.d.ts

@@ -7,19 +7,20 @@ import type { NodeType } from './nodes';
  * If the set is null, all properties are allowed.
  */
 export type PrototypeWhitelist = Map<unknown, Set<string> | null>;
-export type Scope = Map<string, unknown> | Record<string, unknown>;
+export type Scope = Record<string, unknown> | Map<string, unknown>;
 export interface ContextOptions {
     nodeBlacklist?: Set<NodeType>;
     prototypeWhitelist?: PrototypeWhitelist;
     functionCallsAllowed?: boolean;
 }
 export default class Context {
-    readonly scope: Map<string, unknown>;
-    readonly isFunctionCallAllowed: boolean;
-    private readonly options;
-    constructor(scope?: Scope, options?: ContextOptions);
-    isNodeAllowed(nodeType: AnyNode['type']): boolean;
-    newChildContext(scope: Scope): Context;
+    private readonly parent;
+    private readonly scope;
+    private readonly config;
+    static create(scope?: Scope, options?: ContextOptions): Context;
+    private static buildMap;
+    private constructor();
+    newChild(scope?: Scope): Context;
     /** Returns true if the property is allowed to be accessed on the prototype.
      *
      * - If the prototype is not in the whitelist, the property is allowed only if it is an own property.
@@ -30,5 +31,10 @@ export default class Context {
      * @param ownProperty Whether the property is owned by the object it is being accessed on.
      */
     isPrototypeAllowed(prototype: unknown, property: string, ownProperty: boolean): boolean;
-    private static buildMap;
+    isNodeAllowed(nodeType: AnyNode['type']): boolean;
+    get isFunctionCallAllowed(): boolean;
+    private getDescriptor;
+    get(key: string): unknown;
+    set(key: string, value: unknown): void;
+    declare(key: string, value: unknown, constant: boolean): void;
 }

package/esm/helpers/kscript/Context.js

@@ -1,26 +1,27 @@
 export default class Context {
-    constructor(scope, options) {
+    static create(scope, options) {
+        return new Context(null, {
+            functionCallsAllowed: true,
+            nodeBlacklist: new Set(),
+            prototypeWhitelist: new Map(),
+            ...(options ?? {}),
+        }, scope);
+    }
+    static buildMap(obj) {
+        const entries = obj instanceof Map ? Array.from(obj.entries()) : Object.entries(obj);
+        return new Map(entries.map(([key, value]) => [key, { value, constant: false }]));
+    }
+    constructor(parent, config, scope) {
+        this.parent = parent;
         this.scope = scope ? Context.buildMap(scope) : new Map();
-        this.options = {
-            nodeBlacklist: options?.nodeBlacklist ?? new Set(),
-            prototypeWhitelist: options?.prototypeWhitelist ?? new Map(),
-            functionCallsAllowed: options?.functionCallsAllowed ?? true,
+        this.config = {
+            nodeBlacklist: config.nodeBlacklist ?? new Set(),
+            prototypeWhitelist: config.prototypeWhitelist ?? new Map(),
+            functionCallsAllowed: config.functionCallsAllowed ?? true,
         };
-        this.isFunctionCallAllowed = this.options.functionCallsAllowed;
-    }
-    isNodeAllowed(nodeType) {
-        return !this.options.nodeBlacklist.has(nodeType);
     }
-    newChildContext(scope) {
-        let newEntries;
-        if (scope instanceof Map) {
-            newEntries = [...scope.entries()];
-        }
-        else {
-            newEntries = Object.entries(scope);
-        }
-        const newScope = new Map([...this.scope.entries(), ...newEntries]);
-        return new Context(newScope, this.options);
+    newChild(scope) {
+        return new Context(this, this.config, scope);
     }
     /** Returns true if the property is allowed to be accessed on the prototype.
      *
@@ -32,7 +33,7 @@ export default class Context {
      * @param ownProperty Whether the property is owned by the object it is being accessed on.
      */
     isPrototypeAllowed(prototype, property, ownProperty) {
-        const whitelist = this.options.prototypeWhitelist.get(prototype);
+        const whitelist = this.config.prototypeWhitelist.get(prototype);
         // If the object owns the property, and it's not in the whitelist, it's allowed.
         // This prevents direct calls to non-whitelisted properties on allowed prototypes
         // e.g. Object.prototype.hasOwnProperty.call(obj, 'toString')
@@ -43,7 +44,43 @@ export default class Context {
             return true;
         return whitelist.has(property);
     }
-    static buildMap(obj) {
-        return obj instanceof Map ? obj : new Map(Object.entries(obj));
+    isNodeAllowed(nodeType) {
+        return !this.config.nodeBlacklist.has(nodeType);
+    }
+    get isFunctionCallAllowed() {
+        return this.config.functionCallsAllowed;
+    }
+    getDescriptor(key, searchParents = true) {
+        if (this.scope.has(key)) {
+            return this.scope.get(key);
+        }
+        if (searchParents && this.parent) {
+            return this.parent.getDescriptor(key);
+        }
+        return undefined;
+    }
+    get(key) {
+        const descriptor = this.getDescriptor(key);
+        if (descriptor) {
+            return descriptor.value;
+        }
+        return undefined;
+    }
+    set(key, value) {
+        const descriptor = this.getDescriptor(key);
+        if (!descriptor) {
+            throw new Error(`no variable with name ${key}`);
+        }
+        if (descriptor.constant) {
+            throw new Error(`cannot reassign to constant variable ${key}`);
+        }
+        descriptor.value = value;
+    }
+    declare(key, value, constant) {
+        const descriptor = this.getDescriptor(key, false);
+        if (descriptor) {
+            throw new Error(`variable ${key} already declared`);
+        }
+        this.scope.set(key, { value, constant });
     }
 }

package/esm/helpers/kscript/kscript.d.ts

@@ -1,6 +1,11 @@
 import { ContextOptions, PrototypeWhitelist, Scope } from './Context';
 import { Node } from './nodes';
-export declare function execute(code: string, scope?: Scope, options?: ContextOptions): unknown;
-export declare function compile(code: string, scope?: Scope, options?: ContextOptions): Node;
+export declare const compile: (code: string, scope?: Scope, options?: ContextOptions) => Node;
+/** Throws an error if the code uses an invalid node.
+ *
+ * This does not check that the code will run without errors, only that the nodes used are allowed.
+ **/
+export declare const validate: (code: string, options?: ContextOptions) => void;
+export declare const execute: (code: string, scope?: Scope, options?: ContextOptions) => unknown;
 export declare const SAFE_GLOBALS: Map<string, unknown>;
 export declare const SAFE_PROTOTYPES: PrototypeWhitelist;

package/esm/helpers/kscript/kscript.js

@@ -1,30 +1,39 @@
 import Context from './Context';
-import { parse } from 'acorn';
+import { parse as acorn } from 'acorn';
 import { construct } from './utils';
-export function execute(code, scope, options) {
-    return compile(code, scope, options).run();
-}
-export function compile(code, scope, options) {
-    let program;
+const parse = (code) => {
     try {
         // we add the brackets to ensure that object literals are parsed correctly
-        program = parse(`(${code})`, { ecmaVersion: 2020, sourceType: 'script' });
+        // e.g. `{ a: 1 }` is not a valid statement, but `({ a: 1 })` is
+        return acorn(`(${code})`, { ecmaVersion: 2020, sourceType: 'script' });
     }
     catch (error) {
         if (error instanceof SyntaxError) {
             // fallback to non-wrapped code for edge cases where the brackets cause issues
-            program = parse(`${code}`, { ecmaVersion: 2020, sourceType: 'script' });
-        }
-        else {
-            throw error;
+            // e.g. `if (a) 2; else 4;` will throw a syntax error if wrapped in brackets
+            return acorn(`${code}`, { ecmaVersion: 2020, sourceType: 'script' });
         }
+        throw error;
     }
-    const context = new Context(scope ?? SAFE_GLOBALS, {
+};
+export const compile = (code, scope, options) => {
+    const program = parse(code);
+    const context = Context.create(scope ?? SAFE_GLOBALS, {
         prototypeWhitelist: SAFE_PROTOTYPES,
         ...options,
     });
     return construct(program, context);
-}
+};
+/** Throws an error if the code uses an invalid node.
+ *
+ * This does not check that the code will run without errors, only that the nodes used are allowed.
+ **/
+export const validate = (code, options) => {
+    compile(code, undefined, options).validate(true);
+};
+export const execute = (code, scope, options) => {
+    return compile(code, scope, options).run();
+};
 export const SAFE_GLOBALS = new Map(Object.entries({
     console: {
         debug: console.debug,

package/esm/helpers/kscript/nodes/Identifier.d.ts

@@ -8,5 +8,6 @@ import Node from './Node';
  */
 export default class IdentifierNode extends Node<Identifier> {
     constructor(node: AnyNode, context: Context);
+    validate(): void;
     run(): unknown;
 }

package/esm/helpers/kscript/nodes/Identifier.js

@@ -8,7 +8,8 @@ export default class IdentifierNode extends Node {
     constructor(node, context) {
         super(node, context, 'Identifier');
     }
+    validate() { }
     run() {
-        return this.scope.get(this.node.name);
+        return this.context.get(this.node.name);
     }
 }

package/esm/helpers/kscript/nodes/Literal.d.ts

@@ -7,5 +7,6 @@ import Node from './Node';
  */
 export default class LiteralNode extends Node<Literal> {
     constructor(node: AnyNode, context: Context);
+    validate(): void;
     run(): unknown;
 }

package/esm/helpers/kscript/nodes/Literal.js

@@ -7,10 +7,13 @@ export default class LiteralNode extends Node {
     constructor(node, context) {
         super(node, context, 'Literal');
     }
-    run() {
+    validate() {
         if (this.node.bigint) {
             throw new Error('BigInts are not supported');
         }
+    }
+    run() {
+        this.validate();
         return this.node.value;
     }
 }

package/esm/helpers/kscript/nodes/Node.d.ts

@@ -3,7 +3,12 @@ import type Context from '../Context';
 export default abstract class Node<T extends AnyNode = AnyNode> {
     protected context: Context;
     protected node: T;
-    protected scope: Context['scope'];
     protected constructor(node: AnyNode, context: Context, expectedType: T['type'] | T['type'][]);
+    /** Runs the node. */
     abstract run(): unknown;
+    /** Validates the node. If the node is invalid, an error is thrown.
+     *
+     * if `validateChildren` is true, the children of the node will also be validated.
+     */
+    abstract validate(validateChildren: boolean): void;
 }

package/esm/helpers/kscript/nodes/Node.js

@@ -7,6 +7,5 @@ export default class Node {
             throw new Error(`Expected node type to be ${expectedTypes.join(' or ')}, but got ${node.type}`);
         }
         this.node = node;
-        this.scope = context.scope;
     }
 }

package/esm/helpers/kscript/nodes/Program.d.ts

@@ -7,5 +7,6 @@ import Node from './Node';
  */
 export default class ProgramNode extends Node<Program> {
     constructor(node: AnyNode, scope: Context);
+    validate(validateChildren?: boolean): void;
     run(): unknown;
 }

package/esm/helpers/kscript/nodes/Program.js

@@ -8,11 +8,20 @@ export default class ProgramNode extends Node {
     constructor(node, scope) {
         super(node, scope, 'Program');
     }
+    validate(validateChildren = false) {
+        if (!validateChildren) {
+            return;
+        }
+        for (const child of this.node.body) {
+            construct(child, this.context).validate(true);
+        }
+    }
     run() {
-        const children = this.node.body;
-        if (children.length !== 1) {
-            throw new Error('Only a single statement is supported in the sandbox');
+        this.validate();
+        let result;
+        for (const child of this.node.body) {
+            result = construct(child, this.context).run();
         }
-        return construct(children[0], this.context).run();
+        return result;
     }
 }

package/esm/helpers/kscript/nodes/TemplateLiteral.d.ts

@@ -19,5 +19,6 @@ import Node from './Node';
  */
 export default class TemplateLiteralNode extends Node<TemplateLiteral> {
     constructor(node: AnyNode, scope: Context);
+    validate(validateChildren?: boolean): void;
     run(): unknown;
 }

package/esm/helpers/kscript/nodes/TemplateLiteral.js

@@ -20,6 +20,14 @@ export default class TemplateLiteralNode extends Node {
     constructor(node, scope) {
         super(node, scope, 'TemplateLiteral');
     }
+    validate(validateChildren = false) {
+        if (!validateChildren) {
+            return;
+        }
+        for (const expression of this.node.expressions) {
+            construct(expression, this.context).validate(true);
+        }
+    }
     run() {
         return this.node.quasis.reduce((acc, quasi, i) => {
             const expression = this.node.expressions[i] ? construct(this.node.expressions[i], this.context).run() : '';

package/esm/helpers/kscript/nodes/VariableDeclaration.d.ts

@@ -0,0 +1,8 @@
+import { AnyNode, VariableDeclaration } from 'acorn';
+import Context from '../Context';
+import Node from './Node';
+export default class VariableDeclarationNode extends Node<VariableDeclaration> {
+    constructor(node: AnyNode, context: Context);
+    validate(validateChildren?: boolean): void;
+    run(): void;
+}

package/esm/helpers/kscript/nodes/VariableDeclaration.js

@@ -0,0 +1,37 @@
+import Node from './Node';
+import { construct } from '../utils';
+export default class VariableDeclarationNode extends Node {
+    constructor(node, context) {
+        super(node, context, 'VariableDeclaration');
+    }
+    validate(validateChildren = false) {
+        if (this.node.kind === 'var') {
+            throw new Error('var is not supported. use let or const instead');
+        }
+        for (const declaration of this.node.declarations) {
+            if (declaration.id.type !== 'Identifier') {
+                throw new Error('variable name must be an identifier');
+            }
+        }
+        if (!validateChildren) {
+            return;
+        }
+        for (const declaration of this.node.declarations) {
+            construct(declaration.id, this.context).validate(true);
+            if (declaration.init) {
+                construct(declaration.init, this.context).validate(true);
+            }
+        }
+    }
+    run() {
+        this.validate();
+        for (const declaration of this.node.declarations) {
+            const id = declaration.id.name;
+            let init;
+            if (declaration.init) {
+                init = construct(declaration.init, this.context).run();
+            }
+            this.context.declare(id, init, this.node.kind === 'const');
+        }
+    }
+}

package/esm/helpers/kscript/nodes/expressions/ArrayExpression.d.ts

@@ -3,5 +3,6 @@ import type Context from '../../Context';
 import Node from '../Node';
 export default class ArrayExpressionNode extends Node<ArrayExpression> {
     constructor(node: AnyNode, context: Context);
+    validate(validateChildren?: boolean): void;
     run(): unknown;
 }

package/esm/helpers/kscript/nodes/expressions/ArrayExpression.js

@@ -4,6 +4,17 @@ export default class ArrayExpressionNode extends Node {
     constructor(node, context) {
         super(node, context, 'ArrayExpression');
     }
+    validate(validateChildren = false) {
+        if (!validateChildren) {
+            return;
+        }
+        for (const element of this.node.elements) {
+            if (element === null) {
+                continue;
+            }
+            construct(element, this.context).validate(true);
+        }
+    }
     run() {
         const array = [];
         for (const element of this.node.elements) {

package/esm/helpers/kscript/nodes/expressions/ArrowFunctionExpression.d.ts

@@ -10,5 +10,6 @@ import Node from '../Node';
 export default class ArrowFunctionExpressionNode extends Node<ArrowFunctionExpression> {
     private readonly params;
     constructor(node: AnyNode, context: Context);
+    validate(validateChildren?: boolean): void;
     run(): unknown;
 }

package/esm/helpers/kscript/nodes/expressions/ArrowFunctionExpression.js

@@ -1,4 +1,5 @@
 import Node from '../Node';
+import { ReturnStatementError } from '../statements/ReturnStatement';
 import { construct } from '../../utils';
 /** Represents an arrow function expression.
  *
@@ -14,21 +15,34 @@ export default class ArrowFunctionExpressionNode extends Node {
         }
         this.params = this.node.params;
     }
-    run() {
+    validate(validateChildren = false) {
         if (!this.context.isFunctionCallAllowed)
             throw new Error('functions are not allowed');
-        if (!this.node.expression)
-            throw new Error('functions must be an expression');
         if (this.node.async)
             throw new Error('async functions are not supported');
+        if (!validateChildren) {
+            return;
+        }
+        construct(this.node.body, this.context).validate(true);
+    }
+    run() {
+        this.validate();
         return (...args) => {
             const scope = new Map();
             for (const param of this.params) {
                 const index = this.params.indexOf(param);
                 scope.set(param.name, args?.[index]);
             }
-            const newContext = this.context.newChildContext(scope);
-            return construct(this.node.body, newContext).run();
+            const newContext = this.context.newChild(scope);
+            try {
+                return construct(this.node.body, newContext).run();
+            }
+            catch (e) {
+                if (e instanceof ReturnStatementError) {
+                    return e.value;
+                }
+                throw e;
+            }
         };
     }
 }

package/esm/helpers/kscript/nodes/expressions/AssignmentExpression.d.ts

@@ -0,0 +1,8 @@
+import { AnyNode, AssignmentExpression } from 'acorn';
+import Context from '../../Context';
+import Node from '../Node';
+export default class AssignmentExpressionNode extends Node<AssignmentExpression> {
+    constructor(node: AnyNode, context: Context);
+    validate(validateChildren?: boolean): void;
+    run(): unknown;
+}