Initial commit

Author: tdakkota

.github/dependabot.yml

@@ -0,0 +1,17 @@
+# To get started with Dependabot version updates, you'll need to specify which
+# package ecosystems to update and where the package manifests are located.
+# Please see the documentation for all configuration options:
+# https://help.github.com/github/administering-a-repository/configuration-options-for-dependency-updates
+
+version: 2
+updates:
+  - package-ecosystem: "gomod" # See documentation for possible values
+    directory: "/" # Location of package manifests
+    schedule:
+      interval: "daily"
+
+  # Maintain dependencies for GitHub Actions
+  - package-ecosystem: "github-actions"
+    directory: "/"
+    schedule:
+      interval: "daily"

.github/workflows/coverage.yml

@@ -0,0 +1,70 @@
+name: Coverage
+on:
+  push:
+    tags:
+      - v*
+    branches:
+      - master
+  pull_request:
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Install Go
+        uses: actions/setup-go@v2.1.3
+        with:
+          go-version: 1.16
+
+      - name: Get Go environment
+        id: go-env
+        run: |
+          echo "::set-output name=cache::$(go env GOCACHE)"
+          echo "::set-output name=modcache::$(go env GOMODCACHE)"
+      - name: Set up cache
+        uses: actions/cache@v2.1.4
+        with:
+          path: |
+            ${{ steps.go-env.outputs.cache }}
+            ${{ steps.go-env.outputs.modcache }}
+          key: test-${{ runner.os }}-go-${{ hashFiles('**/go.sum') }}
+          restore-keys: |
+            test-${{ runner.os }}-go-
+
+      - name: Checkout code
+        uses: actions/checkout@v2.3.4
+
+      - name: Download dependencies
+        run: go mod download && go mod tidy
+
+      - name: Run tests with coverage
+        run: go test -v -coverpkg=./... -coverprofile=profile.out ./...
+
+      - name: Extract cover profile
+        run: go tool cover -func profile.out
+
+      - name: Upload artifact
+        uses: actions/upload-artifact@v2.2.2
+        with:
+          name: coverage
+          path: profile.out
+          if-no-files-found: error
+          retention-days: 1
+
+  upload:
+    runs-on: ubuntu-latest
+    needs:
+      - test
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v2.3.4
+
+      - name: Download artifact
+        uses: actions/download-artifact@v2.0.8
+        with:
+          name: coverage
+
+      - name: Send coverage
+        uses: codecov/codecov-action@v1.3.1
+        with:
+          file: profile.out

.github/workflows/lint.yml

@@ -0,0 +1,28 @@
+name: Lint
+on:
+  push:
+    tags:
+      - v*
+    branches:
+      - master
+  pull_request:
+  workflow_dispatch:
+
+jobs:
+  golangci-lint:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v2.3.4
+
+      - name: Install Go
+        uses: actions/setup-go@v2.1.3
+        with:
+          go-version: 1.16
+
+      - name: Lint
+        uses: golangci/golangci-lint-action@v2.5.1
+        with:
+          version: latest
+          args: --timeout 5m
+          skip-go-installation: true

.github/workflows/pr-extra.yml

@@ -0,0 +1,27 @@
+name: Extra
+on:
+  push:
+    tags:
+      - v*
+    branches:
+      - master
+  pull_request:
+  workflow_dispatch:
+
+jobs:
+  vulns:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v2.3.4
+
+      - name: Install Go
+        uses: actions/setup-go@v2.1.3
+        with:
+          go-version: 1.16
+
+      - name: List dependencies
+        run: go list -json -m all > go.list
+
+      - name: Run nancy
+        uses: sonatype-nexus-community/nancy-github-action@v1.0.2

.gitignore

@@ -0,0 +1,22 @@
+### Go template
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Env files
+*.env
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# IntelliJ project files
+.idea
+*.iml
+out
+gen

README.md

@@ -0,0 +1,18 @@
+package sun
+
+import (
+	"fmt"
+
+	"go.starlark.net/starlark"
+)
+
+func wantArgs(fnName string, args starlark.Tuple, kwargs []starlark.Tuple, count int) error {
+	switch {
+	case len(kwargs) > 0:
+		return fmt.Errorf("%s does not accept keyword arguments", fnName)
+	case len(args) != count:
+		return fmt.Errorf("%s: got %d arguments, want exactly %d", fnName, len(args), count)
+	default:
+		return nil
+	}
+}

args.go

