From 6b10477ebc57542021382dbfa879dc16c49e0991 Mon Sep 17 00:00:00 2001 From: James Shubin Date: Fri, 2 May 2025 00:24:03 -0400 Subject: [PATCH] lang: core: convert: Add a simple str to int function --- lang/core/convert/str_to_int.go | 69 +++++++++++++++++++++++++++++++++ 1 file changed, 69 insertions(+) create mode 100644 lang/core/convert/str_to_int.go diff --git a/lang/core/convert/str_to_int.go b/lang/core/convert/str_to_int.go new file mode 100644 index 00000000..b06694ab --- /dev/null +++ b/lang/core/convert/str_to_int.go @@ -0,0 +1,69 @@ +// Mgmt +// Copyright (C) James Shubin and the project contributors +// Written by James Shubin 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 . +// +// Additional permission under GNU GPL version 3 section 7 +// +// If you modify this program, or any covered work, by linking or combining it +// with embedded mcl code and modules (and that the embedded mcl code and +// modules which link with this program, contain a copy of their source code in +// the authoritative form) containing parts covered by the terms of any other +// license, the licensors of this program grant you additional permission to +// convey the resulting work. Furthermore, the licensors of this program grant +// the original author, James Shubin, additional permission to update this +// additional permission if he deems it necessary to achieve the goals of this +// additional permission. + +package convert + +import ( + "context" + "fmt" + "strconv" + + "github.com/purpleidea/mgmt/lang/funcs/simple" + "github.com/purpleidea/mgmt/lang/types" +) + +func init() { + simple.ModuleRegister(ModuleName, "str_to_int", &simple.Scaffold{ + I: &simple.Info{ + Pure: true, + Memo: true, + Fast: true, + Spec: true, + }, + T: types.NewType("func(a str) int"), + F: StrToInt, + }) +} + +// StrToInt converts a string to an integer. +func StrToInt(ctx context.Context, input []types.Value) (types.Value, error) { + if len(input) < 1 { + return nil, fmt.Errorf("not enough args") + } + + base := 10 + bits := 64 // TODO: get from runtime? + x, err := strconv.ParseInt(input[0].Str(), base, bits) + if err != nil { + return nil, err + } + return &types.IntValue{ + V: x, + }, nil +}