task: expose Statements through ParsedQuery object

- removed toQueryBuilder from parsedQuery

Author: norberttech

.php-cs-fixer.php

@@ -36,7 +36,8 @@
         'Flow/Parquet/ThriftModel',
         'Flow/CLI/Tests/Integration',
         'Flow/ETL/Tests/Unit/Loader',
-        'Flow/ETL/Tests/Unit/Exception'
+        'Flow/ETL/Tests/Unit/Exception',
+        'extension/pg-query-ext/ext',
     ]);
 
 return (new Config())

src/lib/postgresql/src/Flow/PostgreSql/AST/Nodes/Exception/InvalidStatementException.php

@@ -0,0 +1,9 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Flow\PostgreSql\AST\Nodes\Exception;
+
+final class InvalidStatementException extends \RuntimeException
+{
+}

src/lib/postgresql/src/Flow/PostgreSql/AST/Nodes/Statement.php

@@ -0,0 +1,29 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Flow\PostgreSql\AST\Nodes;
+
+use Flow\PostgreSql\AST\Nodes\Exception\InvalidStatementException;
+
+/**
+ * @template-covariant T
+ */
+interface Statement
+{
+    /**
+     * @template S of Statement<mixed>
+     *
+     * @param class-string<S> $statementClass
+     *
+     * @throws InvalidStatementException
+     *
+     * @return S
+     */
+    public function assert(string $statementClass) : self;
+
+    /**
+     * @return T
+     */
+    public function raw();
+}

src/lib/postgresql/src/Flow/PostgreSql/AST/Nodes/Statement/AlterStatement.php

@@ -0,0 +1,28 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Flow\PostgreSql\AST\Nodes\Statement;
+
+use Flow\PostgreSql\AST\Nodes\{Statement, StatementTrait};
+use Flow\PostgreSql\Protobuf\AST\{AlterCollationStmt, AlterDatabaseRefreshCollStmt, AlterDatabaseSetStmt, AlterDatabaseStmt, AlterDefaultPrivilegesStmt, AlterDomainStmt, AlterEnumStmt, AlterEventTrigStmt, AlterExtensionContentsStmt, AlterExtensionStmt, AlterFdwStmt, AlterForeignServerStmt, AlterFunctionStmt, AlterObjectDependsStmt, AlterObjectSchemaStmt, AlterOpFamilyStmt, AlterOperatorStmt, AlterOwnerStmt, AlterPolicyStmt, AlterPublicationStmt, AlterRoleSetStmt, AlterRoleStmt, AlterSeqStmt, AlterStatsStmt, AlterSubscriptionStmt, AlterSystemStmt, AlterTableMoveAllStmt, AlterTableSpaceOptionsStmt, AlterTableStmt, AlterTypeStmt, AlterUserMappingStmt, RenameStmt};
+
+/**
+ * Represents ALTER statements (ALTER TABLE, ALTER INDEX, ALTER ROLE, ALTER DATABASE, etc.).
+ *
+ * @implements Statement<AlterCollationStmt|AlterDatabaseRefreshCollStmt|AlterDatabaseSetStmt|AlterDatabaseStmt|AlterDefaultPrivilegesStmt|AlterDomainStmt|AlterEnumStmt|AlterEventTrigStmt|AlterExtensionContentsStmt|AlterExtensionStmt|AlterFdwStmt|AlterForeignServerStmt|AlterFunctionStmt|AlterObjectDependsStmt|AlterObjectSchemaStmt|AlterOperatorStmt|AlterOpFamilyStmt|AlterOwnerStmt|AlterPolicyStmt|AlterPublicationStmt|AlterRoleSetStmt|AlterRoleStmt|AlterSeqStmt|AlterStatsStmt|AlterSubscriptionStmt|AlterSystemStmt|AlterTableMoveAllStmt|AlterTableSpaceOptionsStmt|AlterTableStmt|AlterTypeStmt|AlterUserMappingStmt|RenameStmt>
+ */
+final readonly class AlterStatement implements Statement
+{
+    use StatementTrait;
+
+    public function __construct(
+        private AlterTableStmt|AlterDomainStmt|AlterFunctionStmt|AlterRoleStmt|AlterRoleSetStmt|AlterDatabaseStmt|AlterDatabaseSetStmt|AlterDatabaseRefreshCollStmt|AlterSeqStmt|AlterOwnerStmt|AlterObjectSchemaStmt|AlterObjectDependsStmt|AlterExtensionStmt|AlterExtensionContentsStmt|AlterFdwStmt|AlterForeignServerStmt|AlterUserMappingStmt|AlterTableSpaceOptionsStmt|AlterTableMoveAllStmt|AlterPolicyStmt|AlterPublicationStmt|AlterSubscriptionStmt|AlterDefaultPrivilegesStmt|AlterCollationStmt|AlterEnumStmt|AlterOperatorStmt|AlterOpFamilyStmt|AlterTypeStmt|AlterSystemStmt|AlterEventTrigStmt|AlterStatsStmt|RenameStmt $stmt,
+    ) {
+    }
+
+    public function raw() : AlterTableStmt|AlterDomainStmt|AlterFunctionStmt|AlterRoleStmt|AlterRoleSetStmt|AlterDatabaseStmt|AlterDatabaseSetStmt|AlterDatabaseRefreshCollStmt|AlterSeqStmt|AlterOwnerStmt|AlterObjectSchemaStmt|AlterObjectDependsStmt|AlterExtensionStmt|AlterExtensionContentsStmt|AlterFdwStmt|AlterForeignServerStmt|AlterUserMappingStmt|AlterTableSpaceOptionsStmt|AlterTableMoveAllStmt|AlterPolicyStmt|AlterPublicationStmt|AlterSubscriptionStmt|AlterDefaultPrivilegesStmt|AlterCollationStmt|AlterEnumStmt|AlterOperatorStmt|AlterOpFamilyStmt|AlterTypeStmt|AlterSystemStmt|AlterEventTrigStmt|AlterStatsStmt|RenameStmt
+    {
+        return $this->stmt;
+    }
+}

