
import * as basic from "./compiler";
import { EmuHalt } from "../emu";
import { SourceLocation } from "../workertypes";
import { isArray } from "../util";

function isLiteral(arg: basic.Expr): arg is basic.Literal {
    return (arg as any).value != null;
}
function isLookup(arg: basic.Expr): arg is basic.IndOp {
    return (arg as any).name != null;
}
function isBinOp(arg: basic.Expr): arg is basic.BinOp {
    return (arg as any).op != null && (arg as any).left != null && (arg as any).right != null;
}
function isUnOp(arg: basic.Expr): arg is basic.UnOp {
    return (arg as any).op != null && (arg as any).expr != null;
}

export interface InputResponse {
    line: string;
    vals: string[];
    elapsed?: number;
}

// expr2js() options
class ExprOptions {
    isconst?: boolean;      // only allow constant operations
    novalid?: boolean;      // check for valid values when fetching
    locals?: string[];      // pass local variable names when defining functions
}

interface CompiledStatement {
    $run?: () => void;
}

class RNG {
    next : () => number;
    seed : (aa,bb,cc,dd) => void;
    seedfloat : (n) => void;
    randomize() {
        this.seed(Math.random()*0x7fffffff, Math.random()*0x7fffffff, Math.random()*0x7fffffff, Math.random()*0x7fffffff);
    }
    constructor() {
        let f = () => {
            var a, b, c, d : number;
            this.seed = function(aa,bb,cc,dd) {
                a = aa; b = bb; c = cc; d = dd;
            }
            this.seedfloat = function(n) {
                this.seed(n, n*4294, n*429496, n*4294967296);
                this.next(); this.next(); this.next();
            }
            this.next = function() {
                // sfc32
                a >>>= 0; b >>>= 0; c >>>= 0; d >>>= 0; 
                var t = (a + b) | 0;
                a = b ^ b >>> 9;
                b = c + (c << 3) | 0;
                c = (c << 21 | c >>> 11);
                d = d + 1 | 0;
                t = t + d | 0;
                c = c + t | 0;
                return (t >>> 0) / 4294967296;
            }
        };
        f();
        this.seedfloat(-1);
    }
};

const DEFAULT_MAX_ARRAY_ELEMENTS = 1024*1024;

export class BASICRuntime {

    program : basic.BASICProgram;
    allstmts : basic.Statement[];
    pc2label : Map<number,string>;
    label2pc : {[label : string] : number};
    label2dataptr : {[label : string] : number};
    datums : basic.Literal[];
    builtins : {};
    opts : basic.BASICOptions;
    margin : number = 80; // number of columns

    curpc : number;
    dataptr : number;
    vars : {[name:string] : any}; // actually Value, but += doesn't work
    globals : {[name:string] : any};
    arrays : {};
    defs : {};
    subroutines : {};
    forLoops : { [varname:string] : { $next:(name:string) => void } };
    forLoopStack: string[];
    whileLoops : number[];
    returnStack : number[];
    column : number;
    rng : RNG;

    running : boolean = false;
    exited : boolean = true;
    trace : boolean = false;

    load(program: basic.BASICProgram) : boolean {
        // get previous label and offset for hot reload
        let prevlabel = null;
        let prevpcofs = 0;
        if (this.pc2label != null) {
            let pc = this.curpc;
            while (pc > 0 && (prevlabel = this.pc2label.get(pc)) == null) {
                pc--;
            }
            prevpcofs = this.curpc - pc;
            console.log('oldpc=', this.curpc, 'restart @ label', prevlabel, '+', prevpcofs);
        }
        // initialize program
        this.program = program;
        this.opts = program.opts;
        if (!this.opts.maxArrayElements) this.opts.maxArrayElements = DEFAULT_MAX_ARRAY_ELEMENTS;
        this.allstmts = program.stmts;
        this.label2pc = program.labels;
        this.label2dataptr = {};
        this.pc2label = new Map();
        this.datums = [];
        this.subroutines = {};
        this.builtins = this.getBuiltinFunctions();
        // TODO: detect undeclared vars
        // build PC -> label lookup
        for (var label in program.labels) {
            var targetpc = program.labels[label];
            this.pc2label.set(targetpc, label);
        }
        // iterate through all the statements
        this.allstmts.forEach((stmt, pc) => {
            // compile statements ahead of time
            this.curpc = pc + 1; // for error reporting
            this.compileStatement(stmt);
            // parse DATA literals
            if (stmt.command == 'DATA') {
                this.label2dataptr[stmt.$loc.label] = this.datums.length;
                (stmt as basic.DATA_Statement).datums.forEach(datum => {
                    this.curpc = stmt.$loc.offset; // for error reporting
                    this.datums.push(datum);
                });
            }
        });
        // try to resume where we left off after loading
        if (this.label2pc[prevlabel] != null) {
            this.curpc = this.label2pc[prevlabel] + prevpcofs;
            return true;
        } else {
            this.curpc = 0;
            return false;
        }
    }

    reset() {
        this.curpc = 0;
        this.dataptr = 0;
        this.clearVars();
        this.returnStack = [];
        this.column = 0;
        this.running = true;
        this.exited = false;
    }
    clearVars() {
        this.globals = this.vars = {};
        this.arrays = {};
        this.defs = {}; // TODO? only in interpreters
        this.forLoops = {};
        this.forLoopStack = [];
        this.whileLoops = [];
        this.rng = new RNG();
        // initialize arrays?
        if (this.opts && this.opts.staticArrays) {
            this.allstmts.filter((stmt) => stmt.command == 'DIM').forEach((dimstmt: basic.DIM_Statement) => {
                dimstmt.args.forEach( (arg) => this.compileJS(this._DIM(arg))() );
            });
        }
    }
    
