hackmud-script-manager 0.19.0-f21e319 → 0.19.0-fa82f73

package/processScript/transform.js

@@ -1,394 +1 @@
-import babelTraverse from '@babel/traverse';
-import t from '@babel/types';
-import { assert } from '@samual/lib/assert';
-import { clearObject } from '@samual/lib/clearObject';
-import { validDBMethods } from '../constants.js';
-import { getReferencePathsToGlobal } from './shared.js';
-
-// eslint-disable-next-line @typescript-eslint/consistent-type-imports
-const {
-  default: traverse
-} = babelTraverse;
-const globalFunctionsUnder7Characters = [`Map`, `Set`, `Date`, `JSON`, `Math`, `Array`, `Error`, `isNaN`, `Number`, `Object`, `RegExp`, `String`, `Symbol`, `BigInt`];
-
-/**
- * transform a given babel `File` to be hackmud compatible
- *
- * (returned File will need `postprocess()`ing)
- *
- * @param file babel ast node representing a file containing preprocessed code
- * @param sourceCode the original untouched source code
- * @param options {@link TransformOptions details}
- */
-const transform = (file, sourceCode, {
-  uniqueID = `00000000000`,
-  scriptUser = `UNKNOWN`,
-  scriptName = `UNKNOWN`,
-  seclevel = 4
-} = {}) => {
-  const topFunctionName = `_${uniqueID}_SCRIPT_`;
-  const exports = new Map();
-  const liveExports = new Map();
-  let program;
-  traverse(file, {
-    Program(path) {
-      program = path;
-      path.skip();
-    }
-  });
-  if (program.scope.hasGlobal(`_SOURCE`)) {
-    for (const referencePath of getReferencePathsToGlobal(`_SOURCE`, program)) referencePath.replaceWith(t.stringLiteral(sourceCode));
-  }
-  if (program.scope.hasGlobal(`_BUILD_DATE`)) {
-    for (const referencePath of getReferencePathsToGlobal(`_BUILD_DATE`, program)) referencePath.replaceWith(t.numericLiteral(Date.now()));
-  }
-  if (program.scope.hasGlobal(`_SCRIPT_USER`)) {
-    for (const referencePath of getReferencePathsToGlobal(`_SCRIPT_USER`, program)) {
-      if (scriptUser == true) referencePath.replaceWith(t.stringLiteral(`$${uniqueID}$SCRIPT_USER$`));else referencePath.replaceWith(t.stringLiteral(scriptUser));
-    }
-  }
-  if (program.scope.hasGlobal(`_SCRIPT_NAME`)) {
-    for (const referencePath of getReferencePathsToGlobal(`_SCRIPT_NAME`, program)) {
-      if (scriptName == true) referencePath.replaceWith(t.stringLiteral(`$${uniqueID}$SCRIPT_NAME$`));else referencePath.replaceWith(t.stringLiteral(scriptName));
-    }
-  }
-  if (program.scope.hasGlobal(`_FULL_SCRIPT_NAME`)) {
-    for (const referencePath of getReferencePathsToGlobal(`_FULL_SCRIPT_NAME`, program)) {
-      if (scriptUser == true || scriptName == true) referencePath.replaceWith(t.stringLiteral(`$${uniqueID}$FULL_SCRIPT_NAME$`));else referencePath.replaceWith(t.stringLiteral(`${scriptUser}.${scriptName}`));
-    }
-  }
-  let functionDotPrototypeIsReferencedMultipleTimes = false;
-  const createGetFunctionPrototypeNode = () => {
-    for (const globalFunction of globalFunctionsUnder7Characters) {
-      if (program.scope.hasOwnBinding(globalFunction)) continue;
-      return t.memberExpression(t.memberExpression(t.identifier(globalFunction), t.identifier(`constructor`)), t.identifier(`prototype`));
-    }
-    return t.memberExpression(t.memberExpression(t.arrowFunctionExpression([t.identifier(`_`)], t.identifier(`_`)), t.identifier(`constructor`)), t.identifier(`prototype`));
-  };
-  if (program.scope.hasGlobal(`Function`)) {
-    const FunctionReferencePaths = getReferencePathsToGlobal(`Function`, program);
-    if (FunctionReferencePaths.length == 1) {
-      const referencePath = FunctionReferencePaths[0];
-      assert(referencePath.parent.type == `MemberExpression`, `\`Function\` isn't available in hackmud, only \`Function.prototype\` is accessible`);
-      assert(referencePath.parent.property.type == `Identifier`, `\`Function\` isn't available in hackmud, only \`Function.prototype\` is accessible`);
-      assert(referencePath.parent.property.name == `prototype`, `\`Function\` isn't available in hackmud, only \`Function.prototype\` is accessible`);
-      referencePath.parentPath.replaceWith(createGetFunctionPrototypeNode());
-    } else {
-      for (const referencePath of FunctionReferencePaths) {
-        assert(referencePath.parent.type == `MemberExpression`, `\`Function\` isn't available in hackmud, only \`Function.prototype\` is accessible`);
-        assert(referencePath.parent.property.type == `Identifier`, `\`Function\` isn't available in hackmud, only \`Function.prototype\` is accessible`);
-        assert(referencePath.parent.property.name == `prototype`, `\`Function\` isn't available in hackmud, only \`Function.prototype\` is accessible`);
-        functionDotPrototypeIsReferencedMultipleTimes = true;
-        referencePath.parentPath.replaceWith(t.identifier(`_${uniqueID}_FUNCTION_DOT_PROTOTYPE_`));
-      }
-      functionDotPrototypeIsReferencedMultipleTimes = true;
-    }
-  }
-  let detectedSeclevel = 4;
-  const processFakeSubscriptObject = fakeSubscriptObjectName => {
-    for (const referencePath of getReferencePathsToGlobal(fakeSubscriptObjectName, program)) {
-      var _referencePath$parent, _referencePath$parent2;
-      assert(referencePath.parent.type == `MemberExpression`);
-      assert(referencePath.parent.property.type == `Identifier`);
-      assert(((_referencePath$parent = referencePath.parentPath.parentPath) === null || _referencePath$parent === void 0 ? void 0 : _referencePath$parent.node.type) == `MemberExpression`);
-      assert(referencePath.parentPath.parentPath.node.property.type == `Identifier`);
-      assert(/^[_a-z][\d_a-z]{0,24}$/.test(referencePath.parent.property.name), `invalid user "${referencePath.parent.property.name}" in subscript`);
-      assert(/^[_a-z][\d_a-z]{0,24}$/.test(referencePath.parentPath.parentPath.node.property.name), `invalid script name "${referencePath.parentPath.parentPath.node.property.name}" in subscript`);
-      if (((_referencePath$parent2 = referencePath.parentPath.parentPath.parentPath) === null || _referencePath$parent2 === void 0 ? void 0 : _referencePath$parent2.type) == `CallExpression`) {
-        // BUG this is causing typescript to be slow
-        referencePath.parentPath.parentPath.replaceWith(t.identifier(`$${uniqueID}$SUBSCRIPT$${referencePath.parent.property.name}$${referencePath.parentPath.parentPath.node.property.name}$`));
-      } else {
-        // BUG this is causing typescript to be slow
-        referencePath.parentPath.parentPath.replaceWith(t.arrowFunctionExpression([t.restElement(t.identifier(`args`))], t.callExpression(t.identifier(`$${uniqueID}$SUBSCRIPT$${referencePath.parent.property.name}$${referencePath.parentPath.parentPath.node.property.name}$`), [t.spreadElement(t.identifier(`args`))])));
-      }
-    }
-  };
-  for (const fakeSubscriptObjectName of [`$fs`, `$4s`, `$s`]) {
-    if (program.scope.hasGlobal(fakeSubscriptObjectName)) processFakeSubscriptObject(fakeSubscriptObjectName);
-  }
-  for (const fakeSubscriptObjectName of [`$hs`, `$3s`]) {
-    if (program.scope.hasGlobal(fakeSubscriptObjectName)) {
-      detectedSeclevel = 3;
-      processFakeSubscriptObject(fakeSubscriptObjectName);
-    }
-  }
-  for (const fakeSubscriptObjectName of [`$ms`, `$2s`]) {
-    if (program.scope.hasGlobal(fakeSubscriptObjectName)) {
-      detectedSeclevel = 2;
-      processFakeSubscriptObject(fakeSubscriptObjectName);
-    }
-  }
-  for (const fakeSubscriptObjectName of [`$ls`, `$1s`]) {
-    if (program.scope.hasGlobal(fakeSubscriptObjectName)) {
-      detectedSeclevel = 1;
-      processFakeSubscriptObject(fakeSubscriptObjectName);
-    }
-  }
-  for (const fakeSubscriptObjectName of [`$ns`, `$0s`]) {
-    if (program.scope.hasGlobal(fakeSubscriptObjectName)) {
-      detectedSeclevel = 0;
-      processFakeSubscriptObject(fakeSubscriptObjectName);
-    }
-  }
-  seclevel = Math.min(seclevel, detectedSeclevel);
-  if (program.scope.hasGlobal(`$db`)) {
-    for (const referencePath of getReferencePathsToGlobal(`$db`, program)) {
-      var _referencePath$parent3;
-      assert(referencePath.parentPath.node.type == `MemberExpression`);
-      assert(referencePath.parentPath.node.property.type == `Identifier`);
-      const databaseOpMethodName = referencePath.parentPath.node.property.name;
-      assert(validDBMethods.includes(databaseOpMethodName), `invalid db method "${databaseOpMethodName}", valid db methods are "${validDBMethods.join(`", "`)}"`);
-      if (((_referencePath$parent3 = referencePath.parentPath.parentPath) === null || _referencePath$parent3 === void 0 ? void 0 : _referencePath$parent3.type) == `CallExpression`) referencePath.parentPath.replaceWith(t.identifier(`$${uniqueID}$DB$${databaseOpMethodName}$`));else if (databaseOpMethodName == `ObjectId`) {
-        referencePath.parentPath.replaceWith(t.arrowFunctionExpression([], t.callExpression(t.identifier(`$${uniqueID}$DB$${databaseOpMethodName}$`), [])));
-      } else if (databaseOpMethodName == `i` || databaseOpMethodName == `r`) {
-        referencePath.parentPath.replaceWith(t.arrowFunctionExpression([t.identifier(`a`)], t.callExpression(t.identifier(`$${uniqueID}$DB$${databaseOpMethodName}$`), [t.identifier(`a`)])));
-      } else {
-        referencePath.parentPath.replaceWith(t.arrowFunctionExpression([t.identifier(`a`), t.identifier(`b`)], t.callExpression(t.identifier(`$${uniqueID}$DB$${databaseOpMethodName}$`), [t.identifier(`a`), t.identifier(`b`)])));
-      }
-    }
-  }
-  if (program.scope.hasGlobal(`$D`)) {
-    for (const referencePath of getReferencePathsToGlobal(`$D`, program)) {
-      if (referencePath.parentPath.type == `CallExpression`) referencePath.replaceWith(t.identifier(`$${uniqueID}$DEBUG$`));else {
-        referencePath.replaceWith(t.arrowFunctionExpression([t.identifier(`a`)], t.callExpression(t.identifier(`$${uniqueID}$DEBUG$`), [t.identifier(`a`)])));
-      }
-    }
-  }
-  if (program.scope.hasGlobal(`$FMCL`)) {
-    for (const referencePath of getReferencePathsToGlobal(`$FMCL`, program)) referencePath.replaceWith(t.identifier(`$${uniqueID}$FMCL$`));
-  }
-  if (program.scope.hasGlobal(`$G`)) {
-    for (const referencePath of getReferencePathsToGlobal(`$G`, program)) referencePath.replaceWith(t.identifier(`$${uniqueID}$GLOBAL$`));
-  }
-  if (program.scope.hasGlobal(`_SECLEVEL`)) {
-    for (const referencePath of getReferencePathsToGlobal(`_SECLEVEL`, program)) referencePath.replaceWith(t.numericLiteral(seclevel));
-  }
-  let needGetPrototypeOf = false;
-  let needSetPrototypeOf = false;
-  if (program.scope.hasGlobal(`Object`)) {
-    for (const referencePath of getReferencePathsToGlobal(`Object`, program)) {
-      if (referencePath.parent.type != `MemberExpression` || referencePath.parent.computed) continue;
-      assert(referencePath.parent.property.type == `Identifier`);
-      if (referencePath.parent.property.name == `getPrototypeOf`) {
-        referencePath.parentPath.replaceWith(t.identifier(`_${uniqueID}_GET_PROTOTYPE_OF_`));
-        needGetPrototypeOf = true;
-      } else if (referencePath.parent.property.name == `setPrototypeOf`) {
-        referencePath.parentPath.replaceWith(t.identifier(`_${uniqueID}_SET_PROTOTYPE_OF_`));
-        needSetPrototypeOf = true;
-      }
-    }
-  }
-
-  // rollup removes all the inline exports and places a statement at the end instead
-  const lastStatement = program.node.body.at(-1);
-  let exportDefaultName;
-  assert(lastStatement, `program is empty`);
-  if (lastStatement.type == `ExportNamedDeclaration`) {
-    program.node.body.pop();
-    for (const specifier of lastStatement.specifiers) {
-      assert(specifier.type == `ExportSpecifier`, `${specifier.type} is currently unsupported`);
-      const exportedName = specifier.exported.type == `Identifier` ? specifier.exported.name : specifier.exported.value;
-      if (exportedName == `default`) exportDefaultName = specifier.local.name;else exports.set(specifier.local.name, exportedName);
-    }
-  }
-  const globalBlock = t.blockStatement([]);
-  let mainFunction;
-  for (const statement of program.node.body) {
-    if (statement.type == `VariableDeclaration`) {
-      for (const declarator of statement.declarations) {
-        if (declarator.id.type == `Identifier` && declarator.id.name == exportDefaultName && declarator.init && (declarator.init.type == `FunctionExpression` || declarator.init.type == `ArrowFunctionExpression`) && !declarator.init.async && !declarator.init.generator) {
-          mainFunction = t.functionDeclaration(t.identifier(topFunctionName), declarator.init.params, declarator.init.body.type == `BlockStatement` ? declarator.init.body : t.blockStatement([t.returnStatement(declarator.init.body)]));
-          continue;
-        }
-        for (const identifierName in t.getBindingIdentifiers(declarator.id)) {
-          if (identifierName == exportDefaultName) {
-            mainFunction = t.functionDeclaration(t.identifier(topFunctionName), [t.identifier(`context`), t.identifier(`args`)], t.blockStatement([t.returnStatement(t.callExpression(t.identifier(exportDefaultName), []))]));
-          }
-          if (statement.kind != `const` && exports.has(identifierName)) {
-            liveExports.set(identifierName, exports.get(identifierName));
-            exports.delete(identifierName);
-          }
-          globalBlock.body.push(t.variableDeclaration(`let`, [t.variableDeclarator(t.identifier(identifierName))]));
-        }
-        if (declarator.init) {
-          globalBlock.body.push(t.expressionStatement(t.assignmentExpression(`=`, declarator.id, declarator.init)));
-        }
-      }
-    } else if (statement.type == `FunctionDeclaration`) {
-      if (statement.id.name == exportDefaultName) mainFunction = statement;else {
-        globalBlock.body.push(t.variableDeclaration(`let`, [t.variableDeclarator(statement.id, t.functionExpression(undefined, statement.params, statement.body, statement.generator, statement.async))]));
-      }
-    } else globalBlock.body.push(statement);
-  }
-  mainFunction || (mainFunction = t.functionDeclaration(t.identifier(topFunctionName), [t.identifier(`context`), t.identifier(`args`)], t.blockStatement([])));
-  program.node.body = [mainFunction];
-  if (globalBlock.body.length) {
-    if (exports.size || liveExports.size) {
-      mainFunction.body.body.push(t.returnStatement(t.objectExpression([...[...exports].map(([local, exported]) => t.objectProperty(t.identifier(exported), t.identifier(local))), ...[...liveExports].map(([local, exported]) => t.objectMethod(`get`, t.identifier(exported), [], t.blockStatement([t.returnStatement(t.identifier(local))])))])));
-    }
-    program.scope.crawl();
-    const globalBlockVariables = new Set();
-    let hoistedGlobalBlockFunctions = 0;
-    for (const [globalBlockIndex, globalBlockStatement] of [...globalBlock.body.entries()].reverse()) {
-      if (globalBlockStatement.type == `VariableDeclaration`) {
-        assert(globalBlockStatement.declarations.length == 1);
-        const declarator = globalBlockStatement.declarations[0];
-        assert(declarator.id.type == `Identifier`, `declarator.id.type was "${declarator.id.type}"`);
-        program.scope.crawl();
-        if (program.scope.hasGlobal(declarator.id.name)) {
-          globalBlock.body.splice(globalBlockIndex, 1);
-          const [globalBlockPath] = program.unshiftContainer(`body`, globalBlock);
-          const [globalBlockStatementPath] = program.unshiftContainer(`body`, globalBlockStatement);
-          program.scope.crawl();
-          if (!declarator.init || declarator.init.type != `FunctionExpression` && declarator.init.type != `ArrowFunctionExpression` || Object.keys(program.scope.globals).some(global => globalBlockVariables.has(global))) {
-            const binding = program.scope.getBinding(declarator.id.name);
-            assert(binding);
-            for (const referencePath of binding.referencePaths) {
-              assert(referencePath.node.type == `Identifier`);
-              referencePath.replaceWith(t.memberExpression(t.identifier(`$${uniqueID}$GLOBAL$`), t.identifier(referencePath.node.name)));
-            }
-            for (const referencePath of binding.constantViolations) {
-              if (referencePath.node.type != `AssignmentExpression`) continue;
-              for (const [name, node] of Object.entries(t.getBindingIdentifiers(referencePath.node))) {
-                if (name == declarator.id.name) {
-                  clearObject(node);
-                  Object.assign(node, t.memberExpression(t.identifier(`$${uniqueID}$GLOBAL$`), t.identifier(name)));
-                }
-              }
-            }
-            globalBlockPath.remove();
-            globalBlockStatementPath.remove();
-            if (declarator.init) {
-              globalBlock.body.splice(globalBlockIndex, 0, t.expressionStatement(t.assignmentExpression(`=`, t.memberExpression(t.identifier(`$${uniqueID}$GLOBAL$`), t.identifier(declarator.id.name)), declarator.init)));
-            }
-          } else {
-            globalBlockPath.remove();
-            globalBlockStatementPath.remove();
-            mainFunction.body.body.unshift(globalBlockStatement);
-            hoistedGlobalBlockFunctions++;
-          }
-        } else globalBlockVariables.add(declarator.id.name);
-      } else if (globalBlockStatement.type == `ClassDeclaration`) {
-        program.scope.crawl();
-        if (program.scope.hasGlobal(globalBlockStatement.id.name)) {
-          globalBlock.body.splice(globalBlockIndex, 1);
-          const [globalBlockPath] = program.unshiftContainer(`body`, globalBlock);
-          const [globalBlockStatementPath] = program.unshiftContainer(`body`, globalBlockStatement);
-          program.scope.crawl();
-          const binding = program.scope.getBinding(globalBlockStatement.id.name);
-          assert(binding);
-          for (const referencePath of binding.referencePaths) {
-            assert(referencePath.node.type == `Identifier`);
-            referencePath.replaceWith(t.memberExpression(t.identifier(`$${uniqueID}$GLOBAL$`), t.identifier(referencePath.node.name)));
-          }
-          globalBlockPath.remove();
-          globalBlockStatementPath.remove();
-          globalBlock.body.splice(globalBlockIndex, 0, t.expressionStatement(t.assignmentExpression(`=`, t.memberExpression(t.identifier(`$${uniqueID}$GLOBAL$`), t.identifier(globalBlockStatement.id.name)), t.classExpression(undefined, globalBlockStatement.superClass, globalBlockStatement.body, globalBlockStatement.decorators))));
-        }
-      }
-    }
-    if (program.scope.hasGlobal(`_EXPORTS`)) {
-      for (const referencePath of getReferencePathsToGlobal(`_EXPORTS`, program)) {
-        referencePath.replaceWith(t.arrayExpression([...exports.keys(), ...liveExports.keys()].map(name => t.stringLiteral(name))));
-      }
-    }
-    if (globalBlock.body.length) {
-      mainFunction.body.body.splice(hoistedGlobalBlockFunctions, 0, t.ifStatement(t.unaryExpression(`!`, t.identifier(`$${uniqueID}$FMCL$`)), globalBlock));
-    }
-  }
-  if (functionDotPrototypeIsReferencedMultipleTimes) {
-    mainFunction.body.body.unshift(t.variableDeclaration(`let`, [t.variableDeclarator(t.identifier(`_${uniqueID}_FUNCTION_DOT_PROTOTYPE_`), createGetFunctionPrototypeNode())]));
-  }
-  if (needSetPrototypeOf) {
-    mainFunction.body.body.unshift(t.variableDeclaration(`let`, [t.variableDeclarator(t.identifier(`_${uniqueID}_SET_PROTOTYPE_OF_`), t.callExpression(t.memberExpression(t.memberExpression(t.identifier(`Object`), t.identifier(`call`)), t.identifier(`bind`)), [t.identifier(`_${uniqueID}_DUNDER_PROTO_SETTER_`)]))]));
-  }
-  if (needGetPrototypeOf) {
-    mainFunction.body.body.unshift(t.variableDeclaration(`let`, [t.variableDeclarator(t.identifier(`_${uniqueID}_GET_PROTOTYPE_OF_`), t.callExpression(t.memberExpression(t.memberExpression(t.identifier(`Object`), t.identifier(`call`)), t.identifier(`bind`)), [t.identifier(`_${uniqueID}_DUNDER_PROTO_GETTER_`)]))]));
-  }
-  if (needGetPrototypeOf || needSetPrototypeOf) {
-    mainFunction.body.body.unshift(t.variableDeclaration(`let`, [t.variableDeclarator(t.objectPattern(needGetPrototypeOf ? needSetPrototypeOf ? [t.objectProperty(t.identifier(`get`), t.identifier(`_${uniqueID}_DUNDER_PROTO_GETTER_`)), t.objectProperty(t.identifier(`set`), t.identifier(`_${uniqueID}_DUNDER_PROTO_SETTER_`))] : [t.objectProperty(t.identifier(`get`), t.identifier(`_${uniqueID}_DUNDER_PROTO_GETTER_`))] : [t.objectProperty(t.identifier(`set`), t.identifier(`_${uniqueID}_DUNDER_PROTO_SETTER_`))]), t.callExpression(t.memberExpression(t.identifier(`Object`), t.identifier(`getOwnPropertyDescriptor`)), [t.memberExpression(t.identifier(`Object`), t.identifier(`prototype`)), t.stringLiteral(`__proto__`)]))]));
-  }
-  traverse(file, {
-    BlockStatement({
-      node: blockStatement
-    }) {
-      for (const [index, functionDeclaration] of blockStatement.body.entries()) {
-        if (functionDeclaration.type != `FunctionDeclaration` || functionDeclaration.generator) continue;
-        blockStatement.body.splice(index, 1);
-        blockStatement.body.unshift(t.variableDeclaration(`let`, [t.variableDeclarator(functionDeclaration.id, t.arrowFunctionExpression(functionDeclaration.params, functionDeclaration.body, functionDeclaration.async))]));
-      }
-    },
-    ClassBody({
-      node: classBody,
-      scope,
-      parent
-    }) {
-      assert(t.isClass(parent));
-      let thisIsReferenced = false;
-      for (const classMethod of classBody.body) {
-        if (classMethod.type != `ClassMethod`) continue;
-        let methodReferencesThis = false;
-        traverse(classMethod.body, {
-          ThisExpression(path) {
-            methodReferencesThis = true;
-            thisIsReferenced = true;
-            path.replaceWith(t.identifier(`_${uniqueID}_THIS_`));
-          },
-          Function(path) {
-            path.skip();
-          }
-        }, scope);
-        if (!methodReferencesThis) continue;
-        if (classMethod.kind == `constructor`) {
-          const superCalls = [];
-          traverse(classMethod.body, {
-            CallExpression(path) {
-              if (path.node.callee.type == `Super`) superCalls.push(path);
-            }
-          }, scope);
-          if (!superCalls.length) {
-            classMethod.body.body.unshift(t.variableDeclaration(`let`, [t.variableDeclarator(t.identifier(`_${uniqueID}_THIS_`), t.callExpression(t.super(), []))]));
-          } else if (superCalls.length == 1 && superCalls[0].parent.type == `ExpressionStatement` && superCalls[0].parentPath.parentPath.parent == classMethod) {
-            superCalls[0].parentPath.replaceWith(t.variableDeclaration(`let`, [t.variableDeclarator(t.identifier(`_${uniqueID}_THIS_`), superCalls[0].node)]));
-          } else {
-            for (const path of superCalls) {
-              path.replaceWith(t.assignmentExpression(`=`, t.identifier(`_${uniqueID}_THIS_`), path.node));
-            }
-            classMethod.body.body.unshift(t.variableDeclaration(`let`, [t.variableDeclarator(t.identifier(`_${uniqueID}_THIS_`))]));
-          }
-          continue;
-        }
-
-        // BUG if the class or a super class overwrites `valueOf()` (or `Object.prototype` isn't even in the chain), this breaks
-        // TODO track whether the class is extending a class that at some point extends from `Object` (if unsure, assume no)
-        // TODO track whether any class in the chain overwrites `valueOf()` (if unsure, assume yes)
-        // TODO for classes that need it, create a super class for this one to extend from with `valueOf()` assigned to an unused name
-
-        classMethod.body.body.unshift(t.variableDeclaration(`let`, [t.variableDeclarator(t.identifier(`_${uniqueID}_THIS_`), t.callExpression(t.memberExpression(t.super(), t.identifier(`valueOf`)), []))]));
-      }
-      if (!parent.superClass && thisIsReferenced) parent.superClass = t.identifier(`Object`);
-    },
-    VariableDeclaration({
-      node: variableDeclaration
-    }) {
-      if (variableDeclaration.kind == `const`) variableDeclaration.kind = `let`;
-    },
-    ThisExpression(path) {
-      path.replaceWith(t.identifier(`undefined`));
-    },
-    BigIntLiteral(path) {
-      const bigIntAsNumber = Number(path.node.value);
-      if (BigInt(bigIntAsNumber) == BigInt(path.node.value)) {
-        path.replaceWith(t.callExpression(t.identifier(`BigInt`), [t.numericLiteral(bigIntAsNumber)]));
-      } else {
-        path.replaceWith(t.callExpression(t.identifier(`BigInt`), [t.stringLiteral(path.node.value)]));
-      }
-    }
-  });
-  return {
-    file,
-    seclevel
-  };
-};
-
-export { transform as default, transform };
+import e from"@babel/traverse";import i from"@babel/types";import{assert as t}from"@samual/lib/assert";import{clearObject as r}from"@samual/lib/clearObject";import{validDBMethods as n}from"../constants.js";import{getReferencePathsToGlobal as o}from"./shared.js";const{default:a}=e,s=["Map","Set","Date","JSON","Math","Array","Error","isNaN","Number","Object","RegExp","String","Symbol","BigInt"],transform=(e,p,{uniqueID:l="00000000000",scriptUser:c="UNKNOWN",scriptName:d="UNKNOWN",seclevel:f=4}={})=>{const b=`_${l}_SCRIPT_`,y=new Map,m=new Map;let $;if(a(e,{Program(e){$=e,e.skip()}}),$.scope.hasGlobal("_SOURCE"))for(const e of o("_SOURCE",$))e.replaceWith(i.stringLiteral(p));if($.scope.hasGlobal("_BUILD_DATE"))for(const e of o("_BUILD_DATE",$))e.replaceWith(i.numericLiteral(Date.now()));if($.scope.hasGlobal("_SCRIPT_USER"))for(const e of o("_SCRIPT_USER",$))1==c?e.replaceWith(i.stringLiteral(`$${l}$SCRIPT_USER$`)):e.replaceWith(i.stringLiteral(c));if($.scope.hasGlobal("_SCRIPT_NAME"))for(const e of o("_SCRIPT_NAME",$))1==d?e.replaceWith(i.stringLiteral(`$${l}$SCRIPT_NAME$`)):e.replaceWith(i.stringLiteral(d));if($.scope.hasGlobal("_FULL_SCRIPT_NAME"))for(const e of o("_FULL_SCRIPT_NAME",$))1==c||1==d?e.replaceWith(i.stringLiteral(`$${l}$FULL_SCRIPT_NAME$`)):e.replaceWith(i.stringLiteral(`${c}.${d}`));let h=!1;const createGetFunctionPrototypeNode=()=>{for(const e of s)if(!$.scope.hasOwnBinding(e))return i.memberExpression(i.memberExpression(i.identifier(e),i.identifier("constructor")),i.identifier("prototype"));return i.memberExpression(i.memberExpression(i.arrowFunctionExpression([i.identifier("_")],i.identifier("_")),i.identifier("constructor")),i.identifier("prototype"))};if($.scope.hasGlobal("Function")){const e=o("Function",$);if(1==e.length){const i=e[0];t("MemberExpression"==i.parent.type,"`Function` isn't available in hackmud, only `Function.prototype` is accessible"),t("Identifier"==i.parent.property.type,"`Function` isn't available in hackmud, only `Function.prototype` is accessible"),t("prototype"==i.parent.property.name,"`Function` isn't available in hackmud, only `Function.prototype` is accessible"),i.parentPath.replaceWith(createGetFunctionPrototypeNode())}else{for(const r of e)t("MemberExpression"==r.parent.type,"`Function` isn't available in hackmud, only `Function.prototype` is accessible"),t("Identifier"==r.parent.property.type,"`Function` isn't available in hackmud, only `Function.prototype` is accessible"),t("prototype"==r.parent.property.name,"`Function` isn't available in hackmud, only `Function.prototype` is accessible"),h=!0,r.parentPath.replaceWith(i.identifier(`_${l}_FUNCTION_DOT_PROTOTYPE_`));h=!0}}let E=4;const processFakeSubscriptObject=e=>{for(const r of o(e,$))t("MemberExpression"==r.parent.type),t("Identifier"==r.parent.property.type),t("MemberExpression"==r.parentPath.parentPath?.node.type),t("Identifier"==r.parentPath.parentPath.node.property.type),t(/^[_a-z][\d_a-z]{0,24}$/.test(r.parent.property.name),`invalid user "${r.parent.property.name}" in subscript`),t(/^[_a-z][\d_a-z]{0,24}$/.test(r.parentPath.parentPath.node.property.name),`invalid script name "${r.parentPath.parentPath.node.property.name}" in subscript`),"CallExpression"==r.parentPath.parentPath.parentPath?.type?r.parentPath.parentPath.replaceWith(i.identifier(`$${l}$SUBSCRIPT$${r.parent.property.name}$${r.parentPath.parentPath.node.property.name}$`)):r.parentPath.parentPath.replaceWith(i.arrowFunctionExpression([i.restElement(i.identifier("args"))],i.callExpression(i.identifier(`$${l}$SUBSCRIPT$${r.parent.property.name}$${r.parentPath.parentPath.node.property.name}$`),[i.spreadElement(i.identifier("args"))])))};for(const e of["$fs","$4s","$s"])$.scope.hasGlobal(e)&&processFakeSubscriptObject(e);for(const e of["$hs","$3s"])$.scope.hasGlobal(e)&&(E=3,processFakeSubscriptObject(e));for(const e of["$ms","$2s"])$.scope.hasGlobal(e)&&(E=2,processFakeSubscriptObject(e));for(const e of["$ls","$1s"])$.scope.hasGlobal(e)&&(E=1,processFakeSubscriptObject(e));for(const e of["$ns","$0s"])$.scope.hasGlobal(e)&&(E=0,processFakeSubscriptObject(e));if(f=Math.min(f,E),$.scope.hasGlobal("$db"))for(const e of o("$db",$)){t("MemberExpression"==e.parentPath.node.type),t("Identifier"==e.parentPath.node.property.type);const r=e.parentPath.node.property.name;t(n.includes(r),`invalid db method "${r}", valid db methods are "${n.join('", "')}"`),"CallExpression"==e.parentPath.parentPath?.type?e.parentPath.replaceWith(i.identifier(`$${l}$DB$${r}$`)):"ObjectId"==r?e.parentPath.replaceWith(i.arrowFunctionExpression([],i.callExpression(i.identifier(`$${l}$DB$${r}$`),[]))):"i"==r||"r"==r?e.parentPath.replaceWith(i.arrowFunctionExpression([i.identifier("a")],i.callExpression(i.identifier(`$${l}$DB$${r}$`),[i.identifier("a")]))):e.parentPath.replaceWith(i.arrowFunctionExpression([i.identifier("a"),i.identifier("b")],i.callExpression(i.identifier(`$${l}$DB$${r}$`),[i.identifier("a"),i.identifier("b")])))}if($.scope.hasGlobal("$D"))for(const e of o("$D",$))"CallExpression"==e.parentPath.type?e.replaceWith(i.identifier(`$${l}$DEBUG$`)):e.replaceWith(i.arrowFunctionExpression([i.identifier("a")],i.callExpression(i.identifier(`$${l}$DEBUG$`),[i.identifier("a")])));if($.scope.hasGlobal("$FMCL"))for(const e of o("$FMCL",$))e.replaceWith(i.identifier(`$${l}$FMCL$`));if($.scope.hasGlobal("$G"))for(const e of o("$G",$))e.replaceWith(i.identifier(`$${l}$GLOBAL$`));if($.scope.hasGlobal("_SECLEVEL"))for(const e of o("_SECLEVEL",$))e.replaceWith(i.numericLiteral(f));let _=!1,u=!1;if($.scope.hasGlobal("Object"))for(const e of o("Object",$))"MemberExpression"!=e.parent.type||e.parent.computed||(t("Identifier"==e.parent.property.type),"getPrototypeOf"==e.parent.property.name?(e.parentPath.replaceWith(i.identifier(`_${l}_GET_PROTOTYPE_OF_`)),_=!0):"setPrototypeOf"==e.parent.property.name&&(e.parentPath.replaceWith(i.identifier(`_${l}_SET_PROTOTYPE_OF_`)),u=!0));const P=$.node.body.at(-1);let x;if(t(P,"program is empty"),"ExportNamedDeclaration"==P.type){$.node.body.pop();for(const e of P.specifiers){t("ExportSpecifier"==e.type,`${e.type} is currently unsupported`);const i="Identifier"==e.exported.type?e.exported.name:e.exported.value;"default"==i?x=e.local.name:y.set(e.local.name,i)}}const O=i.blockStatement([]);let D;for(const e of $.node.body)if("VariableDeclaration"==e.type)for(const t of e.declarations)if("Identifier"!=t.id.type||t.id.name!=x||!t.init||"FunctionExpression"!=t.init.type&&"ArrowFunctionExpression"!=t.init.type||t.init.async||t.init.generator){for(const r in i.getBindingIdentifiers(t.id))r==x&&(D=i.functionDeclaration(i.identifier(b),[i.identifier("context"),i.identifier("args")],i.blockStatement([i.returnStatement(i.callExpression(i.identifier(x),[]))]))),"const"!=e.kind&&y.has(r)&&(m.set(r,y.get(r)),y.delete(r)),O.body.push(i.variableDeclaration("let",[i.variableDeclarator(i.identifier(r))]));t.init&&O.body.push(i.expressionStatement(i.assignmentExpression("=",t.id,t.init)))}else D=i.functionDeclaration(i.identifier(b),t.init.params,"BlockStatement"==t.init.body.type?t.init.body:i.blockStatement([i.returnStatement(t.init.body)]));else"FunctionDeclaration"==e.type?e.id.name==x?D=e:O.body.push(i.variableDeclaration("let",[i.variableDeclarator(e.id,i.functionExpression(void 0,e.params,e.body,e.generator,e.async))])):O.body.push(e);if(D||=i.functionDeclaration(i.identifier(b),[i.identifier("context"),i.identifier("args")],i.blockStatement([])),$.node.body=[D],O.body.length){(y.size||m.size)&&D.body.body.push(i.returnStatement(i.objectExpression([...[...y].map((([e,t])=>i.objectProperty(i.identifier(t),i.identifier(e)))),...[...m].map((([e,t])=>i.objectMethod("get",i.identifier(t),[],i.blockStatement([i.returnStatement(i.identifier(e))]))))]))),$.scope.crawl();const e=new Set;let n=0;for(const[o,a]of[...O.body.entries()].reverse())if("VariableDeclaration"==a.type){t(1==a.declarations.length);const s=a.declarations[0];if(t("Identifier"==s.id.type,`declarator.id.type was "${s.id.type}"`),$.scope.crawl(),$.scope.hasGlobal(s.id.name)){O.body.splice(o,1);const[p]=$.unshiftContainer("body",O),[c]=$.unshiftContainer("body",a);if($.scope.crawl(),!s.init||"FunctionExpression"!=s.init.type&&"ArrowFunctionExpression"!=s.init.type||Object.keys($.scope.globals).some((i=>e.has(i)))){const e=$.scope.getBinding(s.id.name);t(e);for(const r of e.referencePaths)t("Identifier"==r.node.type),r.replaceWith(i.memberExpression(i.identifier(`$${l}$GLOBAL$`),i.identifier(r.node.name)));for(const t of e.constantViolations)if("AssignmentExpression"==t.node.type)for(const[e,n]of Object.entries(i.getBindingIdentifiers(t.node)))e==s.id.name&&(r(n),Object.assign(n,i.memberExpression(i.identifier(`$${l}$GLOBAL$`),i.identifier(e))));p.remove(),c.remove(),s.init&&O.body.splice(o,0,i.expressionStatement(i.assignmentExpression("=",i.memberExpression(i.identifier(`$${l}$GLOBAL$`),i.identifier(s.id.name)),s.init)))}else p.remove(),c.remove(),D.body.body.unshift(a),n++}else e.add(s.id.name)}else if("ClassDeclaration"==a.type&&($.scope.crawl(),t(a.id,"src/processScript/transform.ts:564:37"),$.scope.hasGlobal(a.id.name))){O.body.splice(o,1);const[e]=$.unshiftContainer("body",O),[r]=$.unshiftContainer("body",a);$.scope.crawl();const n=$.scope.getBinding(a.id.name);t(n);for(const e of n.referencePaths)t("Identifier"==e.node.type),e.replaceWith(i.memberExpression(i.identifier(`$${l}$GLOBAL$`),i.identifier(e.node.name)));e.remove(),r.remove(),O.body.splice(o,0,i.expressionStatement(i.assignmentExpression("=",i.memberExpression(i.identifier(`$${l}$GLOBAL$`),i.identifier(a.id.name)),i.classExpression(void 0,a.superClass,a.body,a.decorators))))}if($.scope.hasGlobal("_EXPORTS"))for(const e of o("_EXPORTS",$))e.replaceWith(i.arrayExpression([...y.keys(),...m.keys()].map((e=>i.stringLiteral(e)))));O.body.length&&D.body.body.splice(n,0,i.ifStatement(i.unaryExpression("!",i.identifier(`$${l}$FMCL$`)),O))}return h&&D.body.body.unshift(i.variableDeclaration("let",[i.variableDeclarator(i.identifier(`_${l}_FUNCTION_DOT_PROTOTYPE_`),createGetFunctionPrototypeNode())])),u&&D.body.body.unshift(i.variableDeclaration("let",[i.variableDeclarator(i.identifier(`_${l}_SET_PROTOTYPE_OF_`),i.callExpression(i.memberExpression(i.memberExpression(i.identifier("Object"),i.identifier("call")),i.identifier("bind")),[i.identifier(`_${l}_DUNDER_PROTO_SETTER_`)]))])),_&&D.body.body.unshift(i.variableDeclaration("let",[i.variableDeclarator(i.identifier(`_${l}_GET_PROTOTYPE_OF_`),i.callExpression(i.memberExpression(i.memberExpression(i.identifier("Object"),i.identifier("call")),i.identifier("bind")),[i.identifier(`_${l}_DUNDER_PROTO_GETTER_`)]))])),(_||u)&&D.body.body.unshift(i.variableDeclaration("let",[i.variableDeclarator(i.objectPattern(_?u?[i.objectProperty(i.identifier("get"),i.identifier(`_${l}_DUNDER_PROTO_GETTER_`)),i.objectProperty(i.identifier("set"),i.identifier(`_${l}_DUNDER_PROTO_SETTER_`))]:[i.objectProperty(i.identifier("get"),i.identifier(`_${l}_DUNDER_PROTO_GETTER_`))]:[i.objectProperty(i.identifier("set"),i.identifier(`_${l}_DUNDER_PROTO_SETTER_`))]),i.callExpression(i.memberExpression(i.identifier("Object"),i.identifier("getOwnPropertyDescriptor")),[i.memberExpression(i.identifier("Object"),i.identifier("prototype")),i.stringLiteral("__proto__")]))])),a(e,{BlockStatement({node:e}){for(const[t,r]of e.body.entries())"FunctionDeclaration"!=r.type||r.generator||(e.body.splice(t,1),e.body.unshift(i.variableDeclaration("let",[i.variableDeclarator(r.id,i.arrowFunctionExpression(r.params,r.body,r.async))])))},ClassBody({node:e,scope:r,parent:n}){t(i.isClass(n));let o=!1;for(const t of e.body){if("ClassMethod"!=t.type)continue;let e=!1;if(a(t.body,{ThisExpression(t){e=!0,o=!0,t.replaceWith(i.identifier(`_${l}_THIS_`))},Function(e){e.skip()}},r),e)if("constructor"!=t.kind)t.body.body.unshift(i.variableDeclaration("let",[i.variableDeclarator(i.identifier(`_${l}_THIS_`),i.callExpression(i.memberExpression(i.super(),i.identifier("valueOf")),[]))]));else{const e=[];if(a(t.body,{CallExpression(i){"Super"==i.node.callee.type&&e.push(i)}},r),e.length)if(1==e.length&&"ExpressionStatement"==e[0].parent.type&&e[0].parentPath.parentPath.parent==t)e[0].parentPath.replaceWith(i.variableDeclaration("let",[i.variableDeclarator(i.identifier(`_${l}_THIS_`),e[0].node)]));else{for(const t of e)t.replaceWith(i.assignmentExpression("=",i.identifier(`_${l}_THIS_`),t.node));t.body.body.unshift(i.variableDeclaration("let",[i.variableDeclarator(i.identifier(`_${l}_THIS_`))]))}else t.body.body.unshift(i.variableDeclaration("let",[i.variableDeclarator(i.identifier(`_${l}_THIS_`),i.callExpression(i.super(),[]))]))}}!n.superClass&&o&&(n.superClass=i.identifier("Object"))},VariableDeclaration({node:e}){"const"==e.kind&&(e.kind="let")},ThisExpression(e){e.replaceWith(i.identifier("undefined"))},BigIntLiteral(e){const t=Number(e.node.value);BigInt(t)==BigInt(e.node.value)?e.replaceWith(i.callExpression(i.identifier("BigInt"),[i.numericLiteral(t)])):e.replaceWith(i.callExpression(i.identifier("BigInt"),[i.stringLiteral(e.node.value)]))}}),{file:e,seclevel:f}};export{transform as default,transform};

package/pull.js

@@ -1,17 +1 @@
-import { copyFilePersistent } from '@samual/lib/copyFilePersistent';
-import { resolve } from 'path';
-
-/**
- * Copies script from hackmud to local source folder.
- *
- * @param sourceFolderPath path to folder containing source files
- * @param hackmudPath path to hackmud directory
- * @param script to pull in `user.name` format
- */
-const pull = async (sourceFolderPath, hackmudPath, script) => {
-  const [user, name] = script.split(`.`);
-  if (!user || !name) throw new Error(`\`script\` argument must be in "user.name" format`);
-  await copyFilePersistent(resolve(hackmudPath, user, `scripts`, `${name}.js`), resolve(sourceFolderPath, user, `${name}.js`));
-};
-
-export { pull as default, pull };
+import{copyFilePersistent as t}from"@samual/lib/copyFilePersistent";import{resolve as r}from"path";const pull=async(s,a,i)=>{const[o,e]=i.split(".");if(!o||!e)throw new Error('`script` argument must be in "user.name" format');await t(r(a,o,"scripts",`${e}.js`),r(s,o,`${e}.js`))};export{pull as default,pull};