GoBuild.py

import sublime, sublime_plugin
import os, glob
import ctypes
from GoBuildCommons import *

MessageBox = ctypes.windll.user32.MessageBoxA

class GoBuildCommand(sublime_plugin.WindowCommand):

	type = "RUN"
	project_name = ""
	project_path = ""
	valid_structure = False

 	def run(self, type="RUN"):
		self.type = type;
		self.file_name = getFileName();
		
		if isGoProject():
			self.executeProject()
		else:
			if isGoFile(self.file_name):
				self.executeFile()
			else:
				sublime.status_message("Cannot " + type.lower() + " a non go file")

	def setEnv(self):
		append = ""
		if "GOPATH" in os.environ:
			self.GOPATH = os.environ['GOPATH']
			append = self.GOPATH + ";"
        
		os.environ['GOPATH'] = append + self.project_path
		os.putenv("GOPATH", append + self.project_path)

	def restoreEnv(self):
		if self.GOPATH != "":
			os.environ['GOPATH'] = self.GOPATH
			os.putenv("GOPATH", self.GOPATH)

		self.GOPATH = ""

	def executeFile(self):
		if self.type == "RUN":
			command = ["go", "run", self.file_name]
		elif self.type == "BUILD":
			command = ["go", "build", "-x", "-v", self.file_name]
		elif self.type == "TEST":
			command = ["go", "test", self.file_name]
		else:
			self.errorMessage("Unknown command: " + self.type)
			self.restoreEnv()
			return

		getView().window().run_command("exec", { 'kill': True })
		getView().window().run_command("exec", {
			'shell': True,
			'cmd': command,
			'working_dir' : os.path.dirname(self.file_name),
			'file_regex': '^(.+\.go):([0-9]+):(?:([0-9]+):)?\s*(.*)',
		})

	def executeProject(self):
		self.project_name, self.project_path = getProject()
		base_path = os.path.dirname(self.project_path)

		project_name = self.getProjectName()

		if project_name == "":
			project_name = self.project_name

		sublime.status_message("Building project: " + project_name)

		self.setEnv()
		main = self.getMainFile()

		if len(main):
			target = main
		else:
			target = self.file_name.replace(os.path.dirname(self.project_path) + "\\", "") 


		if self.type == "RUN":
			command = "go run " + target
		elif self.type == "BUILD":
			if (hasValidStructure(base_path)):
				output = os.path.join("bin", getArch(), project_name + ".exe")
				command = "go build -x -v" + " -o " + output + " " + target
			else:
				command = "go build -x -v " + target

		elif self.type == "TEST":
			command = "go test " + self.file_name
		else:
			self.errorMessage("Unknown command: " + self.type)
			self.restoreEnv()
			return

		getView().window().run_command("exec", { 'kill': True })
		getView().window().run_command("exec", {
			'shell': True,
			'cmd': [command],
			'working_dir' : os.path.dirname(self.project_path),
			'file_regex': '^(.+\.go):([0-9]+):(?:([0-9]+):)?\s*(.*)',
		})

		self.restoreEnv()

	def getProjectName(self):
		infile = open(self.project_path, "r")
		project = json.loads(infile.read())
		if "name" in project["settings"]:
			name = project["settings"]["name"]
		else: 
			name = ""
		infile.close()
		return name

	def getMainFile(self):
		infile = open(self.project_path, "r")
		project = json.loads(infile.read())
		if "main" in project["settings"]:
			main = project["settings"]["main"]
		else:
			main = ""
		infile.close()
		return main