diff --git a/main.go b/main.go index 7b6956e..246f8e9 100644 --- a/main.go +++ b/main.go @@ -19,6 +19,7 @@ type conf struct { Classes []string Imports []string SysImports []string + Pragma []string } var Config conf @@ -166,6 +167,7 @@ func Start() (err error) { w := wrap.NewWrapper(Debug) w.Import(Config.Imports) w.SysImport(Config.SysImports) + w.Pragma(Config.Pragma) for _, n := range(unit.Children()) { switch x := n.(type) { case *ast.ObjCInterfaceDecl: diff --git a/wrap/main.go b/wrap/main.go index e64aeba..25ecbe4 100644 --- a/wrap/main.go +++ b/wrap/main.go @@ -46,9 +46,13 @@ func (w *Wrapper) Import(ss []string) { func (w *Wrapper) SysImport(ss []string) { for _,s := range ss { - w.cCode.WriteString(` -#import <` + s + `> -`) + w.cCode.WriteString("\n#import <" + s + ">\n") + } +} + +func (w *Wrapper) Pragma(ss []string) { + for _,s := range ss { + w.cCode.WriteString("\n#pragma " + s + "\n") } }