Merge pull request #352 from spoo-me/feat/link-tags

feat(links): tags as a per-account registry with ids, colours and icons

Author: Zingzy

dependencies/__init__.py

@@ -72,6 +72,7 @@
     ProfilePictureSvc,
     ReportIntakeSvc,
     StatsSvc,
+    TagSvc,
     UrlExpandSvc,
     UrlPolicy,
     UrlSvc,
@@ -97,6 +98,7 @@
     get_profile_picture_service,
     get_report_intake_service,
     get_stats_service,
+    get_tag_service,
     get_url_policy,
     get_url_service,
     get_user_repo,
@@ -147,6 +149,7 @@
     "ReportIntakeSvc",
     "Settings",
     "StatsSvc",
+    "TagSvc",
     "UrlExpandSvc",
     "UrlPolicy",
     "UrlSvc",
@@ -186,6 +189,7 @@
     "get_report_intake_service",
     "get_settings",
     "get_stats_service",
+    "get_tag_service",
     "get_url_policy",
     "get_url_service",
     "get_user_repo",

dependencies/services.py

@@ -39,6 +39,7 @@
 from services.report_intake_service import ReportIntakeService
 from services.safety.policy import UrlPolicyService
 from services.stats_service import StatsService
+from services.tag_service import TagService
 from services.url_expand_service import UrlExpandService
 from services.url_service import UrlService
 from services.webhooks.service import WebhookService
@@ -52,6 +53,10 @@ def get_url_policy(request: Request) -> UrlPolicyService:
     return request.app.state.url_policy
 
 
+def get_tag_service(request: Request) -> TagService:
+    return request.app.state.tag_service
+
+
 def get_bulk_url_service(request: Request) -> BulkUrlService:
     return request.app.state.bulk_url_service
 
@@ -169,6 +174,7 @@ def get_webhook_service(request: Request) -> WebhookService:
 UrlSvc = Annotated[UrlService, Depends(get_url_service)]
 UrlPolicy = Annotated[UrlPolicyService, Depends(get_url_policy)]
 BulkUrlSvc = Annotated[BulkUrlService, Depends(get_bulk_url_service)]
+TagSvc = Annotated[TagService, Depends(get_tag_service)]
 StatsSvc = Annotated[StatsService, Depends(get_stats_service)]
 PublicStatsSvc = Annotated[PublicStatsService, Depends(get_public_stats_service)]
 ExportSvc = Annotated[ExportService, Depends(get_export_service)]

dependencies/wiring.py

@@ -47,6 +47,7 @@
     ReportSubmissionRepository,
 )
 from repositories.scheduled_task_repository import ScheduledTaskRepository
+from repositories.tag_repository import TagRepository
 from repositories.token_repository import TokenRepository
 from repositories.url_repository import UrlRepository
 from repositories.user_repository import UserRepository
@@ -123,6 +124,7 @@
 from services.scheduler import TaskScheduler
 from services.scheduler.tasks import build_task_registry
 from services.stats_service import StatsService
+from services.tag_service import TagService
 from services.tenant_resolver import CachedMongoTenantResolver
 from services.token_factory import TokenFactory
 from services.url_expand_service import UrlExpandService
@@ -358,6 +360,7 @@ def build_account_erasure_service(
         user_repo=user_repo,
         url_service=url_service,
         domain_service=domain_service,
+        tag_service=TagService(TagRepository(db["tags"]), UrlRepository(db["urlsV2"])),
         click_repo=ClickRepository(db["clicks"]),
         api_key_repo=ApiKeyRepository(db["api-keys"]),
         token_repo=TokenRepository(db["verification-tokens"]),
@@ -668,6 +671,8 @@ def wire_services(app: FastAPI, settings: AppSettings, redis_client) -> None:
     )
     app.state.url_policy = url_policy
     # ── Services ─────────────────────────────────────────────────────────
+    tag_repo = TagRepository(db["tags"])
+    app.state.tag_service = TagService(tag_repo, url_repo)
     app.state.url_service = UrlService(
         url_repo,
         legacy_repo,
@@ -692,6 +697,7 @@ def wire_services(app: FastAPI, settings: AppSettings, redis_client) -> None:
         meta_key_secret=settings.secret_key,
         events=app.state.domain_event_sink,
         user_repo=user_repo,
+        tag_service=app.state.tag_service,
     )
     app.state.bulk_url_service = BulkUrlService(
         url_repo,
@@ -701,11 +707,13 @@ def wire_services(app: FastAPI, settings: AppSettings, redis_client) -> None:
         system_default_domain=settings.system_default_domain,
         og_ttl_seconds=edge.og_ttl_seconds,
         events=app.state.domain_event_sink,
+        tag_service=app.state.tag_service,
     )
     app.state.stats_service = StatsService(
         click_repo,
         url_repo,
         max_date_range_days=settings.max_date_range_days,
+        tag_service=app.state.tag_service,
     )
     # One resolver serves BOTH public read-only surfaces (preview + stats)
     # so they can never disagree about which link a code names or what
@@ -988,6 +996,7 @@ def wire_services(app: FastAPI, settings: AppSettings, redis_client) -> None:
         user_repo=user_repo,
         url_service=app.state.url_service,
         domain_service=app.state.custom_domain_service,
+        tag_service=app.state.tag_service,
         click_repo=click_repo,
         api_key_repo=api_key_repo,
         token_repo=token_repo,

middleware/rate_limiter.py

@@ -99,6 +99,10 @@ class Limits:
     URL_BULK_STATUS = "60 per minute; 200 per day"
     URL_BULK_EXPIRY = "60 per minute; 200 per day"
     URL_BULK_DOMAIN = "60 per minute; 200 per day"
+    URL_BULK_TAGS = "60 per minute; 200 per day"
+    # Tags: writes are cheap single-doc ops; delete fans out over every link.
+    TAG_WRITE = "30 per minute"
+    TAG_DELETE = "10 per minute"
     URL_BULK_MUTATE_DELETE = "30 per minute; 100 per day"
 
     # Claim intake (POST /api/v1/urls/claim). Per submission, 16-item cap;