@@ -0,0 +1,21 @@
+package sun
+
+import (
+	"go.starlark.net/starlark"
+)
+
+func callable(
+	thread *starlark.Thread,
+	b *starlark.Builtin,
+	args starlark.Tuple,
+	kwargs []starlark.Tuple,
+) (starlark.Value, error) {
+	if err := wantArgs(b.Name(), args, kwargs, 1); err != nil {
+		return nil, err
+	}
+
+	if _, ok := args[0].(starlark.Callable); ok {
+		return starlark.True, nil
+	}
+	return starlark.False, nil
+}

callable.go

@@ -0,0 +1,13 @@
+package sun
+
+import (
+	"testing"
+
+	"go.starlark.net/starlark"
+)
+
+func TestCallable(t *testing.T) {
+	runTestData(t, "callable.star", starlark.StringDict{
+		"callable": starlark.NewBuiltin("callable", callable),
+	})
+}

callable_test.go

@@ -0,0 +1,2 @@
+// Package sun contains Python builtins implementations for Starlark.
+package sun

doc.go

@@ -0,0 +1,22 @@
+package sun_test
+
+import (
+	"fmt"
+
+	"go.starlark.net/starlark"
+
+	"github.com/tdakkota/sun"
+)
+
+func Example_filter() {
+	code := "list(filter(lambda x: x % 2 == 0, range(10)))"
+	thread := &starlark.Thread{Name: "main"}
+	result, err := starlark.Eval(thread, "example.star", code, sun.Module.Members)
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(result)
+	// Output:
+	// [0, 2, 4, 6, 8]
+}

example_test.go

@@ -0,0 +1,113 @@
+package sun
+
+import (
+	"fmt"
+
+	"go.starlark.net/starlark"
+)
+
+type filterFunc = func(x starlark.Value) (starlark.Value, error)
+
+type filterIter struct {
+	thread   *starlark.Thread
+	function filterFunc
+	iterator starlark.Iterator
+}
+
+func (f filterIter) Next(p *starlark.Value) bool {
+	var x starlark.Value
+	for {
+		if !f.iterator.Next(&x) {
+			return false
+		}
+
+		v, err := f.function(x)
+		if err != nil {
+			return false
+		}
+
+		if v.Truth() {
+			*p = x
+			return true
+		}
+	}
+}
+
+func (f filterIter) Done() {
+	f.iterator.Done()
+}
+
+type filterObject struct {
+	thread   *starlark.Thread
+	function filterFunc
+	iterable starlark.Iterable
+	iterator starlark.Iterator
+}
+
+func (f filterObject) String() string {
+	return fmt.Sprintf("<filter object>")
+}
+
+func (f filterObject) Type() string {
+	return "filter"
+}
+
+func (f filterObject) Freeze() {
+	f.iterable.Freeze()
+}
+
+func (f filterObject) Truth() starlark.Bool {
+	return starlark.True
+}
+
+func (f filterObject) Hash() (uint32, error) {
+	return 0, fmt.Errorf("unhashable type: filter")
+}
+
+func (f filterObject) Iterate() starlark.Iterator {
+	return filterIter{
+		thread:   f.thread,
+		function: f.function,
+		iterator: f.iterator,
+	}
+}
+
+func filter(
+	thread *starlark.Thread,
+	b *starlark.Builtin,
+	args starlark.Tuple,
+	kwargs []starlark.Tuple,
+) (starlark.Value, error) {
+	var (
+		function filterFunc
+		iterable starlark.Iterable
+	)
+	if err := wantArgs(b.Name(), args, kwargs, 2); err != nil {
+		return nil, err
+	}
+
+	switch fn := args[0].(type) {
+	case starlark.Callable:
+		function = func(x starlark.Value) (starlark.Value, error) {
+			return starlark.Call(thread, fn, starlark.Tuple{x}, nil)
+		}
+	case starlark.NoneType:
+		function = func(x starlark.Value) (starlark.Value, error) {
+			return x.Truth(), nil
+		}
+	default:
+		return nil, fmt.Errorf("got %s, want callable", fn.Type())
+	}
+
+	iterable, ok := args[1].(starlark.Iterable)
+	if !ok {
+		return nil, fmt.Errorf("got %s, want iterable", args[1].Type())
+	}
+
+	return &filterObject{
+		thread:   thread,
+		function: function,
+		iterable: iterable,
+		iterator: iterable.Iterate(),
+	}, nil
+}

filter.go

@@ -0,0 +1,13 @@
+package sun
+
+import (
+	"testing"
+
+	"go.starlark.net/starlark"
+)
+
+func TestFilter(t *testing.T) {
+	runTestData(t, "filter.star", starlark.StringDict{
+		"filter": starlark.NewBuiltin("filter", filter),
+	})
+}

filter_test.go

@@ -0,0 +1,5 @@
+module github.com/tdakkota/sun
+
+go 1.16
+
+require go.starlark.net v0.0.0-20210312235212-74c10e2c17dc