asm.go

// Package asm implements a parser for LLVM IR assembly.
package asm

import (
	"io"
	"io/ioutil"

	"github.com/geode-lang/geode/llvm/asm/internal/ast"
	"github.com/geode-lang/geode/llvm/asm/internal/irx"
	"github.com/geode-lang/geode/llvm/asm/internal/lexer"
	"github.com/geode-lang/geode/llvm/asm/internal/parser"
	"github.com/geode-lang/geode/llvm/ir"
	"github.com/pkg/errors"
)

// ParseFile parses the given LLVM IR assembly file into an LLVM IR module.
func ParseFile(path string) (*ir.Module, error) {
	buf, err := ioutil.ReadFile(path)
	if err != nil {
		return nil, errors.WithStack(err)
	}
	return ParseBytes(buf)
}

// Parse parses the given LLVM IR assembly file into an LLVM IR module, reading
// from r.
func Parse(r io.Reader) (*ir.Module, error) {
	buf, err := ioutil.ReadAll(r)
	if err != nil {
		return nil, errors.WithStack(err)
	}
	return ParseBytes(buf)
}

// ParseBytes parses the given LLVM IR assembly file into an LLVM IR module,
// reading from b.
func ParseBytes(b []byte) (*ir.Module, error) {
	module, err := parseBytes(b)
	if err != nil {
		return nil, errors.WithStack(err)
	}
	// Translate the AST of the module to an equivalent LLVM IR module.
	m, err := irx.Translate(module)
	if err != nil {
		return nil, errors.WithStack(err)
	}
	return m, nil
}

// ParseString parses the given LLVM IR assembly file into an LLVM IR module,
// reading from s.
func ParseString(s string) (*ir.Module, error) {
	return ParseBytes([]byte(s))
}

// parseBytes parses the given LLVM IR assembly file into an AST, reading from
// b.
func parseBytes(b []byte) (*ast.Module, error) {
	l := lexer.NewLexer(b)
	p := parser.NewParser()
	module, err := p.Parse(l)
	if err != nil {
		return nil, errors.WithStack(err)
	}
	m, ok := module.(*ast.Module)
	if !ok {
		return nil, errors.Errorf("invalid module type; expected *ast.Module, got %T", module)
	}
	return m, nil
}