aergoio/aergo

package context

import (
    "fmt"
    "strings"
)

func ParseFirstWord(input string) (string, string) {

    trimedStr := strings.TrimSpace(input) //remove whitespace
    splitedStr := strings.Fields(trimedStr)
    if len(splitedStr) == 0 {
        return "", ""
    } else if len(splitedStr) == 1 {
        return strings.ToLower(splitedStr[0]), ""
    }

    return strings.ToLower(splitedStr[0]), strings.Join(splitedStr[1:], " ")
}

type Chunk struct {
    Accent bool
    Text   string
}

func SplitSpaceAndAccent(input string, addLastInComplete bool) []Chunk {

    var ret []Chunk

    fieldSplit := strings.Fields(input)
    startAccent := -1

    for i, chunk := range fieldSplit {
        if startAccent == -1 {
            if strings.HasPrefix(chunk, "`") {
                if strings.Count(chunk, "`") > 1 && strings.HasSuffix(chunk, "`") {
                    // for example `keyword`
                    ret = append(ret, Chunk{Accent: true, Text: chunk[1 : len(chunk)-1]})
                } else {
                    // for example `white space`
                    startAccent = i
                }
            } else {
                // just normal keyword
                ret = append(ret, Chunk{Accent: false, Text: chunk})
            }
        } else if startAccent != -1 && strings.HasSuffix(chunk, "`") {

            //end of statement
            mergedStr := strings.Join(fieldSplit[startAccent:i+1], " ")
            ret = append(ret, Chunk{Accent: true, Text: mergedStr[1 : len(mergedStr)-1]})

            startAccent = -1 // reset
        }

        // contain last incomplete word
        if addLastInComplete && i+1 == len(fieldSplit) && -1 != startAccent {
            mergedStr := strings.Join(fieldSplit[startAccent:], " ")
            ret = append(ret, Chunk{Accent: true, Text: mergedStr[1:]})
        }
    }

    return ret
}

func IsCompleteCommand(line string, line_no int, isOpen bool) (bool, error) {

    chunks := strings.Fields(line)

    for _, chunk := range chunks {
        if isOpen {
            if chunk == "`" {
                isOpen = false
            } else if strings.HasPrefix(chunk, "`") {
                return false, fmt.Errorf("already open parameter at line %v", line_no)
            } else if strings.HasSuffix(chunk, "`") {
                isOpen = false
            }
        } else {
            if chunk == "`" {
                isOpen = true
            } else if strings.HasPrefix(chunk, "`") {
                if strings.HasSuffix(chunk, "`") {
                    // for example `keyword`
                } else {
                    // for example `white space`
                    isOpen = true
                }
            } else if strings.HasSuffix(chunk, "`") {
                return false, fmt.Errorf("closing not open parameter at line %v", line_no)
            }
        }
    }

    return isOpen, nil
}