feat(messaging): lock code-managed templates and adopt templates into code

products/messaging/backend/facade/contracts.py

@@ -1,3 +1,4 @@
+from collections.abc import Mapping
 from dataclasses import dataclass, field
 from datetime import datetime
 from typing import Any
@@ -321,6 +322,10 @@ class TemplateManagedBy(LabeledStrEnum):
     CODE = "code", "Code"
 
 
+class TemplateReleasedTo(LabeledStrEnum):
+    GUI = "gui", "GUI"
+
+
 class TemplateUploadResult(LabeledStrEnum):
     CREATED = "created", "Created"
     UPDATED = "updated", "Updated"
@@ -335,6 +340,34 @@ class ContentHashState(LabeledStrEnum):
     OUTDATED = "outdated", "Outdated"
 
 
+@frozen
+class TemplateWriteIntent:
+    action: str
+    payload: Mapping[str, Any]
+
+
+class TemplateWriteRefused(Exception):
+    def __init__(self, *, code: str, detail: str, why: str, fix: str) -> None:
+        super().__init__(detail)
+        self.code = code
+        self.detail = detail
+        self.why = why
+        self.fix = fix
+
+
+class TemplateManagedByCode(TemplateWriteRefused):
+    def __init__(self, *, detail: str, why: str, fix: str) -> None:
+        super().__init__(code="immutable", detail=detail, why=why, fix=fix)
+
+
+class TemplateKeyRefused(TemplateWriteRefused):
+    pass
+
+
+class TemplateEditedInPostHog(Exception):
+    code = "template_edited_in_posthog"
+
+
 @frozen
 class TemplateSource:
     repository: str | None
@@ -353,3 +386,4 @@ class TemplateUpload:
     design: dict[str, Any]
     source: TemplateSource | None
     if_not_superseded: bool
+    if_code_managed: bool

products/messaging/backend/facade/templates.py

@@ -10,10 +10,17 @@
 from posthog.api.tagged_item import current_tag_names
 from posthog.dataclasses import frozen
 
-from products.messaging.backend.facade.contracts import TemplateManagedBy, TemplateUpload, TemplateUploadResult
+from products.messaging.backend.facade.contracts import (
+    TemplateManagedBy,
+    TemplateUpload,
+    TemplateUploadResult,
+    TemplateWriteIntent,
+)
 from products.messaging.backend.models.message_template import MessageTemplate
 from products.messaging.backend.services import (
     message_templates as templates_service,
+    template_keys,
+    template_ownership,
     template_saves,
     template_uploads,
 )
@@ -123,10 +130,13 @@ def save_template_draft(
     template_id: UUID,
     draft: dict[str, Any] | None,
     *,
+    intent: TemplateWriteIntent,
     require_workflow_editor: Callable[[UUID], None],
 ) -> MessageTemplateRow:
     return _to_contract(
-        template_saves.save_locked_draft(team_id, template_id, draft, require_workflow_editor=require_workflow_editor)
+        template_saves.save_locked_draft(
+            team_id, template_id, draft, intent=intent, require_workflow_editor=require_workflow_editor
+        )
     )
 
 
@@ -135,6 +145,7 @@ def save_template_fields(
     template_id: UUID,
     fields: dict[str, Any],
     *,
+    intent: TemplateWriteIntent,
     user_id: int | None,
     require_workflow_editor: Callable[[UUID], None],
 ) -> MessageTemplateRow:
@@ -150,6 +161,7 @@ def write(template: MessageTemplate) -> None:
             team_id,
             template_id,
             write,
+            intent=intent,
             user_id=user_id,
             require_workflow_editor=require_workflow_editor,
         )
@@ -161,6 +173,7 @@ def save_template_content(
     template_id: UUID,
     edit: Callable[[dict[str, Any]], dict[str, Any]],
     *,
+    intent: TemplateWriteIntent,
     user_id: int | None,
     require_workflow_editor: Callable[[UUID], None],
 ) -> MessageTemplateRow:
@@ -169,12 +182,53 @@ def save_template_content(
             team_id,
             template_id,
             lambda template: templates_service.edit_content(template, edit),
+            intent=intent,
             user_id=user_id,
             require_workflow_editor=require_workflow_editor,
         )
     )
 
 
+def check_template_write(template: MessageTemplateRow, intent: TemplateWriteIntent) -> None:
+    template_keys.check_key_kept(template, intent)
+    template_ownership.check_write(template, intent)
+
+
+def is_release(intent: TemplateWriteIntent) -> bool:
+    return template_ownership.is_release(intent)
+
+
+def release_template(
+    team_id: int, template_id: UUID, *, require_workflow_editor: Callable[[UUID], None]
+) -> MessageTemplateRow:
+    return _to_contract(
+        template_saves.release_template(team_id, template_id, require_workflow_editor=require_workflow_editor)
+    )
+
+
+def is_adoption(intent: TemplateWriteIntent) -> bool:
+    return template_keys.is_adoption(intent)
+
+
+def check_template_key_write(template: MessageTemplateRow, intent: TemplateWriteIntent) -> None:
+    template_keys.check_key_unchanged(template, intent)
+
+
+def adopt_template(
+    team_id: int,
+    template_id: UUID,
+    key: str,
+    *,
+    intent: TemplateWriteIntent,
+    require_workflow_editor: Callable[[UUID], None],
+) -> MessageTemplateRow:
+    return _to_contract(
+        template_keys.adopt_template(
+            team_id, template_id, key, intent=intent, require_workflow_editor=require_workflow_editor
+        )
+    )
+
+
 def upload_template(
     team_id: int,
     upload: TemplateUpload,