Improved project structure

src/build/Build.go

@@ -0,0 +1,110 @@
+package build
+
+import (
+	"bufio"
+	"bytes"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"git.akyoto.dev/cli/q/src/directory"
+	"git.akyoto.dev/cli/q/src/elf"
+	"git.akyoto.dev/cli/q/src/errors"
+	"git.akyoto.dev/cli/q/src/log"
+)
+
+// Build describes a compiler build.
+type Build struct {
+	Name            string
+	Directory       string
+	Code            bytes.Buffer
+	Data            bytes.Buffer
+	WriteExecutable bool
+}
+
+// New creates a new build.
+func New(directory string) *Build {
+	return &Build{
+		Name:            filepath.Base(directory),
+		Directory:       directory,
+		WriteExecutable: true,
+	}
+}
+
+// Run parses the input files and generates an executable file.
+func (build *Build) Run() error {
+	err := build.Compile()
+
+	if err != nil {
+		return err
+	}
+
+	if build.WriteExecutable {
+		return writeToDisk(build.Executable(), build.Code.Bytes(), build.Data.Bytes())
+	}
+
+	return nil
+}
+
+// Compile compiles all the functions.
+func (build *Build) Compile() error {
+	stat, err := os.Stat(build.Directory)
+
+	if err != nil {
+		return err
+	}
+
+	if !stat.IsDir() {
+		return &errors.InvalidDirectory{Path: build.Directory}
+	}
+
+	directory.Walk(build.Directory, func(file string) {
+		if !strings.HasSuffix(file, ".q") {
+			return
+		}
+
+		log.Info.Println(file)
+	})
+
+	build.Code.Write([]byte{
+		0xb8, 0x01, 0x00, 0x00, 0x00, // mov eax, 1
+		0xbf, 0x01, 0x00, 0x00, 0x00, // mov edi, 1
+		0xbe, 0xa2, 0x00, 0x40, 0x00, // mov esi, 0x4000a2
+		0xba, 0x06, 0x00, 0x00, 0x00, // mov edx, 6
+		0x0f, 0x05, // syscall
+
+		0xb8, 0x3c, 0x00, 0x00, 0x00, // mov eax, 60
+		0xbf, 0x00, 0x00, 0x00, 0x00, // mov edi, 0
+		0x0f, 0x05, // syscall
+	})
+
+	build.Data.Write([]byte{'H', 'e', 'l', 'l', 'o', '\n'})
+	return nil
+}
+
+// Executable returns the path to the executable.
+func (build *Build) Executable() string {
+	return filepath.Join(build.Directory, build.Name)
+}
+
+// writeToDisk writes the executable file to disk.
+func writeToDisk(filePath string, code []byte, data []byte) error {
+	file, err := os.Create(filePath)
+
+	if err != nil {
+		return err
+	}
+
+	buffer := bufio.NewWriter(file)
+	executable := elf.New(code, data)
+	executable.Write(buffer)
+	buffer.Flush()
+
+	err = file.Close()
+
+	if err != nil {
+		return err
+	}
+
+	return os.Chmod(filePath, 0755)
+}