    // TODO: saveState(), loadState()
    saveState() {
        // TODO: linked list loop?
        return $.extend(true, {}, this);
    }
    loadState(state) {
        $.extend(true, this, state);
    }
    
    getBuiltinFunctions() {
        var fnames = this.program && this.opts.validFunctions;
        // if no valid function list, look for ABC...() functions in prototype
        if (!fnames) fnames = Object.getOwnPropertyNames(BASICRuntime.prototype).filter((name) => /^[A-Z]{3,}[$]?$/.test(name));
        var dict = {};
        for (var fn of fnames) if (this.supportsFunction(fn)) dict[fn] = this[fn].bind(this);
        return dict;
    }
    supportsFunction(fnname: string) {
        return typeof this[fnname] === 'function';
    }

    runtimeError(msg : string) {
        this.curpc--; // we did curpc++ before executing statement
        throw new EmuHalt(msg, this.getCurrentSourceLocation());
    }
    
    dialectError(what : string) {
        this.runtimeError(`I can't ${what} in this dialect of BASIC.`);
    }

    getLineForPC(pc:number) {
        var stmt = this.allstmts[pc];
        return stmt && stmt.$loc && stmt.$loc.line;
    }

    getLabelForPC(pc:number) {
        var stmt = this.allstmts[pc];
        return stmt && stmt.$loc && stmt.$loc.label;
    }

    getCurrentSourceLocation() : SourceLocation {
        var stmt = this.getStatement();
        return stmt && stmt.$loc;
    }

    getCurrentLabel() : string {
        var loc = this.getCurrentSourceLocation();
        return loc && loc.label;
    }

    getStatement() {
        return this.allstmts[this.curpc];
    }

    step() : boolean {
        if (!this.running) return false;
        var stmt = this.getStatement();
        // end of program?
        if (!stmt) {
            this.running = false;
            this.exited = true;
            return false;
        }
        if (this.trace) console.log(this.curpc, stmt, this.vars, Object.keys(this.arrays));
        // skip to next statment
        this.curpc++;
        // compile (unless cached) and execute statement
        this.executeStatement(stmt);
        return this.running;
    }

    compileStatement(stmt: basic.Statement & CompiledStatement) {
        if (stmt.$run == null) {
            try {
                var stmtfn = this['do__' + stmt.command];
                if (stmtfn == null) this.runtimeError(`I don't know how to "${stmt.command}".`);
                var functext = stmtfn.bind(this)(stmt);
                if (this.trace) console.log(functext);
                stmt.$run = this.compileJS(functext);
            } catch (e) {
                if (functext) console.log(functext);
                throw e;
            }
        }
    }
    compileJS(functext: string) : () => void {
        return new Function(functext).bind(this);
    }
    executeStatement(stmt: basic.Statement & CompiledStatement) {
        // compile (unless cached)
        this.compileStatement(stmt);
        // run compiled statement
        stmt.$run();
    }

    // TODO: this only works because each line has a label
    skipToEOL() {
        do {
            this.curpc++;
        } while (this.curpc < this.allstmts.length && !this.pc2label.get(this.curpc));
    }

    skipToElse() {
        while (this.curpc < this.allstmts.length) {
            // in Altair BASIC, ELSE is bound to the right-most IF
            // TODO: this is complicated, we should just have nested expressions
            var cmd = this.allstmts[this.curpc].command;
            if (cmd == 'ELSE') { this.curpc++; break; }
            else if (cmd == 'IF') return this.skipToEOL();
            this.curpc++;
            if (this.pc2label.get(this.curpc))
                break;
        }
    }

    skipToEOF() {
        this.curpc = this.allstmts.length;
    }

    skipToAfterNext(forname: string) : void {
        var pc = this.curpc;
        while (pc < this.allstmts.length) {
            var stmt = this.allstmts[pc];
            if (stmt.command == 'NEXT') {
                var nextlexpr = (stmt as basic.NEXT_Statement).lexpr;
                if (nextlexpr && nextlexpr.name == forname) {
                    this.curpc = pc + 1;
                    return;
                }
            }
            pc++;
        }
        this.runtimeError(`I couldn't find a matching NEXT ${forname} to skip this for loop.`);
    }

    skipToAfterWend() {
        var pc = this.curpc - 1;
        var nesting = 0;
        while (pc < this.allstmts.length) {
            var stmt = this.allstmts[pc];
            //console.log(nesting, pc, stmt);
            if (stmt.command == 'WHILE') {
                nesting++;
            } else if (stmt.command == 'WEND') {
                nesting--;
                if (nesting == 0) {
                    this.curpc = pc + 1;
                    return;
                }
            }
            pc++;
        }
        this.runtimeError(`I couldn't find a matching WEND for this WHILE.`);
    }

    gotoLabel(label) {
        var pc = this.label2pc[label];
        if (pc >= 0) {
            this.curpc = pc;
        } else {
            this.runtimeError(`I tried to go to the label "${label}" but couldn't find it.`);
        }
    }

    newLocalScope() {
        this.vars = Object.create(this.vars);
    }
    
    popLocalScope() {
        if (this.vars !== this.globals)
            this.vars = Object.getPrototypeOf(this.vars);
    }

    gosubLabel(label) {
        if (this.returnStack.length > 32767) // TODO: const?
            this.runtimeError(`I did too many GOSUBs without a RETURN.`)
        this.returnStack.push(this.curpc);
        this.gotoLabel(label);
    }

    returnFromGosub() {
        if (this.returnStack.length == 0)
            this.runtimeError("I tried to RETURN, but there wasn't a corresponding GOSUB."); // RETURN BEFORE GOSUB
        var pc = this.returnStack.pop();
        this.curpc = pc;
        this.popLocalScope();
    }

    popReturnStack() {
        if (this.returnStack.length == 0)
            this.runtimeError("I tried to POP, but there wasn't a corresponding GOSUB.");
        this.returnStack.pop();
    }

