gir.go

package main

//go:generate gir -f testdata/test.gir -o testdata/test_amd64.s -proto testdata/test_proto.go
//go:generate gir -f testdata/test1.gir -o testdata/test1_amd64.s -proto testdata/test1_proto.go
//go:generate gir -f testdata/test2.gir -o testdata/test2_amd64.s -proto testdata/test2_proto.go
//go:generate gir -f testdata/test3.gir -o testdata/test3_amd64.s -proto testdata/test3_proto.go
//go:generate gir -f testdata/test4.gir -o testdata/test4_amd64.s -proto testdata/test4_proto.go

import (
	"bufio"
	"flag"
	"fmt"
	"io"
	"io/ioutil"
	"log"
	"os"
	"strings"

	"github.com/bjwbell/gir/codegen"
	"github.com/bjwbell/gir/config"
	"github.com/bjwbell/gir/ctx"
	"github.com/bjwbell/gir/parse"
	"github.com/bjwbell/gir/scan"
	"github.com/bjwbell/gir/token"
	"github.com/bjwbell/gir/value"
)

var (
	conf    config.Config
	context value.Context
)

func filePath(pathName string) string {
	split := strings.Split(pathName, "/")
	dir := ""
	if len(split) == 1 {
		dir = "."
	} else if len(split) == 2 {
		dir = split[0] + "/"
	} else {
		dir = strings.Join(split[0:len(split)-2], "/")
	}
	return dir
}

func main() {

	context = ctx.NewContext(&conf)

	var f = flag.String("f", "", "input *.gir file ")
	var o = flag.String("o", "", "output *.s file ")
	var proto = flag.String("proto", "", "output *.go prototype file")
	flag.Parse()

	file := ""
	outfile := ""
	protofile := ""
	log.SetFlags(log.Lshortfile)
	if *f != "" {
		file = *f
	} else {
		log.Fatalf("Error no file provided")
	}
	if *o != "" {
		outfile = *o
	}
	if *proto != "" {
		protofile = *proto
	}
	var fd io.Reader
	var err error
	fd, err = os.Open(file)
	if err != nil {
		fmt.Fprintf(os.Stderr, "gir: %s\n", err)
		os.Exit(1)
	}
	var fd2 io.Reader
	var err2 error
	fd2, err2 = os.Open(file)
	if err2 != nil {
		fmt.Fprintf(os.Stderr, "gir: %s\n", err2)
		os.Exit(1)
	}
	scanner1 := scan.New(context, file, bufio.NewReader(fd))
	scanner2 := scan.New(context, file, bufio.NewReader(fd2))

	fmt.Println("Tokens: ")
	for tok := scanner1.Next(); tok.Type != token.EOF; tok = scanner1.Next() {
		fmt.Println(tok, "type ", tok.Type)
	}
	parser := parse.NewParser(file, scanner2, context)
	fileDecl := parser.ParseFile()
	pkg := fileDecl.PkgName
	fmt.Println("tree(exprs): ", parse.Tree(fileDecl))
	asm := ""
	protos := ""
	for _, fnDecl := range fileDecl.Decls {
		ssafn, ok := codegen.BuildSSA(&fnDecl, fileDecl.PkgName, false)
		if ssafn == nil || !ok {
			fmt.Println("Error building SSA form")
			return
		} else {
			fmt.Println("ssa:\n", ssafn)
			var ok bool
			asm, ok = codegen.GenAsm(ssafn)
			if !ok {
				fmt.Println("Error generating assembly")
			}

			if fnProto, ok := codegen.GenGoProto(ssafn); ok {
				protos += fnProto
			} else {
				fmt.Println("Error generating Go proto file")
			}

		}
	}

	if outfile != "" {
		err = ioutil.WriteFile(outfile, []byte(asm), 0644)
		if err != nil {
			panic(err)
		}
	}
	if protofile != "" {
		protoTxt := "// +build amd64\n\n"
		protoTxt += fmt.Sprintf("package %s\n", pkg)
		protoTxt += protos
		err = ioutil.WriteFile(protofile, []byte(protoTxt), 0644)
		if err != nil {
			panic(err)
		}
	}
}