lang: Unnested the core package from the functions dir
The core package could contain non-functions, so we might as well move it upwards.
This commit is contained in:
50
lang/core/strings/split_func.go
Normal file
50
lang/core/strings/split_func.go
Normal file
@@ -0,0 +1,50 @@
|
||||
// Mgmt
|
||||
// Copyright (C) 2013-2024+ James Shubin and the project contributors
|
||||
// Written by James Shubin <james@shubin.ca> and the project contributors
|
||||
//
|
||||
// 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 <http://www.gnu.org/licenses/>.
|
||||
|
||||
package corestrings
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"github.com/purpleidea/mgmt/lang/funcs/simple"
|
||||
"github.com/purpleidea/mgmt/lang/types"
|
||||
)
|
||||
|
||||
func init() {
|
||||
simple.ModuleRegister(ModuleName, "split", &types.FuncValue{
|
||||
T: types.NewType("func(a str, b str) []str"),
|
||||
V: Split,
|
||||
})
|
||||
}
|
||||
|
||||
// Split splits the input string using the separator and returns the segments as
|
||||
// a list.
|
||||
func Split(input []types.Value) (types.Value, error) {
|
||||
str, sep := input[0].Str(), input[1].Str()
|
||||
|
||||
segments := strings.Split(str, sep)
|
||||
|
||||
listVal := types.NewList(types.NewType("[]str"))
|
||||
|
||||
for _, segment := range segments {
|
||||
listVal.Add(&types.StrValue{
|
||||
V: segment,
|
||||
})
|
||||
}
|
||||
|
||||
return listVal, nil
|
||||
}
|
||||
64
lang/core/strings/split_func_test.go
Normal file
64
lang/core/strings/split_func_test.go
Normal file
@@ -0,0 +1,64 @@
|
||||
// Mgmt
|
||||
// Copyright (C) 2013-2024+ James Shubin and the project contributors
|
||||
// Written by James Shubin <james@shubin.ca> and the project contributors
|
||||
//
|
||||
// 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 <http://www.gnu.org/licenses/>.
|
||||
|
||||
package corestrings
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
"github.com/purpleidea/mgmt/lang/types"
|
||||
"github.com/purpleidea/mgmt/util"
|
||||
)
|
||||
|
||||
func testSplit(input, sep string, output []string) error {
|
||||
inputVal, sepVal := &types.StrValue{V: input}, &types.StrValue{V: sep}
|
||||
|
||||
val, err := Split([]types.Value{inputVal, sepVal})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
listVal, ok := val.(*types.ListValue)
|
||||
if !ok {
|
||||
return fmt.Errorf("split did not return a list")
|
||||
}
|
||||
for _, segment := range output {
|
||||
if _, ok := listVal.Contains(&types.StrValue{V: segment}); !ok {
|
||||
return fmt.Errorf("output does not contained expected segment %s", segment)
|
||||
}
|
||||
}
|
||||
for _, segment := range listVal.V {
|
||||
if !util.StrInList(segment.Str(), output) {
|
||||
return fmt.Errorf("output contains unexpected segment %s", segment.Str())
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func TestSplit(t *testing.T) {
|
||||
values := map[string][]string{
|
||||
"hello,world": {"hello", "world"},
|
||||
"hello world": {"hello world"},
|
||||
"hello;world": {"hello;world"},
|
||||
}
|
||||
|
||||
for input, output := range values {
|
||||
if err := testSplit(input, ",", output); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
23
lang/core/strings/strings.go
Normal file
23
lang/core/strings/strings.go
Normal file
@@ -0,0 +1,23 @@
|
||||
// Mgmt
|
||||
// Copyright (C) 2013-2024+ James Shubin and the project contributors
|
||||
// Written by James Shubin <james@shubin.ca> and the project contributors
|
||||
//
|
||||
// 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 <http://www.gnu.org/licenses/>.
|
||||
|
||||
package corestrings
|
||||
|
||||
const (
|
||||
// ModuleName is the prefix given to all the functions in this module.
|
||||
ModuleName = "strings"
|
||||
)
|
||||
39
lang/core/strings/to_lower_func.go
Normal file
39
lang/core/strings/to_lower_func.go
Normal file
@@ -0,0 +1,39 @@
|
||||
// Mgmt
|
||||
// Copyright (C) 2013-2024+ James Shubin and the project contributors
|
||||
// Written by James Shubin <james@shubin.ca> and the project contributors
|
||||
//
|
||||
// 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 <http://www.gnu.org/licenses/>.
|
||||
|
||||
package corestrings
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"github.com/purpleidea/mgmt/lang/funcs/simple"
|
||||
"github.com/purpleidea/mgmt/lang/types"
|
||||
)
|
||||
|
||||
func init() {
|
||||
simple.ModuleRegister(ModuleName, "to_lower", &types.FuncValue{
|
||||
T: types.NewType("func(a str) str"),
|
||||
V: ToLower,
|
||||
})
|
||||
}
|
||||
|
||||
// ToLower turns a string to lowercase.
|
||||
func ToLower(input []types.Value) (types.Value, error) {
|
||||
return &types.StrValue{
|
||||
V: strings.ToLower(input[0].Str()),
|
||||
}, nil
|
||||
}
|
||||
44
lang/core/strings/to_lower_func_test.go
Normal file
44
lang/core/strings/to_lower_func_test.go
Normal file
@@ -0,0 +1,44 @@
|
||||
// Mgmt
|
||||
// Copyright (C) 2013-2024+ James Shubin and the project contributors
|
||||
// Written by James Shubin <james@shubin.ca> and the project contributors
|
||||
//
|
||||
// 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 <http://www.gnu.org/licenses/>.
|
||||
|
||||
package corestrings
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/purpleidea/mgmt/lang/types"
|
||||
)
|
||||
|
||||
func testToLower(t *testing.T, input, expected string) {
|
||||
inputStr := &types.StrValue{V: input}
|
||||
value, err := ToLower([]types.Value{inputStr})
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
if value.Str() != expected {
|
||||
t.Errorf("invalid output, expected %s, got %s", expected, value.Str())
|
||||
}
|
||||
}
|
||||
|
||||
func TestToLowerSimple(t *testing.T) {
|
||||
testToLower(t, "Hello", "hello")
|
||||
}
|
||||
|
||||
func TestToLowerSameString(t *testing.T) {
|
||||
testToLower(t, "HELLO 22", "hello 22")
|
||||
}
|
||||
Reference in New Issue
Block a user