decaffeinate-parser/dist/mappers/mapLiteral.mjs

A better AST for CoffeeScript, inspired by CoffeeScriptRedux.

import { BooleanLiteral, IdentifierLiteral, NullLiteral, NumberLiteral, PassthroughLiteral, PropertyName, RegexLiteral, StatementLiteral, StringLiteral, ThisLiteral, UndefinedLiteral } from 'decaffeinate-coffeescript2/lib/coffeescript/nodes';
import { Bool, Break, Continue, Float, Identifier, Int, JavaScript, Null, Regex, RegexFlags, This, Undefined } from '../nodes';
import getLocation from '../util/getLocation';
import makeHeregex from '../util/makeHeregex';
import makeString from '../util/makeString';
import parseNumber from '../util/parseNumber';
import parseRegExp from '../util/parseRegExp';
var HEREGEX_PATTERN = /^\/\/\/((?:.|\s)*)\/\/\/([gimuy]*)$/;
export default function mapLiteral(context, node) {
    var _a = getLocation(context, node), line = _a.line, column = _a.column, start = _a.start, end = _a.end, raw = _a.raw;
    if (node instanceof ThisLiteral) {
        return new This(line, column, start, end, raw);
    }
    else if (node instanceof NullLiteral) {
        return new Null(line, column, start, end, raw);
    }
    else if (node instanceof UndefinedLiteral) {
        return new Undefined(line, column, start, end, raw);
    }
    else if (node instanceof BooleanLiteral) {
        return new Bool(line, column, start, end, raw, JSON.parse(node.value));
    }
    else if (node instanceof IdentifierLiteral ||
        node instanceof PropertyName) {
        // Sometimes the CoffeeScript AST contains a string object instead of a
        // string primitive. Convert to string primitive if necessary.
        var value = node.value.valueOf();
        return new Identifier(line, column, start, end, raw, value);
    }
    else if (node instanceof PassthroughLiteral) {
        return new JavaScript(line, column, start, end, raw, node.value);
    }
    else if (node instanceof NumberLiteral) {
        if (raw.includes('.')) {
            return new Float(line, column, start, end, raw, parseNumber(node.value));
        }
        else {
            return new Int(line, column, start, end, raw, parseNumber(node.value));
        }
    }
    else if (node instanceof RegexLiteral) {
        var heregexMatch = raw.match(HEREGEX_PATTERN);
        if (heregexMatch) {
            var flags = heregexMatch[2];
            return makeHeregex(context, node, flags);
        }
        else {
            var regExp = parseRegExp(node.value);
            return new Regex(line, column, start, end, raw, regExp.pattern, RegexFlags.parse(regExp.flags || ''));
        }
    }
    else if (node instanceof StringLiteral) {
        return makeString(context, node);
    }
    else if (node instanceof StatementLiteral) {
        if (node.value === 'break') {
            return new Break(line, column, start, end, raw);
        }
        else if (node.value === 'continue') {
            return new Continue(line, column, start, end, raw);
        }
    }
    throw new Error("Unexpected literal: " + JSON.stringify(node));
}