import 'package:sqlparser/sqlparser.dart';
import 'package:sqlparser/src/reader/parser/parser.dart';
import 'package:sqlparser/src/reader/tokenizer/scanner.dart';
import 'package:sqlparser/src/utils/ast_equality.dart';
import 'package:test/test.dart';

void main() {
  test('can parse multiple statements', () {
    final sql = 'a: UPDATE tbl SET a = b; b: SELECT * FROM tbl;';
    final tokens = Scanner(sql).scanTokens();
    final moorFile = Parser(tokens).moorFile();

    final statements = moorFile.statements;

    enforceEqual(
      statements[0],
      DeclaredStatement(
        'a',
        UpdateStatement(
          table: TableReference('tbl', null),
          set: [
            SetComponent(
              column: Reference(columnName: 'a'),
              expression: Reference(columnName: 'b'),
            ),
          ],
        ),
      ),
    );
    enforceEqual(
      statements[1],
      DeclaredStatement(
        'b',
        SelectStatement(
          columns: [StarResultColumn(null)],
          from: [TableReference('tbl', null)],
        ),
      ),
    );
  });

  test('recovers from invalid statements', () {
    final sql = 'a: UPDATE tbl SET a = * d; b: SELECT * FROM tbl;';
    final tokens = Scanner(sql).scanTokens();
    final statements = Parser(tokens).moorFile().statements;

    expect(statements, hasLength(1));
    enforceEqual(
      statements[0],
      DeclaredStatement(
        'b',
        SelectStatement(
          columns: [StarResultColumn(null)],
          from: [TableReference('tbl', null)],
        ),
      ),
    );
  });

  test('parses imports and declared statements in moor mode', () {
    final sql = r'''
    import 'test.dart';
    query: SELECT * FROM tbl;
     ''';

    final tokens = Scanner(sql, scanMoorTokens: true).scanTokens();
    final statements = Parser(tokens, useMoor: true).moorFile().statements;

    expect(statements, hasLength(2));

    final parsedImport = statements[0] as ImportStatement;
    enforceEqual(parsedImport, ImportStatement('test.dart'));
    expect(parsedImport.importToken, tokens[0]);
    expect(parsedImport.importString, tokens[1]);
    expect(parsedImport.semicolon, tokens[2]);

    final declared = statements[1] as DeclaredStatement;
    enforceEqual(
      declared,
      DeclaredStatement(
        'query',
        SelectStatement(
          columns: [StarResultColumn(null)],
          from: [TableReference('tbl', null)],
        ),
      ),
    );
  });
}