Add lua block check for builder

Author: xynicole

build.go

@@ -131,37 +131,52 @@ func buildBlock(sb io.StringWriter, parent *Directive, block Directives, depth i
 			directive := Enquote(stmt.Directive)
 			_, _ = sb.WriteString(directive)
 
-			// special handling for if statements
-			if directive == "if" {
-				_, _ = sb.WriteString(" (")
-				for i, arg := range stmt.Args {
-					if i > 0 {
+			// special handling for'set_by_lua_block' directive
+			if directive == "set_by_lua_block" {
+				_, _ = sb.WriteString(" ")
+				_, _ = sb.WriteString(stmt.Args[0]) // argument for return value
+				_, _ = sb.WriteString(" {")
+				_, _ = sb.WriteString(stmt.Args[1]) // argument containing block content
+				_, _ = sb.WriteString("}")
+				// handling other lua block directives
+			} else if strings.Contains(directive, "_by_lua_block") {
+				_, _ = sb.WriteString(" {")
+				_, _ = sb.WriteString(stmt.Args[0])
+				_, _ = sb.WriteString("}")
+			} else {
+				// special handling for if statements
+				if directive == "if" {
+					_, _ = sb.WriteString(" (")
+					for i, arg := range stmt.Args {
+						if i > 0 {
+							_, _ = sb.WriteString(" ")
+						}
+						_, _ = sb.WriteString(Enquote(arg))
+					}
+					_, _ = sb.WriteString(")")
+				} else {
+					for _, arg := range stmt.Args {
 						_, _ = sb.WriteString(" ")
+						_, _ = sb.WriteString(Enquote(arg))
 					}
-					_, _ = sb.WriteString(Enquote(arg))
-				}
-				_, _ = sb.WriteString(")")
-			} else {
-				for _, arg := range stmt.Args {
-					_, _ = sb.WriteString(" ")
-					_, _ = sb.WriteString(Enquote(arg))
 				}
-			}
 
-			if !stmt.IsBlock() {
-				_, _ = sb.WriteString(";")
-			} else {
-				_, _ = sb.WriteString(" {")
-				stmt := stmt
-				buildBlock(sb, stmt, stmt.Block, depth+1, stmt.Line, options)
-				_, _ = sb.WriteString("\n")
-				_, _ = sb.WriteString(margin(options, depth))
-				_, _ = sb.WriteString("}")
+				if !stmt.IsBlock() {
+					_, _ = sb.WriteString(";")
+				} else {
+					_, _ = sb.WriteString(" {")
+					stmt := stmt
+					buildBlock(sb, stmt, stmt.Block, depth+1, stmt.Line, options)
+					_, _ = sb.WriteString("\n")
+					_, _ = sb.WriteString(margin(options, depth))
+					_, _ = sb.WriteString("}")
+				}
 			}
 		}
 		lastLine = stmt.Line
 	}
 }
+
 func margin(options *BuildOptions, depth int) string {
 	indent := depth * options.Indent
 	if indent < MaxIndent {

build_test.go

@@ -35,7 +35,7 @@ type compareFixture struct {
 	options ParseOptions
 }
 
-//nolint:gochecknoglobals
+//nolint:gochecknoglobals, goconst
 var buildFixtures = []buildFixture{
 	{
 		name:    "nested-and-multiple-args",
@@ -258,6 +258,44 @@ var buildFixtures = []buildFixture{
 		},
 		expected: "#comment1\nuser root; #comment2 #comment3",
 	},
+	{
+		name:    "lua block",
+		options: BuildOptions{},
+		parsed: Directives{
+			{
+				Directive: "content_by_lua_block",
+				Line:      1,
+				Args:      []string{"\n                ngx.say(\"I need no extra escaping here, for example: \\r\\nblah\")\n            "},
+			},
+		},
+
+		expected: "content_by_lua_block {\n                ngx.say(\"I need no extra escaping here, for example: \\r\\nblah\")\n            }",
+	},
+	{
+		name:    "set_by_lua_block",
+		options: BuildOptions{},
+		parsed: Directives{
+			{
+				Directive: "set_by_lua_block",
+				Line:      1,
+				Args: []string{"$res", " -- irregular lua block directive" +
+					"\n            local a = 32" +
+					"\n            local b = 56" +
+					"\n" +
+					"\n            ngx.var.diff = a - b;  -- write to $diff directly" +
+					"\n            return a + b;          -- return the $sum value normally" +
+					"\n        "},
+			},
+		},
+
+		expected: "set_by_lua_block $res { -- irregular lua block directive" +
+			"\n            local a = 32" +
+			"\n            local b = 56" +
+			"\n" +
+			"\n            ngx.var.diff = a - b;  -- write to $diff directly" +
+			"\n            return a + b;          -- return the $sum value normally" +
+			"\n        }",
+	},
 }
 
 func TestBuild(t *testing.T) {