    valueToString(obj:basic.Value, padding:boolean) : string {
        var str;
        if (typeof obj === 'number') {
            var numstr = this.float2str(obj, this.opts.printZoneLength - 4);
            if (!padding)
                return numstr;
            else if (numstr.startsWith('-'))
                return `${numstr} `;
            else
                return ` ${numstr} `;
        } else if (obj == '\n') {
            this.column = 0;
            str = obj;
        } else if (obj == '\t') {
            var l = this.opts.printZoneLength;
            var curgroup = Math.floor(this.column / l);
            var nextcol = (curgroup + 1) * this.opts.printZoneLength;
            if (nextcol+l > this.margin) { this.column = 0; str = "\n"; } // return to left margin
            else str = this.TAB(nextcol); // next column
        } else {
            str = `${obj}`;
        }
        return str;
    }

    float2str(arg: number, numlen: number) : string {
        var numstr = arg.toString().toUpperCase();
        if (numlen > 0) {
            var prec = numlen;
            while (numstr.length > numlen) {
                numstr = arg.toPrecision(prec--);
            }
            if (numstr.startsWith('0.'))
                numstr = numstr.substr(1);
            else if (numstr.startsWith('-0.'))
                numstr = '-'+numstr.substr(2);
        }
        return numstr;
    }

    printExpr(obj) {
        var str = this.valueToString(obj, this.opts.numericPadding);
        this.column += str.length;
        this.print(str);
    }

    // override this
    print(str: string) {
        console.log(str);
    }

    // override this
    async input(prompt: string, nargs: number) : Promise<InputResponse> {
        return {line:"", vals:[]};
    }

    // override this
    resume() { }

    expr2js(expr: basic.Expr, opts?: ExprOptions) : string {
        if (!opts) opts = {};
        if (isLiteral(expr)) {
            return JSON.stringify(expr.value);
        } else if (isLookup(expr)) {
            if (!expr.args && opts.locals && opts.locals.indexOf(expr.name) >= 0) {
                return expr.name; // local arg in DEF
            } else {
                if (opts.isconst)
                    this.runtimeError(`I expected a constant value here.`); // TODO: check at compile-time?
                var s = '';
                var qname = JSON.stringify(expr.name);
                let jsargs = expr.args ? expr.args.map((arg) => this.expr2js(arg, opts)).join(', ') : [];
                if (expr.name.startsWith("FN")) { // is it a user-defined function?
                    // TODO: check argument count?
                    s += `this.getDef(${qname})(${jsargs})`;
                    // TODO: detect recursion?
                } else if (this.builtins[expr.name]) { // is it a built-in function?
                    this.checkFuncArgs(expr, this.builtins[expr.name]);
                    s += `this.builtins.${expr.name}(${jsargs})`;
                } else if (expr.args) {
                    // get array slice (HP BASIC)
                    if (this.opts.arraysContainChars && expr.name.endsWith('$'))
                        s += `this.getStringSlice(this.vars.${expr.name}, ${jsargs})`;
                    else
                        s += `this.arrayGet(${qname}, ${jsargs})`;
                } else { // just a variable
                    s += `this.vars.${expr.name}`;
                }
                return opts.novalid ? s : `this.checkValue(${s}, ${qname})`;
            }
        } else if (isBinOp(expr)) {
            var left = this.expr2js(expr.left, opts);
            var right = this.expr2js(expr.right, opts);
            return `this.${expr.op}(${left}, ${right})`;
        } else if (isUnOp(expr)) {
            var e = this.expr2js(expr.expr, opts);
            return `this.${expr.op}(${e})`;
        }
    }

    assign2js(expr: basic.IndOp, opts?: ExprOptions) : string {
        if (!opts) opts = {};
        var s = '';
         // is it a function? not allowed
        if (expr.name.startsWith("FN") || this.builtins[expr.name])
            this.runtimeError(`I can't call a function here.`);
        // is it a subscript?
        if (expr.args) {
            // TODO: set array slice (HP BASIC)
            if (this.opts.arraysContainChars && expr.name.endsWith('$')) {
                this.runtimeError(`I can't set array slices via this command yet.`);
            } else {
                s += this.array2js(expr, opts);
            }
        } else { // just a variable
            s = `this.globals.${expr.name}`;
        }
        return s;
    }

    array2js(expr: basic.IndOp, opts?: ExprOptions) : string {
        var qname = JSON.stringify(expr.name);
        var args = expr.args || [];
        return this.expr2js(expr, {novalid:true}) // check array bounds
             + `;this.getArray(${qname}, ${args.length})`
             + args.map((arg) => '[this.ROUND('+this.expr2js(arg, opts)+')]').join('');
    }

    checkFuncArgs(expr: basic.IndOp, fn: Function) {
        // TODO: check types?
        var nargs = expr.args ? expr.args.length : 0;
        // exceptions
        if (expr.name == 'RND' && nargs == 0) return;
        if (expr.name == 'MID$' && nargs == 2) return;
        if (expr.name == 'INSTR' && nargs == 2) return;
        if (fn.length != nargs)
            this.runtimeError(`I expected ${fn.length} arguments for the ${expr.name} function, but I got ${nargs}.`);
    }

