Add possibility to get field directly on struct initialization

Author: monstermichl

parser/parser.go

@@ -3405,7 +3405,17 @@ func (p *Parser) evaluateStructInitialization(ctx context) (Expression, error) {
 	if err != nil {
 		return nil, err
 	}
-	return structInitialization, nil
+	var expr Expression = structInitialization
+	nextToken = p.peek()
+
+	if nextToken.Type() == lexer.DOT {
+		expr, _, err = p.evaluateStructFieldsFromExpression(expr, nextToken, "", false, ctx) // TODO: Find out if importAlias must be passed correctly.
+
+		if err != nil {
+			return nil, err
+		}
+	}
+	return expr, nil
 }
 
 func (p *Parser) evaluateSubscript(ctx context) (Expression, error) {

tests/struct.go

@@ -60,6 +60,20 @@ func testDeclareAndDefineStructWithValuesOneLineSuccess(t *testing.T, transpiler
 	})
 }
 
+func testGetFieldDirectlyFromStructInitializationSuccess(t *testing.T, transpilerFunc transpilerFunc) {
+	transpilerFunc(t, `
+		type myStruct struct {
+			a, b string
+			c    bool
+			d    int
+		}
+		print(myStruct{a: "Hello", b: "World"}.b)
+	`, func(output string, err error) {
+		require.Nil(t, err)
+		require.Equal(t, "World", output)
+	})
+}
+
 func testDeclareAndDefineStructSliceSuccess(t *testing.T, transpilerFunc transpilerFunc) {
 	transpilerFunc(t, `
 		type myStruct struct {

tests/struct_linux_test.go

@@ -16,6 +16,10 @@ func TestDeclareAndDefineStructWithValuesOneLineSuccess(t *testing.T) {
 	testDeclareAndDefineStructWithValuesOneLineSuccess(t, transpileBash)
 }
 
+func TestGetFieldDirectlyFromStructInitializationSuccess(t *testing.T) {
+	testGetFieldDirectlyFromStructInitializationSuccess(t, transpileBash)
+}
+
 func TestDeclareAndDefineStructSliceSuccess(t *testing.T) {
 	testDeclareAndDefineStructSliceSuccess(t, transpileBash)
 }

tests/struct_windows_test.go

@@ -16,6 +16,10 @@ func TestDeclareAndDefineStructWithValuesOneLineSuccess(t *testing.T) {
 	testDeclareAndDefineStructWithValuesOneLineSuccess(t, transpileBatch)
 }
 
+func TestGetFieldDirectlyFromStructInitializationSuccess(t *testing.T) {
+	testGetFieldDirectlyFromStructInitializationSuccess(t, transpileBatch)
+}
+
 func TestDeclareAndDefineStructSliceSuccess(t *testing.T) {
 	testDeclareAndDefineStructSliceSuccess(t, transpileBatch)
 }