Open-Argon/argon-v3
1
0
Fork 0
mirror of https://github.com/Open-Argon/argon-v3.git synced 2025-12-06 08:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

add more features

Browse Source
This commit is contained in:
William Bell
2023-07-17 20:52:42 +01:00
parent 425b73b681
commit aa21a53791
4 changed files with 338 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

145
src/buffer.go
View File

@@ -1,6 +1,7 @@
package main
import (
"bytes"
"fmt"
)
@@ -150,6 +151,150 @@ func ArBuffer(buf []byte) ArObject {
return obj, ArErr{}
},
}
obj.obj["splitN"] = builtinFunc{
"splitN",
func(a ...any) (any, ArErr) {
if len(a) != 2 {
return nil, ArErr{
TYPE: "TypeError",
message: "expected 1 argument, got " + fmt.Sprint(len(a)),
EXISTS: true,
}
}
splitVal := ArValidToAny(a[0])
if typeof(splitVal) != "buffer" {
return nil, ArErr{
TYPE: "TypeError",
message: "expected buffer, got " + typeof(splitVal),
EXISTS: true,
}
}
var separator = splitVal.([]byte)
nVal := ArValidToAny(a[1])
if typeof(nVal) != "number" {
return nil, ArErr{
TYPE: "TypeError",
message: "expected number, got " + typeof(nVal),
EXISTS: true,
}
}
nNum := nVal.(number)
if nNum.Denom().Cmp(one.Denom()) != 0 {
return nil, ArErr{
TYPE: "TypeError",
message: "expected integer, got " + fmt.Sprint(nNum),
EXISTS: true,
}
}
n := nNum.Num().Int64()
var result [][]byte
start := 0
var i int64
for i = 0; i < n; i++ {
index := bytes.Index(buf[start:], separator)
if index == -1 {
result = append(result, buf[start:])
break
}
end := start + index
result = append(result, buf[start:end])
start = end + len(separator)
}
if int64(len(result)) != n {
result = append(result, buf[start:])
}
var bufoutput = []any{}
for _, v := range result {
bufoutput = append(bufoutput, ArBuffer(v))
}
return ArArray(bufoutput), ArErr{}
},
}
obj.obj["split"] = builtinFunc{
"split",
func(a ...any) (any, ArErr) {
if len(a) != 1 {
return nil, ArErr{
TYPE: "TypeError",
message: "expected 1 argument, got " + fmt.Sprint(len(a)),
EXISTS: true,
}
}
splitVal := ArValidToAny(a[0])
if typeof(splitVal) != "buffer" {
return nil, ArErr{
TYPE: "TypeError",
message: "expected buffer, got " + typeof(splitVal),
EXISTS: true,
}
}
var separator = splitVal.([]byte)
var result [][]byte
start := 0
for {
index := bytes.Index(buf[start:], separator)
if index == -1 {
result = append(result, buf[start:])
break
}
end := start + index
result = append(result, buf[start:end])
start = end + len(separator)
}
var bufoutput = []any{}
for _, v := range result {
bufoutput = append(bufoutput, ArBuffer(v))
}
return ArArray(bufoutput), ArErr{}
},
}
obj.obj["slice"] = builtinFunc{
"slice",
func(a ...any) (any, ArErr) {
if len(a) != 2 {
return nil, ArErr{
TYPE: "TypeError",
message: "expected 2 arguments, got " + fmt.Sprint(len(a)),
EXISTS: true,
}
}
startVal := ArValidToAny(a[0])
if typeof(startVal) != "number" {
return nil, ArErr{
TYPE: "TypeError",
message: "expected number, got " + typeof(startVal),
EXISTS: true,
}
}
start := startVal.(number)
if start.Denom().Cmp(one.Denom()) != 0 {
return nil, ArErr{
TYPE: "TypeError",
message: "expected integer, got " + fmt.Sprint(start),
EXISTS: true,
}
}
endVal := ArValidToAny(a[1])
if typeof(endVal) != "number" {
return nil, ArErr{
TYPE: "TypeError",
message: "expected number, got " + typeof(endVal),
EXISTS: true,
}
}
end := endVal.(number)
if end.Denom().Cmp(one.Denom()) != 0 {
return nil, ArErr{
TYPE: "TypeError",
message: "expected integer, got " + fmt.Sprint(end),
EXISTS: true,
}
}
return ArBuffer(buf[floor(start).Num().Int64():floor(end).Num().Int64()]), ArErr{}
},
}
obj.obj["to"] = builtinFunc{
"to",
func(a ...any) (any, ArErr) {

29
src/file.go
View File

@@ -123,6 +123,13 @@ func ArRead(args ...any) (any, ArErr) {
}
return ArTimeClass(info.ModTime()), ArErr{}
}},
"close": builtinFunc{"close", func(...any) (any, ArErr) {
err := file.Close()
if err != nil {
return ArObject{}, ArErr{TYPE: "Runtime Error", message: err.Error(), EXISTS: true}
}
return nil, ArErr{}
}},
}), ArErr{}
}
@@ -148,7 +155,10 @@ func ArWrite(args ...any) (any, ArErr) {
return ArObject{}, ArErr{TYPE: "Runtime Error", message: "text takes a string not type '" + typeof(args[0]) + "'", EXISTS: true}
}
args[0] = ArValidToAny(args[0])
file.Write([]byte(args[0].(string)))
_, err = file.Write([]byte(args[0].(string)))
if err != nil {
return ArObject{}, ArErr{TYPE: "Runtime Error", message: err.Error(), EXISTS: true}
}
return nil, ArErr{}
}},
"buffer": builtinFunc{"buffer", func(args ...any) (any, ArErr) {
@@ -159,7 +169,10 @@ func ArWrite(args ...any) (any, ArErr) {
return ArObject{}, ArErr{TYPE: "Runtime Error", message: "buffer takes a buffer not type '" + typeof(args[0]) + "'", EXISTS: true}
}
args[0] = ArValidToAny(args[0])
file.Write(args[0].([]byte))
_, err = file.Write(args[0].([]byte))
if err != nil {
return ArObject{}, ArErr{TYPE: "Runtime Error", message: err.Error(), EXISTS: true}
}
return nil, ArErr{}
}},
"json": builtinFunc{"json", func(args ...any) (any, ArErr) {
@@ -170,7 +183,17 @@ func ArWrite(args ...any) (any, ArErr) {
if err != nil {
return ArObject{}, ArErr{TYPE: "Runtime Error", message: err.Error(), EXISTS: true}
}
file.Write([]byte(jsonstr))
_, err = file.Write([]byte(jsonstr))
if err != nil {
return ArObject{}, ArErr{TYPE: "Runtime Error", message: err.Error(), EXISTS: true}
}
return nil, ArErr{}
}},
"close": builtinFunc{"close", func(...any) (any, ArErr) {
err := file.Close()
if err != nil {
return ArObject{}, ArErr{TYPE: "Runtime Error", message: err.Error(), EXISTS: true}
}
return nil, ArErr{}
}},
}), ArErr{}