    startForLoop(forname:string, init:number, targ:number, step?:number, endpc?:number) {
        // save start PC and label in case of hot reload (only works if FOR is first stmt in line)
        var looppc = this.curpc - 1;
        var looplabel = this.pc2label.get(looppc);
        if (!step) step = 1;
        this.vars[forname] = init;
        if (this.trace) console.log(`FOR ${forname} = ${init} TO ${targ} STEP ${step}`);
        // create done function
        var loopdone = () => {
            return step >= 0 ? this.vars[forname] > targ : this.vars[forname] < targ;
        }
        // skip entire for loop before first iteration? (Minimal BASIC)
        if (this.opts.testInitialFor && loopdone()) {
            if (endpc != null)
                this.curpc = endpc+1;
            else
                this.skipToAfterNext(forname);
        }
        // save for var name on stack, remove existing entry
        if (this.forLoopStack[forname] != null)
            this.forLoopStack = this.forLoopStack.filter((n) => n == forname);
        this.forLoopStack.push(forname);
        // create for loop record
        this.forLoops[forname] = {
            $next: (nextname:string) => {
                if (nextname && forname != nextname)
                    this.runtimeError(`I executed NEXT "${nextname}", but the last FOR was for "${forname}".`)
                this.vars[forname] += step;
                var done = loopdone();
                if (done) {
                    // delete entry, pop FOR off the stack and continue
                    this.forLoopStack.pop();
                    delete this.forLoops[forname];
                } else {
                    // go back to FOR loop, adjusting for hot reload (fetch pc by label)
                    this.curpc = ((looplabel != null && this.label2pc[looplabel]) || looppc) + 1;
                }
                if (this.trace) console.log(`NEXT ${forname}: ${this.vars[forname]} TO ${targ} STEP ${step} DONE=${done}`);
            }
        };
    }

    nextForLoop(name) {
        // get FOR loop entry, or get top of stack if NEXT var is optional 
        var fl = this.forLoops[name || (this.opts.optionalNextVar && this.forLoopStack[this.forLoopStack.length-1])];
        if (!fl) this.runtimeError(`I couldn't find a matching FOR for this NEXT.`)
        fl.$next(name);
    }

    whileLoop(cond) {
        if (cond) {
            this.whileLoops.push(this.curpc-1);
        } else {
            this.skipToAfterWend();
        }
    }

    nextWhileLoop() {
        var pc = this.whileLoops.pop();
        if (pc == null) this.runtimeError(`I couldn't find a matching WHILE for this WEND.`);
        else this.curpc = pc;
    }

    // converts a variable to string/number based on var name
    assign(name: string, right: number|string, isRead?:boolean) : number|string {
        // convert data? READ always converts if read into string
        if (isRead && name.endsWith("$"))
            return this.checkValue(this.convert(name, right), name);
        // TODO: use options
        if (name.endsWith("$")) {
            return this.convertToString(right, name);
        } else {
            return this.convertToNumber(right, name);
        }
    }

    convert(name: string, right: number|string) : number|string {
        if (name.endsWith("$")) {
            return right == null ? "" : right.toString();
        } else if (typeof right === 'number') {
            return right;
        } else {
            return parseFloat(right+"");
        }
    }

    convertToString(right: number|string, name?: string) {
        if (typeof right !== 'string') this.runtimeError(`I can't convert ${right} to a string.`);
        else return right;
    }

    convertToNumber(right: number|string, name?: string) {
        if (typeof right !== 'number') this.runtimeError(`I can't convert ${right} to a number.`);
        else return this.checkNum(right);
    }

    // dimension array
    dimArray(name: string, ...dims:number[]) {
        // TODO: maybe do this check at compile-time?
        dims = dims.map(Math.round);
        if (this.arrays[name] != null) {
            if (this.opts.staticArrays) return;
            else this.runtimeError(`I already dimensioned this array (${name}) earlier.`)
        }
        var total = this.getTotalArrayLength(dims);
        if (total > this.opts.maxArrayElements)
            this.runtimeError(`I can't create an array with this many elements.`);
        var isstring = name.endsWith('$');
        // if numeric value, we use Float64Array which inits to 0
        var arrcons = isstring ? Array : Float64Array;
        if (dims.length == 1) {
            this.arrays[name] = new arrcons(dims[0]+1);
        } else if (dims.length == 2) {
            this.arrays[name] = new Array(dims[0]+1);
            for (var i=0; i<dims[0]+1; i++) {
                this.arrays[name][i] = new arrcons(dims[1]+1);
            }
        } else {
            this.runtimeError(`I only support arrays of one or two dimensions.`)
        }
    }

    getTotalArrayLength(dims:number[]) {
        var n = 1;
        for (var i=0; i<dims.length; i++) {
            if (dims[i] < this.opts.defaultArrayBase)
                this.runtimeError(`I can't create an array with a dimension less than ${this.opts.defaultArrayBase}.`);
            n *= dims[i];
        }
        return n;
    }

    getArray(name: string, order: number) : [] {
        if (!this.arrays[name]) {
            if (this.opts.defaultArraySize == 0)
                this.dialectError(`automatically declare arrays without a DIM statement (or did you mean to call a function?)`);
            if (order == 1)
                this.dimArray(name, this.opts.defaultArraySize-1);
            else if (order == 2)
                this.dimArray(name, this.opts.defaultArraySize-1, this.opts.defaultArraySize-1);
            else
                this.runtimeError(`I only support arrays of one or two dimensions.`); // TODO
        }
        return this.arrays[name];
    }

    arrayGet(name: string, ...indices: number[]) : basic.Value {
        var arr = this.getArray(name, indices.length);
        indices = indices.map(this.ROUND.bind(this));
        var v = arr;
        for (var i=0; i<indices.length; i++) {
            var idx = indices[i];
            if (!isArray(v))
                this.runtimeError(`I tried to lookup ${name}(${indices}) but used too many dimensions.`);
            if (idx < this.opts.defaultArrayBase)
                this.runtimeError(`I tried to lookup ${name}(${indices}) but an index was less than ${this.opts.defaultArrayBase}.`);
            if (idx >= v.length) // TODO: also can happen when mispelling function name
                this.runtimeError(`I tried to lookup ${name}(${indices}) but it exceeded the dimensions of the array.`);
            v = v[indices[i]];
        }
        if (isArray(v)) // i.e. is an array?
            this.runtimeError(`I tried to lookup ${name}(${indices}) but used too few dimensions.`);
        return (v as any) as basic.Value;
    }

