pkg/compiler: switch attributes from Ident to Type

This allows parametrized attributes like size[10].
But this is not used for now.

5 files changed, 16 insertions, 11 deletions

diff --git a/pkg/ast/ast.go b/pkg/ast/ast.go
index f7d5b8b32..723d6d9c4 100644
--- a/pkg/ast/ast.go
+++ b/pkg/ast/ast.go
@@ -102,7 +102,7 @@ type Struct struct {
 	Pos      Pos
 	Name     *Ident
 	Fields   []*Field
-	Attrs    []*Ident
+	Attrs    []*Type
 	Comments []*Comment
 	IsUnion  bool
 }
diff --git a/pkg/ast/clone.go b/pkg/ast/clone.go
index b754b7650..4c938b807 100644
--- a/pkg/ast/clone.go
+++ b/pkg/ast/clone.go
@@ -105,9 +105,9 @@ func (n *Struct) Clone() Node {
 	for _, f := range n.Fields {
 		fields = append(fields, f.Clone().(*Field))
 	}
-	var attrs []*Ident
+	var attrs []*Type
 	for _, a := range n.Attrs {
-		attrs = append(attrs, a.Clone().(*Ident))
+		attrs = append(attrs, a.Clone().(*Type))
 	}
 	var comments []*Comment
 	for _, c := range n.Comments {
diff --git a/pkg/ast/format.go b/pkg/ast/format.go
index a7e0f568f..cf7edf975 100644
--- a/pkg/ast/format.go
+++ b/pkg/ast/format.go
@@ -68,7 +68,7 @@ func (res *Resource) serialize(w io.Writer) {
 }
 
 func (typedef *TypeDef) serialize(w io.Writer) {
-	fmt.Fprintf(w, "type %v%v", typedef.Name.Name, fmtIdentList(typedef.Args, false))
+	fmt.Fprintf(w, "type %v%v", typedef.Name.Name, fmtIdentList(typedef.Args))
 	if typedef.Type != nil {
 		fmt.Fprintf(w, " %v\n", fmtType(typedef.Type))
 	}
@@ -120,7 +120,11 @@ func (str *Struct) serialize(w io.Writer) {
 	for _, com := range str.Comments {
 		fmt.Fprintf(w, "#%v\n", com.Text)
 	}
-	fmt.Fprintf(w, "%c%v\n", closing, fmtIdentList(str.Attrs, true))
+	fmt.Fprintf(w, "%c", closing)
+	if attrs := fmtTypeList(str.Attrs); attrs != "" {
+		fmt.Fprintf(w, " %v", attrs)
+	}
+	fmt.Fprintf(w, "\n")
 }
 
 func (flags *IntFlags) serialize(w io.Writer) {
@@ -182,14 +186,11 @@ func fmtTypeList(args []*Type) string {
 	return w.String()
 }
 
-func fmtIdentList(args []*Ident, space bool) string {
+func fmtIdentList(args []*Ident) string {
 	if len(args) == 0 {
 		return ""
 	}
 	w := new(bytes.Buffer)
-	if space {
-		fmt.Fprintf(w, " ")
-	}
 	fmt.Fprintf(w, "[")
 	for i, arg := range args {
 		fmt.Fprintf(w, "%v%v", comma(i, ""), arg.Name)
diff --git a/pkg/ast/parser.go b/pkg/ast/parser.go
index a7cfdbf02..b28658de8 100644
--- a/pkg/ast/parser.go
+++ b/pkg/ast/parser.go
@@ -376,9 +376,9 @@ func (p *parser) parseStruct(name *Ident) *Struct {
 		p.consume(tokNewLine)
 	}
 	if p.tryConsume(tokLBrack) {
-		str.Attrs = append(str.Attrs, p.parseIdent())
+		str.Attrs = append(str.Attrs, p.parseType())
 		for p.tryConsume(tokComma) {
-			str.Attrs = append(str.Attrs, p.parseIdent())
+			str.Attrs = append(str.Attrs, p.parseType())
 		}
 		p.consume(tokRBrack)
 	}
diff --git a/pkg/ast/testdata/all.txt b/pkg/ast/testdata/all.txt
index ede11a88a..7e1db7309 100644
--- a/pkg/ast/testdata/all.txt
+++ b/pkg/ast/testdata/all.txt
@@ -46,6 +46,10 @@ s2 {
 
 }
 
+s3 {
+	f1	int8
+} [attribute[1, "foo"], another[and[another]]]
+
 type mybool8 int8
 type net_port proc[1, 2, int16be]
 type mybool16				### unexpected '\n', expecting '[', identifier