chore(data-warehouse): joins on views

posthog/hogql/ast.py

@@ -187,10 +187,24 @@
 class SelectQueryAliasType(Type):
     alias: str
     select_query_type: SelectQueryType | SelectUnionQueryType
+    view_name: Optional[str] = None
 
     def get_child(self, name: str, context: HogQLContext) -> Type:
         if name == "*":
             return AsteriskType(table_type=self)
+        if self.view_name:
+            field = context.database.get_table(self.view_name).get_field(name)
+            if isinstance(field, LazyJoin):
+                return LazyJoinType(table_type=self, field=name, lazy_join=field)
+            if isinstance(field, LazyTable):
+                return LazyTableType(table=field)
+            if isinstance(field, FieldTraverser):
+                return FieldTraverserType(table_type=self, chain=field.chain)
+            if isinstance(field, VirtualTable):
+                return VirtualTableType(table_type=self, field=name, virtual_table=field)
+            if isinstance(field, ExpressionField):
+                return ExpressionFieldType(table_type=self, name=name, expr=field.expr)
+            return FieldType(name=name, table_type=self)
         if self.select_query_type.has_child(name, context):
             return FieldType(name=name, table_type=self)
         raise HogQLException(f"Field {name} not found on query with alias {self.alias}")
@@ -575,6 +589,7 @@
     limit_with_ties: Optional[bool] = None
     offset: Optional[Expr] = None
     settings: Optional[HogQLQuerySettings] = None
+    view_name: Optional[str] = None
 
 
 @dataclass(kw_only=True)

posthog/hogql/resolver.py

@@ -207,6 +207,7 @@ def visit_select_query(self, node: ast.SelectQuery):
             {name: self.visit(expr) for name, expr in node.window_exprs.items()} if node.window_exprs else None
         )
         new_node.settings = node.settings.model_copy() if node.settings is not None else None
+        new_node.view_name = node.view_name
 
         self.scopes.pop()
 
@@ -276,6 +277,10 @@ def visit_join_expr(self, node: ast.JoinExpr):
                         raise ResolverException("Nested views are not supported")
 
                     node.table = parse_select(str(database_table.query))
+
+                    if isinstance(node.table, ast.SelectQuery):
+                        node.table.view_name = database_table.name
+
                     node.alias = table_alias or database_table.name
                     node = self.visit(node)
 
@@ -334,6 +339,9 @@ def visit_join_expr(self, node: ast.JoinExpr):
                         f'Already have joined a table called "{node.alias}". Can\'t join another one with the same name.'
                     )
                 node.type = ast.SelectQueryAliasType(alias=node.alias, select_query_type=node.table.type)
+                if isinstance(node.table, ast.SelectQuery):
+                    node.type.view_name = node.table.view_name
+
                 scope.tables[node.alias] = node.type
             else:
                 node.type = node.table.type

posthog/hogql/visitor.py

@@ -485,6 +485,7 @@ def visit_select_query(self, node: ast.SelectQuery):
             if node.window_exprs
             else None,
             settings=node.settings.model_copy() if node.settings is not None else None,
+            view_name=node.view_name,
         )
 
     def visit_select_union_query(self, node: ast.SelectUnionQuery):