    // for HP BASIC string slicing (TODO?)
    modifyStringSlice(orig: string, add: string, start: number, end: number) : string {
        orig = orig || "";
        this.checkString(orig);
        this.checkString(add);
        if (!end) end = start;
        start = this.ROUND(start);
        end = this.ROUND(end);
        if (start < 1) this.dialectError(`accept a string slice index less than 1`);
        if (end < start) this.dialectError(`accept a string slice index less than the starting index`);
        return (orig + ' '.repeat(start)).substr(0, start-1) + add.substr(0, end+1-start) + orig.substr(end);
    }

    getStringSlice(s: string, start: number, end: number) {
        s = this.checkString(s);
        start = this.ROUND(start);
        if (start < 1) this.dialectError(`accept a string slice index less than 1`);
        if (end != null) {
            end = this.ROUND(end);
            if (end < start) this.dialectError(`accept a string slice index less than the starting index`);
            return s.substr(start-1, end+1-start);
        } else {
            return s.substr(start-1);
        }
    }

    checkOnGoto(value: number, labels: string[]) {
        value = this.ROUND(value);
        if (value < 0) // > 255 ?
            this.runtimeError(`I needed a number between 1 and ${labels.length}, but I got ${value}.`);
        if (this.opts.checkOnGotoIndex && (value < 1 || value > labels.length))
            this.runtimeError(`I needed a number between 1 and ${labels.length}, but I got ${value}.`);
        if (value < 1 || value > labels.length)
            return 0;
        return value;
    }
    
    onGotoLabel(value: number, ...labels: string[]) {
        value = this.checkOnGoto(value, labels);
        if (value) this.gotoLabel(labels[value-1]);
    }
    onGosubLabel(value: number, ...labels: string[]) {
        value = this.checkOnGoto(value, labels);
        if (value) this.gosubLabel(labels[value-1]);
    }

    nextDatum() : basic.Value {
        if (this.dataptr >= this.datums.length)
            this.runtimeError("I tried to READ, but ran out of data.");
        return this.datums[this.dataptr++].value;
    }

    //// STATEMENTS

    do__PRINT(stmt : basic.PRINT_Statement) {
        var s = '';
        for (var arg of stmt.args) {
            var expr = this.expr2js(arg);
            var name = (expr as any).name;
            s += `this.printExpr(this.checkValue(${expr}, ${JSON.stringify(name)}));`;
        }
        return s;
    }

    preInput() {
        this.running=false;
        this.curpc--;
    }

    postInput(valid : boolean) {
        if (valid) this.curpc++;
        this.running = true;
        this.resume();
    }

    do__INPUT(stmt : basic.INPUT_Statement) {
        var prompt = stmt.prompt != null ? this.expr2js(stmt.prompt) : '""';
        var elapsed = stmt.elapsed != null ? this.assign2js(stmt.elapsed) : "let ___";
        var setvals = '';
        stmt.args.forEach((arg, index) => {
            var lexpr = this.assign2js(arg);
            setvals += `
            var value = this.convert(${JSON.stringify(arg.name)}, response.vals[${index}]);
            valid &= this.isValid(value);
            ${lexpr} = value;
            `
        });
        return `this.preInput();
                this.input(${prompt}, ${stmt.args.length}).then((response) => {
                    let valid = 1;
                    ${setvals}
                    this.postInput(valid);
                    this.column = 0; // assume linefeed
                    ${elapsed} = response.elapsed;
                })`;
    }

    do__LET(stmt : basic.LET_Statement) {
        var right = this.expr2js(stmt.right);
        var s = `let _right = ${right};`;
        for (var lexpr of stmt.lexprs) {
            // HP BASIC string-slice syntax?
            if (this.opts.arraysContainChars && lexpr.args && lexpr.name.endsWith('$')) {
                s += `this.globals.${lexpr.name} = this.modifyStringSlice(this.vars.${lexpr.name}, _right, `
                s += lexpr.args.map((arg) => this.expr2js(arg)).join(', ');
                s += ');';
            } else {
                var ljs = this.assign2js(lexpr);
                s += `${ljs} = this.assign(${JSON.stringify(lexpr.name)}, _right);`;
            }
        }
        return s;
    }

    do__FOR(stmt : basic.FOR_Statement) {
        var name = JSON.stringify(stmt.lexpr.name);
        var init = this.expr2js(stmt.initial);
        var targ = this.expr2js(stmt.target);
        var step = stmt.step ? this.expr2js(stmt.step) : 'null';
        return `this.startForLoop(${name}, ${init}, ${targ}, ${step}, ${stmt.endpc})`;
    }

    do__NEXT(stmt : basic.NEXT_Statement) {
        var name = stmt.lexpr && JSON.stringify(stmt.lexpr.name);
        return `this.nextForLoop(${name})`;
    }

    do__IF(stmt : basic.IF_Statement) {
        var cond = this.expr2js(stmt.cond);
        if (stmt.endpc != null)
            return `if (!(${cond})) { this.curpc = ${stmt.endpc}; }`
        else
            return `if (!(${cond})) { this.skipToElse(); }`
    }

    do__ELSE(stmt : basic.ELSE_Statement) {
        if (stmt.endpc != null)
            return `this.curpc = ${stmt.endpc}`
        else
            return `this.skipToEOL()`
    }

    do__WHILE(stmt : basic.WHILE_Statement) {
        var cond = this.expr2js(stmt.cond);
        if (stmt.endpc != null)
            return `if (!(${cond})) { this.curpc = ${stmt.endpc+1}; }`;
        else
            return `this.whileLoop(${cond})`;
    }

    do__WEND(stmt : basic.WEND_Statement) {
        if (stmt.startpc != null)
            return `this.curpc = ${stmt.startpc}`;
        else
            return `this.nextWhileLoop()`
    }

