marzeq/qk revision: fb00aaf

package main

import (
  "fmt"
  "os"
  "os/exec"
  "path"

  "github.com/marzeq/quokka/codegen"
  "github.com/marzeq/quokka/import_resolve"
  "github.com/marzeq/quokka/parser"
  "github.com/marzeq/quokka/tokeniser"
  "github.com/marzeq/quokka/typechecker"
)

func _check(err error) {
  if err != nil {
    fmt.Fprintln(os.Stderr, err)
    os.Exit(1)
  }
}

func runCmd(args ...string) error {
  if len(args) == 0 {
    return nil
  }
  cmd := exec.Command(args[0], args[1:]...)
  cmd.Stdout = os.Stdout
  cmd.Stderr = os.Stderr
  err := cmd.Run()
  return err
}

func main() {
  if len(os.Args) < 3 {
    fmt.Println("Usage: quokka [output binary] [src file]")
    return
  }

  binfile := os.Args[1]
  srcfile := os.Args[2]
  rest := os.Args[3:]
  binname := path.Base(binfile)

  t, err := tokeniser.NewTokeniserFromFile(srcfile)
  _check(err)

  toks, err := t.Tokenise()
  _check(err)

  p := parser.NewParser(toks)

  ast, err := p.Parse()
  _check(err)

  merged, err := import_resolve.ProcessImports(ast, map[string]*parser.RootNode{}, map[string]bool{})
  _check(err)

  tc := typechecker.NewTypeChecker()
  ast, funcTable, typeTable, err := tc.TypeCheck(merged)
  _check(err)

  cg := codegen.NewCodeGen(ast, funcTable, typeTable)
  ir, err := cg.EmitIR()
  _check(err)

  os.Mkdir("build", 0755)
  os.WriteFile(fmt.Sprintf("build/%s.ssa", binname), []byte(ir), 0644)

  _check(runCmd("qbe", "-o", "build/"+binname+".s", "build/"+binname+".ssa"))

  _check(runCmd("cc", "-c", "-o", "build/"+binname+".o", "build/"+binname+".s"))

  args := append([]string{"cc", "-o", binfile, "build/" + binname + ".o"}, rest...)
  _check(runCmd(args...))
}