This commit is contained in:
parent
f7be86a3d9
commit
31c5ed614c
27 changed files with 548 additions and 61 deletions
63
src/core/ToNumber.go
Normal file
63
src/core/ToNumber.go
Normal file
|
@ -0,0 +1,63 @@
|
|||
package core
|
||||
|
||||
import (
|
||||
"strconv"
|
||||
"strings"
|
||||
"unicode/utf8"
|
||||
|
||||
"git.urbach.dev/cli/q/src/errors"
|
||||
"git.urbach.dev/cli/q/src/fs"
|
||||
"git.urbach.dev/cli/q/src/token"
|
||||
)
|
||||
|
||||
// ToNumber tries to convert the token into a numeric value.
|
||||
func (f *Function) ToNumber(t token.Token) (int, error) {
|
||||
return ToNumber(t, f.File)
|
||||
}
|
||||
|
||||
// ToNumber tries to convert the token into a numeric value.
|
||||
func ToNumber(t token.Token, file *fs.File) (int, error) {
|
||||
switch t.Kind {
|
||||
case token.Number:
|
||||
var (
|
||||
digits = t.String(file.Bytes)
|
||||
number int64
|
||||
err error
|
||||
)
|
||||
|
||||
switch {
|
||||
case strings.HasPrefix(digits, "0x"):
|
||||
number, err = strconv.ParseInt(digits[2:], 16, 64)
|
||||
case strings.HasPrefix(digits, "0o"):
|
||||
number, err = strconv.ParseInt(digits[2:], 8, 64)
|
||||
case strings.HasPrefix(digits, "0b"):
|
||||
number, err = strconv.ParseInt(digits[2:], 2, 64)
|
||||
default:
|
||||
number, err = strconv.ParseInt(digits, 10, 64)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
return 0, errors.New(InvalidNumber, file, t.Position)
|
||||
}
|
||||
|
||||
return int(number), nil
|
||||
|
||||
case token.Rune:
|
||||
r := t.Bytes(file.Bytes)
|
||||
r = Unescape(r)
|
||||
|
||||
if len(r) == 0 {
|
||||
return 0, errors.New(InvalidRune, file, t.Position+1)
|
||||
}
|
||||
|
||||
number, size := utf8.DecodeRune(r)
|
||||
|
||||
if len(r) > size {
|
||||
return 0, errors.New(InvalidRune, file, t.Position+1)
|
||||
}
|
||||
|
||||
return int(number), nil
|
||||
}
|
||||
|
||||
return 0, errors.New(InvalidNumber, file, t.Position)
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue