lang: Add to_lower and to_upper functions

Signed-off-by: Julien Pivotto <roidelapluie@inuits.eu>

lang/funcs/core/corestrings/to_lower_func.go

@@ -0,0 +1,39 @@
+// Mgmt
+// Copyright (C) 2013-2018+ 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
+}

lang/funcs/core/corestrings/to_lower_func_test.go

@@ -0,0 +1,44 @@
+// Mgmt
+// Copyright (C) 2013-2018+ 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")
+}

lang/funcs/core/corestrings/to_upper_func.go

@@ -0,0 +1,39 @@
+// Mgmt
+// Copyright (C) 2013-2018+ 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_upper", &types.FuncValue{
+		T: types.NewType("func(a str) str"),
+		V: ToUpper,
+	})
+}
+
+// ToUpper turns a string to uppercase.
+func ToUpper(input []types.Value) (types.Value, error) {
+	return &types.StrValue{
+		V: strings.ToUpper(input[0].Str()),
+	}, nil
+}

lang/funcs/core/corestrings/to_upper_func_test.go

@@ -0,0 +1,44 @@
+// Mgmt
+// Copyright (C) 2013-2018+ 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 testToUpper(t *testing.T, input, expected string) {
+	inputStr := &types.StrValue{V: input}
+	value, err := ToUpper([]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 TestToUpperSimple(t *testing.T) {
+	testToUpper(t, "Hello", "HELLO")
+}
+
+func TestToUpperSameString(t *testing.T) {
+	testToUpper(t, "HELLO 22", "HELLO 22")
+}