grammar org.xtext.example.mydsl.MyDsl hidden(WS)

import "URL_REMOVED" as ecore
generate myDsl "URL_REMOVED"

Model:
    greeting=Greeting
;

Greeting:
    'Hello' ','? name=SIMPLE_WORD '!' NL+
    lines+=Line*
;

Line: {Line}
    SIMPLE_WORD+
    NL
;

terminal SIMPLE_WORD: 
    ('0'..'9' | 'a'..'z' | 'A'..'Z') 
    ('0'..'9' | 'a'..'z' | 'A'..'Z' | '-' | '_')*
;
terminal NL: ('\r'? '\n');
terminal WS: (' ' | '\t');

class MyDslFormatter extends AbstractFormatter2 {

    @Inject extension MyDslGrammarAccess

    def dispatch void format(Model model, extension IFormattableDocument document) {
        model.greeting.format()

        println(document)
    }

    def dispatch void format(Greeting greeting, extension IFormattableDocument document) {
        for (line : greeting.lines) {
            line.format()
            line.prepend[indent]
        }
    }

    def dispatch void format(Line line, extension IFormattableDocument document) {
        // TODO...
    }
}

@RunWith(XtextRunner)
@InjectWith(MyDslInjectorProvider)
class MyDslFormattingTest {

    @Inject extension FormatterTestHelper

    @Test
    def void indentLines() {
        assertFormatted[
            toBeFormatted = '''
                Hello, World!
                The quick brown fox
                Jumps over the lazy dog
            '''
            expectation = '''
                Hello, World!
                    The quick brown fox
                    Jumps over the lazy dog
            '''
        ]
    }
}

def dispatch void format(Greeting greeting, extension IFormattableDocument document)  {
	val begin = greeting.regionFor.ruleCallTo(NLRule)
	val end = greeting.lines.last.regionFor.ruleCallTo(NLRule)
		
	interior(begin, end)[indent]
	for (line : greeting.lines) {
		line.format
		line.prepend[indent]
	}
}

Step:
	keyword=ANY_KEYWORD
	description=Text EOL+
	content=DocString?
;

terminal ANY_KEYWORD: '*';

DocString:
	value=DOC_STRING
	EOL+
;

terminal DOC_STRING: ('"""' -> '"""' | "'''" -> "'''");

Text returns ecore::EString:
 	TextLiteral+
;

@Test
def void indentStepContent() {
    // fail
    assertFormatted[
        toBeFormatted = '''
            Scenario: Hello, World!
            * The quick brown fox
            * Jumps over the lazy dog
            """
            At -9.8m/s^2
            """
        '''
        expectation = '''
            Scenario: Hello, World!
                * The quick brown fox
                * Jumps over the lazy dog
                """
                At -9.8m/s^2
                """
        '''
    ]
}