ast,build: add build constraints to File

Updates #3
This commit is contained in:
Michael McLoughlin
2019-01-02 21:35:17 -08:00
parent d92a147644
commit 6202c2c9aa
5 changed files with 58 additions and 9 deletions

View File

@@ -22,8 +22,7 @@ func NewGoAsm(cfg Config) Printer {
}
func (p *goasm) Print(f *avo.File) ([]byte, error) {
p.header()
p.includes(f.Includes)
p.header(f)
for _, s := range f.Sections {
switch s := s.(type) {
case *avo.Function:
@@ -37,15 +36,21 @@ func (p *goasm) Print(f *avo.File) ([]byte, error) {
return p.Result()
}
func (p *goasm) header() {
func (p *goasm) header(f *avo.File) {
p.Comment(p.cfg.GeneratedWarning())
if len(f.Constraints) > 0 {
p.NL()
p.Printf(f.Constraints.GoString())
}
if len(f.Includes) > 0 {
p.NL()
p.includes(f.Includes)
}
}
func (p *goasm) includes(paths []string) {
if len(paths) == 0 {
return
}
p.NL()
for _, path := range paths {
p.Printf("#include \"%s\"\n", path)
}

View File

@@ -67,3 +67,17 @@ func TestTextDecl(t *testing.T) {
"",
})
}
func TestConstraints(t *testing.T) {
ctx := build.NewContext()
ctx.BuildConstraintExpr("linux,386 darwin,!cgo")
ctx.BuildConstraintExpr("!noasm")
AssertPrintsLines(t, ctx, printer.NewGoAsm, []string{
"// Code generated by avo. DO NOT EDIT.",
"",
"// +build linux,386 darwin,!cgo",
"// +build !noasm",
"",
})
}