Merge pull request #577 from nevalang/feature-http-get

nevalang · May 1, 2024 · a5fb1ca · a5fb1ca
2 parents b80bf3c + 924aa0f
commit a5fb1ca
Show file tree

Hide file tree

Showing 5 changed files with 123 additions and 0 deletions.
diff --git a/examples/http/get/e2e_test.go b/examples/http/get/e2e_test.go
@@ -0,0 +1,30 @@
+package test
+
+import (
+	"os"
+	"os/exec"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func Test(t *testing.T) {
+	err := os.Chdir("../..")
+	require.NoError(t, err)
+
+	wd, err := os.Getwd()
+	require.NoError(t, err)
+	defer os.Chdir(wd)
+
+	cmd := exec.Command("neva", "run", "http/get")
+
+	out, err := cmd.CombinedOutput()
+	require.NoError(t, err)
+	require.Contains(
+		t,
+		string(out),
+		"<html>",
+	)
+
+	require.Equal(t, 0, cmd.ProcessState.ExitCode())
+}
diff --git a/examples/http/get/main.neva b/examples/http/get/main.neva
@@ -0,0 +1,14 @@
+import {
+  http
+}
+
+component Main(start) (stop) {
+  nodes {
+    http.Get
+    Println
+  }
+  net {
+    :start -> [('http://www.example.com' -> get:url)]
+    [get:resp.body, get:err] -> println -> :stop
+  }
+}
diff --git a/internal/runtime/funcs/http.go b/internal/runtime/funcs/http.go
@@ -0,0 +1,70 @@
+package funcs
+
+import (
+	"context"
+	goio "io"
+	"net/http"
+
+	"github.com/nevalang/neva/internal/runtime"
+)
+
+type httpGet struct{}
+
+func (httpGet) Create(io runtime.FuncIO, _ runtime.Msg) (func(ctx context.Context), error) {
+	urlIn, err := io.In.Port("url")
+	if err != nil {
+		return nil, err
+	}
+
+	respOut, err := io.Out.Port("resp")
+	if err != nil {
+		return nil, err
+	}
+
+	errOut, err := io.Out.Port("err")
+	if err != nil {
+		return nil, err
+	}
+
+	return func(ctx context.Context) {
+		for {
+			var u string
+			select {
+			case m := <-urlIn:
+				u = m.Str()
+			case <-ctx.Done():
+				return
+			}
+			resp, err := http.Get(u)
+			if err != nil {
+				select {
+				case errOut <- runtime.NewMapMsg(map[string]runtime.Msg{
+					"text": runtime.NewStrMsg(err.Error()),
+				}):
+					continue
+				case <-ctx.Done():
+					return
+				}
+			}
+			body, err := goio.ReadAll(resp.Body)
+			if err != nil {
+				select {
+				case errOut <- runtime.NewMapMsg(map[string]runtime.Msg{
+					"text": runtime.NewStrMsg(err.Error()),
+				}):
+					continue
+				case <-ctx.Done():
+					return
+				}
+			}
+			select {
+			case respOut <- runtime.NewMapMsg(map[string]runtime.Msg{
+				"statusCode": runtime.NewIntMsg(int64(resp.StatusCode)),
+				"body":       runtime.NewStrMsg(string(body)),
+			}):
+			case <-ctx.Done():
+				return
+			}
+		}
+	}, nil
+}
diff --git a/internal/runtime/funcs/registry.go b/internal/runtime/funcs/registry.go
@@ -50,5 +50,7 @@ func CreatorRegistry() map[string]runtime.FuncCreator {
 		// io/file
 		"read_all":  readAll{},
 		"write_all": writeAll{},
+		// http
+		"http_get": httpGet{},
 	}
 }
diff --git a/std/http/http.neva b/std/http/http.neva
@@ -0,0 +1,7 @@
+pub type Response struct {
+  statusCode int
+  body string
+}
+
+#extern(http_get)
+pub component Get(url string) (resp Response, err error)