Include sql server fields in generate endpoint (#438)

thinkst · May 24, 2024 · d1d7461 · d1d7461
1 parent 23cc125
commit d1d7461
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/canarytokens/models.py b/canarytokens/models.py
@@ -872,6 +872,11 @@ class WebBugTokenResponse(TokenResponse):
 
 class SQLServerTokenResponse(TokenResponse):
     token_type: Literal[TokenTypes.SQL_SERVER] = TokenTypes.SQL_SERVER
+    sql_server_sql_action: Optional[Literal["INSERT", "DELETE", "UPDATE", "SELECT"]]
+    sql_server_table_name: Optional[str]
+    sql_server_view_name: Optional[str]
+    sql_server_function_name: Optional[str]
+    sql_server_trigger_name: Optional[str]
 
 
 class ClonedWebTokenResponse(TokenResponse):
@@ -2546,6 +2551,7 @@ class ManageResponse(BaseModel):
     force_https: Optional[bool]
     clonedsite_js: Optional[str]
     clonedsite_css: Optional[str]
+    client_id: Optional[str]
 
 
 class HistoryResponse(BaseModel):

diff --git a/frontend/app.py b/frontend/app.py
@@ -780,6 +780,7 @@ async def api_manage_canarytoken(token: str, auth: str) -> ManageResponse:
         response["clonedsite_css"] = canarydrop.get_cloned_site_css(
             frontend_settings.CLOUDFRONT_URL
         )
+        response["client_id"] = frontend_settings.AZUREAPP_ID
 
     return ManageResponse(**response)
 
@@ -1291,6 +1292,11 @@ def _(token_request_details: SQLServerTokenRequest, canarydrop: Canarydrop):
         auth_token=canarydrop.auth,
         hostname=canarydrop.generated_hostname,
         url_components=list(canarydrop.get_url_components()),
+        sql_server_sql_action=canarydrop.sql_server_sql_action,
+        sql_server_table_name=canarydrop.sql_server_table_name,
+        sql_server_view_name=canarydrop.sql_server_view_name,
+        sql_server_function_name=canarydrop.sql_server_function_name,
+        sql_server_trigger_name=canarydrop.sql_server_trigger_name,
     )