Return cloned js/css snippets in API manage endpoint (#433)

thinkst · May 22, 2024 · 23cc125 · 23cc125
1 parent f0e5d40
commit 23cc125
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/canarytokens/models.py b/canarytokens/models.py
@@ -2544,6 +2544,8 @@ class ManageResponse(BaseModel):
     wg_qr_code: Optional[str]
     qr_code: Optional[str]
     force_https: Optional[bool]
+    clonedsite_js: Optional[str]
+    clonedsite_css: Optional[str]
 
 
 class HistoryResponse(BaseModel):

diff --git a/frontend/app.py b/frontend/app.py
@@ -773,6 +773,13 @@ async def api_manage_canarytoken(token: str, auth: str) -> ManageResponse:
         response["qr_code"] = qr_code
     elif canarydrop.type == TokenTypes.CLONEDSITE:
         response["force_https"] = switchboard_settings.FORCE_HTTPS
+        response["clonedsite_js"] = canarydrop.get_cloned_site_javascript(
+            switchboard_settings.FORCE_HTTPS
+        )
+    elif canarydrop.type == TokenTypes.CSSCLONEDSITE:
+        response["clonedsite_css"] = canarydrop.get_cloned_site_css(
+            frontend_settings.CLOUDFRONT_URL
+        )
 
     return ManageResponse(**response)