146
src/path.go
View File

@@ -41,6 +41,152 @@ var ArPath = Map(
}
return ret, ArErr{}
}},
"exists": builtinFunc{
"exists",
func(args ...any) (any, ArErr) {
if len(args) != 1 {
return nil, ArErr{
TYPE: "runtime",
message: "exists takes exactly 1 argument, got " + fmt.Sprint(len(args)),
EXISTS: true,
}
}
args[0] = ArValidToAny(args[0])
if typeof(args[0]) != "string" {
return nil, ArErr{
TYPE: "runtime",
message: "exists argument must be a string, got " + typeof(args[0]),
EXISTS: true,
}
}
_, err := os.Stat(args[0].(string))
if err != nil {
if os.IsNotExist(err) {
return false, ArErr{}
}
return nil, ArErr{
TYPE: "runtime",
message: err.Error(),
EXISTS: true,
}
}
return true, ArErr{}
}},
"mkAllDir": builtinFunc{
"mkAllDir",
func(args ...any) (any, ArErr) {
if len(args) != 1 {
return nil, ArErr{
TYPE: "runtime",
message: "mkAllDir takes exactly 1 argument, got " + fmt.Sprint(len(args)),
EXISTS: true,
}
}
args[0] = ArValidToAny(args[0])
if typeof(args[0]) != "string" {
return nil, ArErr{
TYPE: "runtime",
message: "mkAllDir argument must be a string, got " + typeof(args[0]),
EXISTS: true,
}
}
err := os.MkdirAll(args[0].(string), os.ModePerm)
if err != nil {
return nil, ArErr{
TYPE: "runtime",
message: err.Error(),
EXISTS: true,
}
}
return nil, ArErr{}
}},
"mkDir": builtinFunc{
"mkDir",
func(args ...any) (any, ArErr) {
if len(args) != 1 {
return nil, ArErr{
TYPE: "runtime",
message: "mkDir takes exactly 1 argument, got " + fmt.Sprint(len(args)),
EXISTS: true,
}
}
args[0] = ArValidToAny(args[0])
if typeof(args[0]) != "string" {
return nil, ArErr{
TYPE: "runtime",
message: "mkDir argument must be a string, got " + typeof(args[0]),
EXISTS: true,
}
}
err := os.Mkdir(args[0].(string), os.ModePerm)
if err != nil {
return nil, ArErr{
TYPE: "runtime",
message: err.Error(),
EXISTS: true,
}
}
return nil, ArErr{}
}},
"remove": builtinFunc{
"remove",
func(args ...any) (any, ArErr) {
if len(args) != 1 {
return nil, ArErr{
TYPE: "runtime",
message: "remove takes exactly 1 argument, got " + fmt.Sprint(len(args)),
EXISTS: true,
}
}
args[0] = ArValidToAny(args[0])
if typeof(args[0]) != "string" {
return nil, ArErr{
TYPE: "runtime",
message: "remove argument must be a string, got " + typeof(args[0]),
EXISTS: true,
}
}
err := os.Remove(args[0].(string))
if err != nil {
return nil, ArErr{
TYPE: "runtime",
message: err.Error(),
EXISTS: true,
}
}
return nil, ArErr{}
}},
"isDir": builtinFunc{
"isDir",
func(args ...any) (any, ArErr) {
if len(args) != 1 {
return nil, ArErr{
TYPE: "runtime",
message: "isDir takes exactly 1 argument, got " + fmt.Sprint(len(args)),
EXISTS: true,
}
}
args[0] = ArValidToAny(args[0])
if typeof(args[0]) != "string" {
return nil, ArErr{
TYPE: "runtime",
message: "isDir argument must be a string, got " + typeof(args[0]),
EXISTS: true,
}
}
stat, err := os.Stat(args[0].(string))
if err != nil {
if os.IsNotExist(err) {
return false, ArErr{}
}
return nil, ArErr{
TYPE: "runtime",
message: err.Error(),
EXISTS: true,
}
}
return stat.IsDir(), ArErr{}
}},
"join": builtinFunc{
"join",
func(args ...any) (any, ArErr) {

21
src/string.go
View File

@@ -262,6 +262,27 @@ func ArString(str string) ArObject {
}
return output, ArErr{}
}}
obj.obj["splitN"] = builtinFunc{
"splitN",
func(a ...any) (any, ArErr) {
if len(a) != 2 {
return nil, ArErr{"TypeError", "expected 2 or more argument, got " + fmt.Sprint(len(a)), 0, "", "", true}
}
if typeof(a[0]) != "string" {
return nil, ArErr{"TypeError", "expected string, got " + typeof(a[0]), 0, "", "", true}
}
if typeof(a[1]) != "number" || !a[1].(number).IsInt() {
return nil, ArErr{"TypeError", "expected integer, got " + typeof(a[1]), 0, "", "", true}
}
splitby := ArValidToAny(a[0]).(string)
n := int(a[1].(number).Num().Int64())
output := []any{}
splitted := (strings.SplitN(str, splitby, n))
for _, v := range splitted {
output = append(output, ArString(v))
}
return output, ArErr{}
}}
obj.obj["capitalise"] = builtinFunc{
"capitalise",
func(a ...any) (any, ArErr) {