SHS/stdlib/string.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 stdlib

import (
    "fmt"
    "strconv"
    "gitlab.com/whom/shs/ast"
    "gitlab.com/whom/shs/log"
)

/* Concatenates N stringables
 *
 * Example: (concat "hello" " " "world")
 */
func Concat(in *ast.Token, vt ast.VarTable, ft ast.FuncTable) *ast.Token {
    in = in.Eval(ft, vt, false)

    var res string
    for i := in; i != nil; i = i.Next {
        if i.Tag == ast.LIST {
            log.Log(log.ERR, "Not concatenating list", "conc")
            log.Log(log.DEBUG, "Try using the expand operator (...)", "conc")
            continue
        }

        res += i.Value()
    }

    t := &ast.Token{Tag: ast.STRING}
    t.Set(res)
    return t
}

/* Takes 1 argument, returns its value as a string
 * works on lists too.
 *
 * Example: (string 1) -> 1.0
 */
func StrCast(in *ast.Token, vt ast.VarTable, ft ast.FuncTable) *ast.Token {
    body := in.Eval(ft, vt, false).String()
    res := &ast.Token{ Tag: ast.STRING }
    res.Set(body)
    return res
}

/* Takes one arg, returns nil
 * Prints a string to stdout
 * Unquotes string so user can add escaped chars like \n, \t, etc
 *
 * Example: (print "Line: \n, Tab: \t")
 */
func PrintStr(in *ast.Token, vt ast.VarTable, ft ast.FuncTable) *ast.Token {
    body := in.Eval(ft, vt, false).String()
    if body[0] != body[len(body)-1] && body[0] != '"' {
        body = "`" + body + "`"
    }

    text, err := strconv.Unquote(body)
    if err != nil {
        log.Log(log.ERR,
            "error unquoting string",
            "print")
        return nil
    }

    fmt.Printf(text + "\n")
    return nil
}