    do__DEF(stmt : basic.DEF_Statement) {
        var args = [];
        for (var arg of stmt.lexpr.args || []) {
            if (isLookup(arg)) {
                args.push(arg.name);
            } else {
                this.runtimeError("I found a DEF statement with arguments other than variable names.");
            }
        }
        var functext = this.expr2js(stmt.def, {locals:args});
        //this.defs[stmt.lexpr.name] = new Function(args.join(','), functext).bind(this);
        return `this.defs.${stmt.lexpr.name} = function(${args.join(',')}) { return ${functext}; }.bind(this)`;
    }

    _DIM(dim : basic.IndOp) {
        // HP BASIC doesn't really have string arrays, just strings
        if (this.opts.arraysContainChars && dim.name.endsWith('$'))
            return '';
        // dimension an array
        var argsstr = '';
        for (var arg of dim.args) {
            argsstr += ', ' + this.expr2js(arg, {isconst: this.opts.staticArrays});
        }
        return `this.dimArray(${JSON.stringify(dim.name)}${argsstr});`;
    }

    do__DIM(stmt : basic.DIM_Statement) {
        if (this.opts.staticArrays) return; // DIM at reset()
        var s = '';
        stmt.args.forEach((dim) => s += this._DIM(dim));
        return s;
    }

    do__GOTO(stmt : basic.GOTO_Statement) {
        var label = this.expr2js(stmt.label);
        return `this.gotoLabel(${label})`;
    }

    do__GOSUB(stmt : basic.GOSUB_Statement) {
        var label = this.expr2js(stmt.label);
        return `this.gosubLabel(${label})`;
    }

    do__RETURN(stmt : basic.RETURN_Statement) {
        return `this.returnFromGosub()`;
    }

    do__ONGOTO(stmt : basic.ONGO_Statement) {
        var expr = this.expr2js(stmt.expr);
        var labels = stmt.labels.map((arg) => this.expr2js(arg, {isconst:true})).join(', ');
        if (stmt.command == 'ONGOTO')
            return `this.onGotoLabel(${expr}, ${labels})`;
        else
            return `this.onGosubLabel(${expr}, ${labels})`;
    }

    do__ONGOSUB(stmt : basic.ONGO_Statement) {
        return this.do__ONGOTO(stmt);
    }

    do__DATA() {
        // data is preprocessed
    }

    do__READ(stmt : basic.READ_Statement) {
        var s = '';
        stmt.args.forEach((arg) => {
            s += `${this.assign2js(arg)} = this.assign(${JSON.stringify(arg.name)}, this.nextDatum(), true);`;
        });
        return s;
    }

    do__RESTORE(stmt : basic.RESTORE_Statement) {
        if (stmt.label != null)
            return `this.dataptr = this.label2dataptr[${this.expr2js(stmt.label, {isconst:true})}] || 0`;
        else
            return `this.dataptr = 0`;
    }

    do__END() {
        return `this.skipToEOF()`;
    }

    do__STOP() {
        return `this.skipToEOF()`;
    }

    do__OPTION(stmt: basic.OPTION_Statement) {
        // already parsed in compiler
    }

    do__POP() {
        return `this.popReturnStack()`;
    }

    do__GET(stmt : basic.GET_Statement) {
        var lexpr = this.assign2js(stmt.lexpr);
        // TODO: single key input
        return `this.preInput();
                this.input().then((vals) => {
                    ${lexpr} = this.convert(${JSON.stringify(stmt.lexpr.name)}, vals[0]);
                    this.postInput(true);
                })`;
    }

    do__CLEAR() {
        return 'this.clearVars()';
    }

    do__RANDOMIZE() {
        return `this.rng.randomize()`;
    }

    do__CHANGE(stmt : basic.CHANGE_Statement) {
        var arr2str = stmt.dest.name.endsWith('$');
        if (arr2str) { // array -> string
            let arrname = (stmt.src as basic.IndOp).name || this.runtimeError("I can only change to a string from an array.");
            let dest = this.assign2js(stmt.dest);
            return `
            let arrname = ${JSON.stringify(arrname)};
            let len = this.arrayGet(arrname, 0);
            let s = '';
            for (let i=0; i<len; i++) {
                s += String.fromCharCode(this.arrayGet(arrname, i+1));
            }
            ${dest} = s;
            `;
        } else { // string -> array
            let src = this.expr2js(stmt.src);
            let dest = this.array2js(stmt.dest);
            return `
            let src = ${src}+"";
            ${dest}[0] = src.length;
            for (let i=0; i<src.length; i++) {
                ${dest}[i+1] = src.charCodeAt(i);
            }
            `;
        }
    }

    do__CONVERT(stmt : basic.CONVERT_Statement) {
        var num2str = stmt.dest.name.endsWith('$');
        let src = this.expr2js(stmt.src);
        let dest = this.assign2js(stmt.dest);
        if (num2str) {
            return `${dest} = this.valueToString(${src}, false)`;
        } else {
            return `${dest} = this.VAL(${src})`;
        }
    }

    do__SUB(stmt: basic.SUB_Statement) {
        this.subroutines[stmt.lexpr.name] = stmt;
        // skip the SUB definition
        return `this.curpc = ${stmt.endpc}`
    }

    do__CALL(stmt: basic.CALL_Statement) {
        var substmt : basic.SUB_Statement = this.subroutines[stmt.call.name];
        if (substmt == null)
            this.runtimeError(`I can't find a subroutine named "${stmt.call.name}".`);
        var subargs = substmt.lexpr.args || [];
        var callargs = stmt.call.args || [];
        if (subargs.length != callargs.length)
            this.runtimeError(`I tried to call ${stmt.call.name} with the wrong number of parameters.`);
        var s = '';
        s += `this.gosubLabel(${JSON.stringify(stmt.call.name)});`
        s += `this.newLocalScope();`
        for (var i=0; i<subargs.length; i++) {
            var arg = subargs[i] as basic.IndOp;
            s += `this.vars.${arg.name} = ${this.expr2js(callargs[i])};`
        }
        return s;
    }

