SHS/io/debug.go

/*   SHS: Syntactically Homogeneous Shell
 *   Copyright (C) 2019  Aidan Hahn
 *
 *   This program is free software: you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation, either version 3 of the License, or
 *   (at your option) any later version.
 *
 *   This program is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

package io

import (
    "strings"
    "fmt"
    "git.callpipe.com/aidan/shs/ast"
    "git.callpipe.com/aidan/shs/datatypes"
)

func FmtToken(arg *ast.Token) string {
    suffix := "->"
    if arg.Next == nil {
        suffix = ""
    }

    switch arg.Tag {
    case ast.LIST:
        return fmt.Sprintf("(%s, [List])%s", "LIST", suffix)

    default:
        return fmt.Sprintf("(%s, %s)%s", GetTagAsStr(arg.Tag), arg.Inner, suffix)
    }
}

func GetTagAsStr(tag ast.Token_t) string {
    switch tag {
    case ast.LIST:
        return "LIST"
    case ast.STRING:
        return "STRING"
    case ast.NUMBER:
        return "NUMBER"
    case ast.SYMBOL:
        return "SYMBOL"
    }
    return "UNKNOWN"
}

func PrintSExpression(arg *ast.Token) {
    if arg == nil {
        return //TODO: Handle error here?
    }

    var lists datatypes.TokenStack;
    lists.Push(arg)

loop:
    var constructor strings.Builder
    i := lists.Pop()
    if i == nil {
        return
    }

    for iter := i; iter != nil; iter = iter.Next {
        if iter.Tag == ast.LIST {
            lists.Push(iter.Inner.(*ast.Token))
        }

        constructor.WriteString(FmtToken(iter))
    }

    println(constructor.String())
    goto loop
}