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"
    "strings"
    "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 2 arguments, a string and a delimiter
 * returns a list of substrings found delimited by the delimiter from the parent string
 * Filters out all empty segments between delimiters
 *
 * Example: (split "/path/to/file" "/") -> ("path" "to" "file")
 */
func Split(in *ast.Token, vt ast.VarTable, ft ast.FuncTable) *ast.Token {
    in = in.Eval(ft, vt, false)
    if in == nil || in.Tag != ast.STRING || in.Next == nil || in.Next.Tag != ast.STRING {
        log.Log(log.ERR,
            "Args must be two strings",
            "split")
        return nil
    }

    body := in.Value()
    delim := in.Next.Value()
    if len(body) < len(delim) {
        log.Log(log.DEBUG,
            "possibly mismatched args" +
                "delimiter longer than body",
            "split")
    }

    var res *ast.Token
    builder := &res
    strtoks := strings.Split(body, delim)
    for _, i := range strtoks {
        if i == "" {
            continue
        }

        *builder = &ast.Token{Tag: ast.STRING}
        (*builder).Set(i)
        builder = &(*builder).Next
    }

    return res
}

/* Takes two args, a delimiter and a list of strings
 * Returns the list of strings concatenated together with the delimiter in between each element
 * On error returns nil
 *
 * Example: (join ", " ("apple" "ananas" "pear")) -> "apple, ananas, pear"
 */
func Join(in *ast.Token, vt ast.VarTable, ft ast.FuncTable) *ast.Token {
    in = in.Eval(ft, vt, false)
    if in == nil || in.Next == nil {
        log.Log(log.ERR,
            "one or more arguments evaluated to nil",
            "join")
        return nil
    }

    delim := in
    strs := in.Next
    if delim.Tag != ast.STRING || strs.Tag != ast.LIST {
        log.Log(log.ERR,
            "first argument must be a string (delimiter) and second argument must be a list (strings)",
            "join")
        return nil
    }

    de := delim.Value()
    res := ""
    for i := strs.Expand(); i != nil; i = i.Next {
        if i.Tag != ast.STRING {
            log.Log(log.ERR,
                "all items to be joined must be strings",
                "join")
            return nil
        }

        res += i.Value()
        if i.Next != nil {
            res += de
        }
    }

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

/* 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
}