    // TODO: ONERR, ON ERROR GOTO
    // TODO: memory quota
    // TODO: useless loop (! 4th edition)
    // TODO: other 4th edition errors
    // TODO: ecma55 all-or-none input checking?

    // FUNCTIONS

    isValid(obj:number|string) : boolean {
        if (typeof obj === 'number' && !isNaN(obj) && (!this.opts.checkOverflow || isFinite(obj)))
            return true;
        else if (typeof obj === 'string')
            return true;
        else
            return false;
    }
    checkValue(obj:number|string, exprname:string) : number|string {
        // check for unreferenced value
        if (typeof obj !== 'number' && typeof obj !== 'string') {
            // assign default value?
            if (obj == null && this.opts.defaultValues) {
                return exprname.endsWith("$") ? "" : 0;
            }
            if (exprname != null && obj == null) {
                this.runtimeError(`I haven't assigned a value to ${exprname}.`);
            } else if (exprname != null) {
                this.runtimeError(`I got an invalid value for ${exprname}: ${obj}`);
            } else {
                this.runtimeError(`I got an invalid value: ${obj}`);
            }
        }
        return obj;
    }
    getDef(exprname: string) {
        var fn = this.defs[exprname];
        if (!fn) this.runtimeError(`I haven't run a DEF statement for ${exprname}.`);
        return fn;
    }
    checkNum(n:number) : number {
        this.checkValue(n, 'this');
        if (n === Infinity) this.runtimeError(`I computed a number too big to store.`);
        if (isNaN(n)) this.runtimeError(`I computed an invalid number.`);
        return n;
    }
    checkString(s:string) : string {
        this.checkValue(s, 'this');
        if (typeof s !== 'string')
            this.runtimeError(`I expected a string here.`);
        else if (s.length > this.opts.maxStringLength)
            this.dialectError(`create strings longer than ${this.opts.maxStringLength} characters`);
        return s;
    }
    
    add(a, b) : number|string {
        // TODO: if string-concat
        if (typeof a === 'number' && typeof b === 'number')
            return this.checkNum(a + b);
        else if (this.opts.stringConcat)
            return this.checkString(a + b);
        else
            this.dialectError(`use the "+" operator to concatenate strings`)
    }
    sub(a:number, b:number) : number {
        return this.checkNum(a - b);
    }
    mul(a:number, b:number) : number {
        return this.checkNum(a * b);
    }
    div(a:number, b:number) : number {
        if (b == 0) this.runtimeError(`I can't divide by zero.`);
        return this.checkNum(a / b);
    }
    idiv(a:number, b:number) : number {
        return this.FIX(this.INT(a) / this.INT(b));
    }
    mod(a:number, b:number) : number {
        return this.checkNum(a % b);
    }
    pow(a:number, b:number) : number {
        if (a == 0 && b < 0) this.runtimeError(`I can't raise zero to a negative power.`);
        return this.checkNum(Math.pow(a, b));
    }
    band(a:number, b:number) : number {
        return a & b;
    }
    bor(a:number, b:number) : number {
        return a | b;
    }
    bnot(a:number) : number {
        return ~a;
    }
    bxor(a:number, b:number) : number {
        return a ^ b;
    }
    bimp(a:number, b:number) : number {
        return this.bor(this.bnot(a), b);
    }
    beqv(a:number, b:number) : number {
        return this.bnot(this.bxor(a, b));
    }
    land(a:number, b:number) : number {
        return a && b ? (this.opts.bitwiseLogic ? -1 : 1) : 0;
    }
    lor(a:number, b:number) : number {
        return a || b ? (this.opts.bitwiseLogic ? -1 : 1) : 0;
    }
    lnot(a:number) : number {
        return a ? 0 : (this.opts.bitwiseLogic ? -1 : 1);
    }
    neg(a:number) : number {
        return -a;
    }
    eq(a:number, b:number) : number {
        return a == b ? (this.opts.bitwiseLogic ? -1 : 1) : 0;
    }
    ne(a:number, b:number) : number {
        return a != b ? (this.opts.bitwiseLogic ? -1 : 1) : 0;
    }
    lt(a:number, b:number) : number {
        return a < b ? (this.opts.bitwiseLogic ? -1 : 1) : 0;
    }
    gt(a:number, b:number) : number {
        return a > b ? (this.opts.bitwiseLogic ? -1 : 1) : 0;
    }
    le(a:number, b:number) : number {
        return a <= b ? (this.opts.bitwiseLogic ? -1 : 1) : 0;
    }
    ge(a:number, b:number) : number {
        return a >= b ? (this.opts.bitwiseLogic ? -1 : 1) : 0;
    }
    min(a:number, b:number) : number {
        return a < b ? a : b;
    }
    max(a:number, b:number) : number {
        return a > b ? a : b;
    }

    // FUNCTIONS (uppercase)
    // TODO: swizzle names for type-checking