src/lib/postgresql/src/Flow/PostgreSql/AST/Nodes/Statement/AnalyzeStatement.php

@@ -0,0 +1,29 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Flow\PostgreSql\AST\Nodes\Statement;
+
+use Flow\PostgreSql\AST\Nodes\{Statement, StatementTrait};
+use Flow\PostgreSql\Protobuf\AST\VacuumStmt;
+
+/**
+ * Represents ANALYZE statements.
+ * Note: ANALYZE is parsed as VacuumStmt in PostgreSQL.
+ *
+ * @implements Statement<VacuumStmt>
+ */
+final readonly class AnalyzeStatement implements Statement
+{
+    use StatementTrait;
+
+    public function __construct(
+        private VacuumStmt $stmt,
+    ) {
+    }
+
+    public function raw() : VacuumStmt
+    {
+        return $this->stmt;
+    }
+}

src/lib/postgresql/src/Flow/PostgreSql/AST/Nodes/Statement/CallStatement.php

@@ -0,0 +1,26 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Flow\PostgreSql\AST\Nodes\Statement;
+
+use Flow\PostgreSql\AST\Nodes\{Statement, StatementTrait};
+use Flow\PostgreSql\Protobuf\AST\CallStmt;
+
+/**
+ * @implements Statement<CallStmt>
+ */
+final readonly class CallStatement implements Statement
+{
+    use StatementTrait;
+
+    public function __construct(
+        private CallStmt $stmt,
+    ) {
+    }
+
+    public function raw() : CallStmt
+    {
+        return $this->stmt;
+    }
+}

src/lib/postgresql/src/Flow/PostgreSql/AST/Nodes/Statement/CheckpointStatement.php

@@ -0,0 +1,26 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Flow\PostgreSql\AST\Nodes\Statement;
+
+use Flow\PostgreSql\AST\Nodes\{Statement, StatementTrait};
+use Flow\PostgreSql\Protobuf\AST\CheckPointStmt;
+
+/**
+ * @implements Statement<CheckPointStmt>
+ */
+final readonly class CheckpointStatement implements Statement
+{
+    use StatementTrait;
+
+    public function __construct(
+        private CheckPointStmt $stmt,
+    ) {
+    }
+
+    public function raw() : CheckPointStmt
+    {
+        return $this->stmt;
+    }
+}

src/lib/postgresql/src/Flow/PostgreSql/AST/Nodes/Statement/CloseStatement.php

@@ -0,0 +1,26 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Flow\PostgreSql\AST\Nodes\Statement;
+
+use Flow\PostgreSql\AST\Nodes\{Statement, StatementTrait};
+use Flow\PostgreSql\Protobuf\AST\ClosePortalStmt;
+
+/**
+ * @implements Statement<ClosePortalStmt>
+ */
+final readonly class CloseStatement implements Statement
+{
+    use StatementTrait;
+
+    public function __construct(
+        private ClosePortalStmt $stmt,
+    ) {
+    }
+
+    public function raw() : ClosePortalStmt
+    {
+        return $this->stmt;
+    }
+}

src/lib/postgresql/src/Flow/PostgreSql/AST/Nodes/Statement/ClusterStatement.php

@@ -0,0 +1,26 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Flow\PostgreSql\AST\Nodes\Statement;
+
+use Flow\PostgreSql\AST\Nodes\{Statement, StatementTrait};
+use Flow\PostgreSql\Protobuf\AST\ClusterStmt;
+
+/**
+ * @implements Statement<ClusterStmt>
+ */
+final readonly class ClusterStatement implements Statement
+{
+    use StatementTrait;
+
+    public function __construct(
+        private ClusterStmt $stmt,
+    ) {
+    }
+
+    public function raw() : ClusterStmt
+    {
+        return $this->stmt;
+    }
+}

src/lib/postgresql/src/Flow/PostgreSql/AST/Nodes/Statement/CommentStatement.php

@@ -0,0 +1,26 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Flow\PostgreSql\AST\Nodes\Statement;
+
+use Flow\PostgreSql\AST\Nodes\{Statement, StatementTrait};
+use Flow\PostgreSql\Protobuf\AST\CommentStmt;
+
+/**
+ * @implements Statement<CommentStmt>
+ */
+final readonly class CommentStatement implements Statement
+{
+    use StatementTrait;
+
+    public function __construct(
+        private CommentStmt $stmt,
+    ) {
+    }
+
+    public function raw() : CommentStmt
+    {
+        return $this->stmt;
+    }
+}