support go local module

Makefile

@@ -15,22 +15,22 @@ test: install ## commands test
 pytest: ## python test
 	python setup.py test
 
-tox: install-test-deps
+tox: install-test-deps ## run tox
 	tox .
 
-flake8: install-test-deps
+flake8: install-test-deps ## run tox flake8 only
 	tox -e flake8 .
 
 help: ## Display this help screen
 	@grep -E '^[a-zA-Z][a-zA-Z0-9_-]+:.*?## .*$$' $(MAKEFILE_LIST) | sed -e 's/^GNUmakefile://' | awk 'BEGIN {FS = ":.*?## "}; {printf "\033[36m%-20s\033[0m %s\n", $$1, $$2}'
 
-docker-build:
+docker-build: ## build development docker image
 	docker build -t wandbox-api-dev .
 
-docker-run:
+docker-run: ## run development docker container
 	docker run -it --rm -v ${PWD}:/work -w /work wandbox-api-dev bash
 
-docker-alpine:
+docker-alpine: ## run python alpine container
 	docker run -it --rm -v ${PWD}:/work -w /work python:3.8-alpine sh
 	# apk add make
 

samples/command/src/go/Makefile

@@ -1,7 +1,7 @@
 #
 # Makefile
 
-GOBUILD=go build
+GOBUILD=GO111MODULE=off go run
 # NOTE: head version is not stable
 WANDBOX_CLI_OPTIONS:=--no-head
 

samples/command/src/go/sample.go

@@ -3,10 +3,17 @@ package main
 
 import (
 	"fmt"
+
+	test1 "./test"
+	"./test2"
+	"./test3"
 )
 
 func main() {
 	fmt.Println("Hello, Wandbox!")
+	test1.Test()
+	test2.Test()
+	test3.Test()
 }
 
 // Go language references:

samples/command/src/go/test/test.go

@@ -0,0 +1,7 @@
+package test1
+
+import "fmt"
+
+func Test() {
+	fmt.Println("Test1")
+}

samples/command/src/go/test2/a.go

@@ -0,0 +1,7 @@
+package test2
+
+import "fmt"
+
+func Test() {
+	fmt.Println("Test2")
+}

samples/command/src/go/test3/a.go

@@ -0,0 +1,7 @@
+package test3
+
+import "./test"
+
+func Test() {
+	test.Test()
+}

samples/command/src/go/test3/test/test.go

@@ -0,0 +1,7 @@
+package test
+
+import "fmt"
+
+func Test() {
+	fmt.Println("Test3")
+}

wandbox/__go__.py

@@ -1,6 +1,62 @@
+import os
+import re
+
+from .runner import Runner
 from .cli import CLI
 
 
+class GoRunner(Runner):
+
+    IMPORT_REGEX = re.compile(r'^\s*import\s+(.*?)$')
+    IMPORT_MULTILINE_REGEX = re.compile(r'import\s+\((.*?)\)', re.MULTILINE | re.DOTALL)
+
+    def reset(self):
+        self.imported = []
+        self.incdirs = []
+
+    def make_code(self, file, filepath, filename):
+        files = dict()
+        code = ''
+        for line in file:
+            m = self.IMPORT_REGEX.match(line)
+            if m:
+                files.update(self.on_import_single(filepath, m.group(1).strip('(\'")')))
+            code += line
+        for m in self.IMPORT_MULTILINE_REGEX.finditer(code):
+            files.update(self.on_import_multiline(filepath, m.group(1)))
+        files[filename] = code
+        return files
+
+    def on_import_multiline(self, filepath, imports):
+        files = dict()
+        for line in imports.splitlines():
+            names = line.split()
+            if len(names) == 1:
+                files.update(self.on_import_single(filepath, names[0].strip('(\'")')))
+            elif len(names) > 1:
+                files.update(self.on_import_single(filepath, names[1].strip('(\'")')))
+        return files
+
+    def on_import_single(self, filepath, filename):
+        files = dict()
+        if filename.startswith('.'):
+            path = os.path.join(os.path.dirname(filepath), filename)
+            if os.path.isdir(path):
+                path = os.path.normpath(path)
+                if path not in self.imported:
+                    self.imported.append(path)
+                    files.update(self.on_import(path, filename))
+        return files
+
+    def on_import(self, filepath, packagepath):
+        files = dict()
+        for f in os.listdir(filepath):
+            go_filepath = os.path.join(filepath, f)
+            if os.path.isfile(go_filepath) and f.endswith('.go'):
+                files.update(self.open_code(go_filepath, go_filepath))
+        return files
+
+
 class GoCLI(CLI):
 
     def __init__(self, compiler=None):
@@ -14,6 +70,9 @@ def go_setup(self, lang, compiler):
             help='set -gcflags -m'
         )
 
+    def get_runner(self, args, options):
+        return GoRunner(args.language, args.compiler, args.save, args.encoding, args.retry, args.retry_wait)
+
     def setup_runner(self, args, enable_options, disable_options, runner):
         self.check_bool_option(args, 'gcflags-m', enable_options, disable_options, 'go-')
         runner.has_runtime_option_raw = False