    ABS(arg : number) : number {
        return this.checkNum(Math.abs(arg));
    }
    ASC(arg : string) : number {
        arg = this.checkString(arg);
        if (arg == '') this.runtimeError(`I tried to call ASC() on an empty string.`);
        return arg.charCodeAt(0);
    }
    ATN(arg : number) : number {
        return this.checkNum(Math.atan(arg));
    }
    CHR$(arg : number) : string {
        return String.fromCharCode(this.checkNum(arg));
    }
    CINT(arg : number) : number {
        return this.ROUND(arg);
    }
    COS(arg : number) : number {
        return this.checkNum(Math.cos(arg));
    }
    COT(arg : number) : number {
        return this.checkNum(1.0 / Math.tan(arg)); // 4th edition only
    }
    CTL(arg : number) : string {
        return this.CHR$(arg);
    }
    EXP(arg : number) : number {
        return this.checkNum(Math.exp(arg));
    }
    FIX(arg : number) : number {
        return this.checkNum(arg < 0 ? Math.ceil(arg) : Math.floor(arg));
    }
    HEX$(arg : number) : string {
        return this.ROUND(arg).toString(16);
    }
    INSTR(a, b, c) : number {
        if (c != null) {
            return this.checkString(b).indexOf(this.checkString(c), this.checkNum(a) - 1) + 1;
        } else {
            return this.checkString(a).indexOf(this.checkString(b)) + 1;
        }
    }
    INT(arg : number) : number {
        return this.checkNum(Math.floor(arg));
    }
    LEFT$(arg : string, count : number) : string {
        arg = this.checkString(arg);
        count = this.ROUND(count);
        return arg.substr(0, count);
    }
    LEN(arg : string) : number {
        return this.checkString(arg).length;
    }
    LIN(arg : number) : string {
        return this.STRING$(arg, '\n');
    }
    LOG(arg : number) : number {
        if (arg == 0) this.runtimeError(`I can't take the logarithm of zero (${arg}).`)
        if (arg < 0) this.runtimeError(`I can't take the logarithm of a negative number (${arg}).`)
        return this.checkNum(Math.log(arg));
    }
    LOG10(arg : number) : number {
        if (arg == 0) this.runtimeError(`I can't take the logarithm of zero (${arg}).`)
        if (arg < 0) this.runtimeError(`I can't take the logarithm of a negative number (${arg}).`)
        return this.checkNum(Math.log10(arg));
    }
    MID$(arg : string, start : number, count : number) : string {
        arg = this.checkString(arg);
        if (!count) count = arg.length;
        start = this.ROUND(start);
        count = this.ROUND(count);
        if (start < 1) this.runtimeError(`I can't compute MID$ if the starting index is less than 1.`)
        return arg.substr(start-1, count);
    }
    OCT$(arg : number) : string {
        return this.ROUND(arg).toString(8);
    }
    PI() : number {
        return Math.PI;
    }
    // TODO: POS(haystack, needle, start)
    POS(arg1, arg2) { // arg ignored
        if (typeof arg1 == 'string' && typeof arg2 == 'string')
            return arg1.indexOf(arg2) >= 0 + 1;
        else
            return this.column + 1;
    }
    RIGHT$(arg : string, count : number) : string {
        arg = this.checkString(arg);
        count = this.ROUND(count);
        return arg.substr(arg.length - count, count);
    }
    RND(arg : number) : number {
        // TODO: X<0 restart w/ seed, X=0 repeats
        if (arg < 0) this.rng.seedfloat(arg);
        return this.rng.next();
    }
    ROUND(arg : number) : number {
        return this.checkNum(Math.round(arg));
    }
    SGN(arg : number) : number {
        this.checkNum(arg);
        return (arg < 0) ? -1 : (arg > 0) ? 1 : 0;
    }
    SIN(arg : number) : number {
        return this.checkNum(Math.sin(arg));
    }
    SPACE$(arg : number) : string {
        return this.STRING$(arg, ' ');
    }
    SPC(arg : number) : string {
        return this.SPACE$(arg);
    }
    SQR(arg : number) : number {
        if (arg < 0) this.runtimeError(`I can't take the square root of a negative number (${arg}).`)
        return this.checkNum(Math.sqrt(arg));
    }
    STR$(arg : number) : string {
        return this.valueToString(this.checkNum(arg), false);
    }
    STRING$(len : number, chr : number|string) : string {
        len = this.ROUND(len);
        if (len <= 0) return '';
        if (len > this.opts.maxStringLength)
            this.dialectError(`create a string longer than ${this.opts.maxStringLength} characters`);
        if (typeof chr === 'string')
            return chr.substr(0,1).repeat(len);
        else
            return String.fromCharCode(chr).repeat(len);
    }
    TAB(arg : number) : string {
        if (arg < 1) { arg = 1; } // TODO: SYSTEM MESSAGE IDENTIFYING THE EXCEPTION
        var spaces = this.ROUND(arg) - 1 - this.column;
        return this.SPACE$(spaces);
    }
    TAN(arg : number) : number {
        return this.checkNum(Math.tan(arg));
    }
    TIM(arg : number) : number { // only HP BASIC?
        var d = new Date();
        switch (this.ROUND(arg)) {
            case 0: return d.getMinutes();
            case 1: return d.getHours();
            case 2:
                var dayCount = [0, 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334];
                var mn = d.getMonth();
                var dn = d.getDate();
                var dayOfYear = dayCount[mn] + dn;
                var isLeapYear = (d.getFullYear() & 3) == 0; // TODO: wrong
                if(mn > 1 && isLeapYear) dayOfYear++;
                return dayOfYear;
            case 3: return d.getFullYear() % 100; // Y@K!
            case 4: return d.getSeconds();
            default: return 0;
        }
    }
    TIMER() : number {
        return Date.now() / 1000;
    }
    UPS$(arg : string) : string {
        return this.checkString(arg).toUpperCase();
    }
    VAL(arg : string) : number {
        var n = parseFloat(this.checkString(arg));
        return isNaN(n) ? 0 : n; // TODO? altair works this way
    }
    LPAD$(arg : string, len : number) : string {
        arg = this.checkString(arg);
        while (arg.length < len) arg = " " + arg;
        return arg;
    }
    RPAD$(arg : string, len : number) : string {
        arg = this.checkString(arg);
        while (arg.length < len) arg = arg + " ";
        return arg;
    }
    NFORMAT$(arg : number, numlen : number) : string {
        var s = this.float2str(arg, numlen);
        return (numlen > 0) ? this.LPAD$(s, numlen) : this.RPAD$(s, -numlen);
    }
}
