summaryrefslogtreecommitdiff
path: root/int.go
diff options
context:
space:
mode:
Diffstat (limited to 'int.go')
-rw-r--r--int.go41
1 files changed, 41 insertions, 0 deletions
diff --git a/int.go b/int.go
new file mode 100644
index 0000000..dd91a16
--- /dev/null
+++ b/int.go
@@ -0,0 +1,41 @@
+package cobol
+
+import (
+ "strconv"
+ "strings"
+)
+
+// just give back an int
+func Int(maybeInt any) int {
+ switch v := maybeInt.(type) {
+ case int:
+ return v
+ case *int:
+ return *v
+ case string:
+ s := strings.TrimSpace(v)
+ if s == "" {
+ return 0
+ }
+ i, err := strconv.Atoi(s)
+ if err == nil {
+ return i
+ }
+ parts := strings.Fields(s)
+ if len(parts) == 0 {
+ return 0
+ }
+ // check every field for an int. sure, why not
+ for _, part := range parts {
+ if part == "" {
+ continue
+ }
+ i, err := strconv.Atoi(part)
+ if err == nil {
+ return i
+ }
+ }
+ return i
+ }
+ return 0
+}