Support star (pointer) *.

Author: hsluoyz

parse/expression.go

@@ -22,6 +22,7 @@ const (
 	ExpressionTypeArray
 	ExpressionTypeCall
 	ExpressionTypeIndex
+	ExpressionTypeStar
 	ExpressionTypeParameter
 	ExpressionTypeResult
 )
@@ -45,6 +46,9 @@ func (expr *Expression) String() string {
 	case ExpressionTypeIndex:
 		// array[123]
 		res = fmt.Sprintf("%s[%s]", expr.Name, expr.Children)
+	case ExpressionTypeStar:
+		// *flag
+		res = fmt.Sprintf("*%s", expr.Children[0])
 	default:
 		res = expr.Name
 	}

parse/expression_parse.go

@@ -73,7 +73,7 @@ func parseArrayExpression(expr *ast.CompositeLit) *Expression {
 	// []int{1, 2}
 	nameType := parseNameType(&expr.Type)
 	res := &Expression{
-		Type: ExpressionTypeArray,
+		Type:     ExpressionTypeArray,
 		NameType: nameType, // int*
 	}
 
@@ -112,6 +112,18 @@ func parseIndexExpression(expr *ast.IndexExpr) *Expression {
 	return res
 }
 
+func parseStarExpression(expr *ast.StarExpr) *Expression {
+	// Star expression like:
+	// *flag
+	inside := parseExpression(&expr.X)
+
+	res := &Expression{
+		Type:     ExpressionTypeStar,
+		Children: Expressions{inside}, // flag
+	}
+	return res
+}
+
 func parseExpression(expr *ast.Expr) *Expression {
 	e, ok := (*expr).(*ast.BasicLit)
 	if ok {
@@ -148,5 +160,10 @@ func parseExpression(expr *ast.Expr) *Expression {
 		return parseIndexExpression(e7)
 	}
 
+	e8, ok := (*expr).(*ast.StarExpr)
+	if ok {
+		return parseStarExpression(e8)
+	}
+
 	panic("parseExpression(): unknown expression type")
 }

parse/name_type_parse.go

@@ -6,11 +6,21 @@ package parse
 import "go/ast"
 
 func parseIdentifierNameType(nameType *ast.Ident) string {
+	// Normal type like:
+	// int, string, ..
 	return nameType.Name
 }
 
 func parseArrayNameType(nameType *ast.ArrayType) string {
-	return nameType.Elt.(*ast.Ident).Name + "*"
+	// Array type like:
+	// []int
+	return parseNameType(&nameType.Elt) + "*"
+}
+
+func parseStarNameType(nameType *ast.StarExpr) string {
+	// Star type like:
+	// *int
+	return parseNameType(&nameType.X) + "*"
 }
 
 func parseNameType(nameType *ast.Expr) string {
@@ -24,5 +34,10 @@ func parseNameType(nameType *ast.Expr) string {
 		return parseArrayNameType(e2)
 	}
 
+	e3, ok := (*nameType).(*ast.StarExpr)
+	if ok {
+		return parseStarNameType(e3)
+	}
+
 	panic("parseNameType(): unknown name type")
 }

parse/test.go

@@ -2,14 +2,14 @@ package parse
 
 import "fmt"
 
-func test(a string, bb []int) int {
+func test(a string, bb []*int) int {
 	fmt.Printf("Hello, World!\n")
 	res := 0
 	array := []int{1, 2}
 	for i := 0; i < 10; i++ {
 		res += 10 / 2
 		array = append(array, 1)
-		if res != 10 && (array[0] >= 10 || bb[3] == 5) {
+		if res != 10 && (array[0] >= 10 || *bb[3] == 5) {
 			continue
 		}
 	}