feat: add bin, oct, hex support

int_fmt.go

@@ -0,0 +1,53 @@
+package sun
+
+import (
+	"fmt"
+
+	"go.starlark.net/starlark"
+)
+
+func formatInt(
+	thread *starlark.Thread,
+	b *starlark.Builtin,
+	args starlark.Tuple,
+	kwargs []starlark.Tuple,
+	f string,
+) (starlark.Value, error) {
+	var i starlark.Int
+	if err := starlark.UnpackPositionalArgs(b.Name(), args, kwargs, 1, &i); err != nil {
+		return nil, err
+	}
+
+	if v, ok := i.Int64(); ok {
+		return starlark.String(fmt.Sprintf(f, v)), nil
+	}
+
+	return starlark.String(fmt.Sprintf(f, i.BigInt())), nil
+}
+
+func bin(
+	thread *starlark.Thread,
+	b *starlark.Builtin,
+	args starlark.Tuple,
+	kwargs []starlark.Tuple,
+) (starlark.Value, error) {
+	return formatInt(thread, b, args, kwargs, "%#b")
+}
+
+func oct(
+	thread *starlark.Thread,
+	b *starlark.Builtin,
+	args starlark.Tuple,
+	kwargs []starlark.Tuple,
+) (starlark.Value, error) {
+	return formatInt(thread, b, args, kwargs, "%O")
+}
+
+func hex(
+	thread *starlark.Thread,
+	b *starlark.Builtin,
+	args starlark.Tuple,
+	kwargs []starlark.Tuple,
+) (starlark.Value, error) {
+	return formatInt(thread, b, args, kwargs, "%#x")
+}

int_fmt_test.go

@@ -0,0 +1,7 @@
+package sun
+
+import "testing"
+
+func TestFmt(t *testing.T) {
+	runTestData(t, "fmt.star")
+}

module.go

@@ -13,5 +13,8 @@ var Module = &starlarkstruct.Module{
 		"next":     starlark.NewBuiltin("next", next),
 		"filter":   starlark.NewBuiltin("filter", filter),
 		"callable": starlark.NewBuiltin("callable", callable),
+		"hex":      starlark.NewBuiltin("hex", hex),
+		"oct":      starlark.NewBuiltin("oct", oct),
+		"bin":      starlark.NewBuiltin("bin", bin),
 	},
 }

testdata/fmt.star

@@ -0,0 +1,29 @@
+load("assert.star", "assert")
+
+def test_main():
+    assert.fails(lambda: hex([]), "want int")
+    assert.fails(lambda: oct([]), "want int")
+    assert.fails(lambda: bin([]), "want int")
+
+    tests = [(-3, '-0x3'), (-2, '-0x2'), (-1, '-0x1'), (0, '0x0'), (1, '0x1'), (2, '0x2')]
+    for test in tests:
+        assert.eq(
+            hex(test[0]),
+            test[1],
+        )
+
+    tests = [(-3, '-0o3'), (-2, '-0o2'), (-1, '-0o1'), (0, '0o0'), (1, '0o1'), (2, '0o2')]
+    for test in tests:
+        assert.eq(
+            oct(test[0]),
+            test[1],
+        )
+
+    tests = [(-3, '-0b11'), (-2, '-0b10'), (-1, '-0b1'), (0, '0b0'), (1, '0b1'), (2, '0b10')]
+    for test in tests:
+        assert.eq(
+            bin(test[0]),
+            test[1],
+        )
+
+test_main()