From 1b229a8f8b657f97ee45256f5b953784b53620ce Mon Sep 17 00:00:00 2001
From: Ayke van Laethem <aykevanlaethem@gmail.com>
Date: Mon, 24 Sep 2018 15:35:50 +0200
Subject: [PATCH] compiler: support compiling individual .go files

For example:

    tinygo run ./src/examples/test/test.go
---
 compiler/compiler.go |  6 +++++-
 ir/ir.go             | 26 ++++++++++++++++++++++++--
 2 files changed, 29 insertions(+), 3 deletions(-)

diff --git a/compiler/compiler.go b/compiler/compiler.go
index b56962fe..ee2d5e9b 100644
--- a/compiler/compiler.go
+++ b/compiler/compiler.go
@@ -169,7 +169,11 @@ func (c *Compiler) Parse(mainPath string) error {
 		ParserMode: parser.ParseComments,
 	}
 	config.Import("runtime")
-	config.Import(mainPath)
+	if strings.HasSuffix(mainPath, ".go") {
+		config.CreateFromFilenames("main", mainPath)
+	} else {
+		config.Import(mainPath)
+	}
 	lprogram, err := config.Load()
 	if err != nil {
 		return err
diff --git a/ir/ir.go b/ir/ir.go
index b495f77c..9f0fc814 100644
--- a/ir/ir.go
+++ b/ir/ir.go
@@ -109,13 +109,35 @@ func NewProgram(lprogram *loader.Program, mainPath string) *Program {
 	program := ssautil.CreateProgram(lprogram, ssa.SanityCheckFunctions|ssa.BareInits|ssa.GlobalDebug)
 	program.Build()
 
+	// Find the main package, which is a bit difficult when running a .go file
+	// directly.
+	mainPkg := program.ImportedPackage(mainPath)
+	if mainPkg == nil {
+		for _, pkgInfo := range program.AllPackages() {
+			if pkgInfo.Pkg.Name() == "main" {
+				if mainPkg != nil {
+					panic("more than one main package found")
+				}
+				mainPkg = pkgInfo
+			}
+		}
+	}
+	if mainPkg == nil {
+		panic("could not find main package")
+	}
+
 	// Make a list of packages in import order.
 	packageList := []*ssa.Package{}
 	packageSet := map[string]struct{}{}
 	worklist := []string{"runtime", mainPath}
 	for len(worklist) != 0 {
 		pkgPath := worklist[0]
-		pkg := program.ImportedPackage(pkgPath)
+		var pkg *ssa.Package
+		if pkgPath == mainPath {
+			pkg = mainPkg // necessary for compiling individual .go files
+		} else {
+			pkg = program.ImportedPackage(pkgPath)
+		}
 		if pkg == nil {
 			// Non-SSA package (e.g. cgo).
 			packageSet[pkgPath] = struct{}{}
@@ -152,7 +174,7 @@ func NewProgram(lprogram *loader.Program, mainPath string) *Program {
 
 	p := &Program{
 		Program:              program,
-		mainPkg:              program.ImportedPackage(mainPath),
+		mainPkg:              mainPkg,
 		functionMap:          make(map[*ssa.Function]*Function),
 		globalMap:            make(map[*ssa.Global]*Global),
 		methodSignatureNames: make(map[string]int),