// QuickScript 语法文件
// ANTLR 4.13.2

grammar QuickScriptFixed;

// 自定义令牌类型
CONST: 'const';

// ===== 解析器规则 =====

// 程序入口
program
    : statement* EOF
    ;

// 语句
statement
    : expressionStmt
    | variableDeclaration
    | constDeclaration
    | ifStatement
    | whileStatement
    | forStatement
    | doStatement
    | returnStatement
    | breakStatement
    | continueStatement
    | blockStatement
    | emptyStatement
    ;

// 表达式语句
expressionStmt
    : expression Semicolon
    ;

// 变量声明
variableDeclaration
    : type Identifier Assign expression Semicolon
    | VAR Identifier Assign expression Semicolon
    ;

// 常量声明
constDeclaration
    : CONST type Identifier Assign expression Semicolon
    ;

// If 语句
ifStatement
    : IF LeftParen expression RightParen statement (ELSE statement)?
    ;

// While 语句
whileStatement
    : WHILE LeftParen expression RightParen statement
    ;

// For 语句
forStatement
    : FOR LeftParen (expression Semicolon | varDecl Semicolon) expression? Semicolon expression? RightParen statement
    ;

// Do 语句
doStatement
    : DO statement WHILE LeftParen expression RightParen Semicolon
    ;

// Return 语句
returnStatement
    : RETURN expression? Semicolon
    ;

// Break 语句
breakStatement
    : BREAK Semicolon
    ;

// Continue 语句
continueStatement
    : CONTINUE Semicolon
    ;

// 代码块
blockStatement
    : LeftBrace statement* RightBrace
    ;

// 空语句
emptyStatement
    : Semicolon
    ;

// 变量声明简化版
varDecl
    : type Identifier Assign expression
    ;

// ===== 表达式 =====

expression
    : LogicalOrExpression
    ;

LogicalOrExpression
    : LogicalAndExpression (LogicalOr LogicalAndExpression)*
    ;

LogicalAndExpression
    : BitwiseOrExpression (LogicalAnd BitwiseOrExpression)*
    ;

BitwiseOrExpression
    : BitwiseXorExpression (BitwiseOr BitwiseXorExpression)*
    ;

BitwiseXorExpression
    : BitwiseAndExpression (BitwiseXor BitwiseAndExpression)*
    ;

BitwiseAndExpression
    : EqualityExpression (BitwiseAnd EqualityExpression)*
    ;

EqualityExpression
    : RelationalExpression (Equal RelationalExpression)*
    ;

RelationalExpression
    : ShiftExpression (Relational ShiftExpression)*
    ;

ShiftExpression
    : AdditiveExpression (Shift AdditiveExpression)*
    ;

AdditiveExpression
    : MultiplicativeExpression (Add MultiplicativeExpression)*
    ;

MultiplicativeExpression
    : PowerExpression (Mul PowerExpression)*
    ;

PowerExpression
    : UnaryExpression (Power UnaryExpression)*
    ;

UnaryExpression
    : Unary primary
    | primary
    ;

// 一元运算符
Unary
    : Plus | Minus | LogicalNot | BitwiseNot
    ;

// 二元运算符
LogicalOr: '||';
LogicalAnd: '&&';
BitwiseOr: '|';
BitwiseXor: '^';
BitwiseAnd: '&';
Equal: '==' | '!=';
Relational: '<' | '>' | '<=' | '>=';
Shift: '<<' | '>>';
Add: '+' | '-';
Mul: '*' | '/' | '%';
Power: '**';

// 字面量
literal
    : IntegerLiteral
    | FloatLiteral
    | StringLiteral
    | BooleanLiteral
    | NullLiteral
    ;

// 类型
type
    : INT | INT64 | UINT | UINT64 | FLOAT | DOUBLE | BOOL | STRING | CHAR | VOID
    ;

// ===== Lexer 规则 =====

// 字面量
IntegerLiteral: [0-9]+;
FloatLiteral: [0-9]+ '.' [0-9]+;
StringLiteral: '"' (~["\\\r\n] | '\\' .)* '"';
BooleanLiteral: TRUE | FALSE;
NullLiteral: NULL;

// 关键字（优先级最高）
IF: 'if';
ELSE: 'else';
WHILE: 'while';
DO: 'do';
FOR: 'for';
BREAK: 'break';
CONTINUE: 'continue';
RETURN: 'return';
VAR: 'var';

// 类型关键字
INT: 'int';
INT64: 'int64';
UINT: 'uint';
UINT64: 'uint64';
FLOAT: 'float';
DOUBLE: 'double';
BOOL: 'bool';
STRING: 'string';
CHAR: 'char';
VOID: 'void';

// 其他关键字
TRUE: 'true';
FALSE: 'false';
NULL: 'null';

// 标识符（必须在所有关键字之后）
Identifier
    : [a-zA-Z_][a-zA-Z0-9_]*
    ;

// 运算符
Plus: '+';
Minus: '-';
Multiply: '*';
Divide: '/';
Modulo: '%';
Assign: '=';
LogicalNot: '!';
BitwiseNot: '~';

// 标点符号
Semicolon: ';';
LeftParen: '(';
RightParen: ')';
LeftBrace: '{';
RightBrace: '}';

// 空白符（跳过）
WS
    : [ \t\r\n\u000C]+ -> skip
    ;

// 注释（跳过）
LINE_COMMENT
    : '//' ~[\r\n]* -> skip
    ;