Files
Joel Salmon b8f176bb31 Initial commit: ImpactFlow Discovery + Google OAuth auth layer
Discovery service (pre-existing): FastAPI + async SQLAlchemy + Alembic +
SQLite + Anthropic, with a five-prompt static UI that produces an Enneagram
+ Ikigai profile.

Auth implementation (this change set) follows
Impact_Flow_Auth_Plan_OAuth.html, adapted to the discovery_conversation /
discovery_profile schema:

- app/auth.py: Google OAuth registration, JWT issue/decode, dual-auth
  dependency (Bearer JWT or X-API-Key), refresh-token hashing, domain
  allow-list, synthetic api-key-admin user
- app/tracking.py: ActivityTrackingMiddleware + log_activity helper;
  tags machine-to-machine calls source=mcp
- app/routers/auth.py: /api/auth/{login,callback,refresh,logout},
  /api/me, /api/me/{stats,sessions,sessions/{id}}
- app/routers/activity.py: /api/activity, /api/activity/summary,
  /api/admin/activity, plus prune_old_activity (90-day retention)
- app/routers/discovery.py: every route now user-scoped via the auth
  dependency; /discovery/profile/{user_id} -> /discovery/profile/me
- alembic/versions/002_add_auth.py: users, refresh_tokens, activity_log
- tests/test_auth.py: 8 tests covering 401 paths, X-API-Key admin
  resolution, JWT round-trip, admin gating, domain allow-list
- README.md: Authentication section, expanded env-var table, updated
  data-model and API-reference tables
- .env.example: new GOOGLE_*, JWT_*, IMPACTFLOW_API_KEY, CORS_*,
  ALLOWED_EMAIL_DOMAINS placeholders
- .gitignore: also exclude data/*.log

Tests: 19/19 pass (11 pre-existing + 8 new). smoke_test.py exercises the
full discovery flow under X-API-Key plus 401 paths, OAuth login redirect,
activity logging, and /api/me/stats.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
2026-05-27 10:59:41 -05:00

71 lines
2.1 KiB
Python

"""Helpers for bringing older local SQLite databases under Alembic control."""
import os
import sqlite3
from dotenv import load_dotenv
DEFAULT_DATABASE_URL = "sqlite+aiosqlite:///./data/discovery.db"
CURRENT_REVISION = "001"
REQUIRED_TABLES = {"discovery_conversation", "discovery_profile"}
def _sqlite_path(database_url: str) -> str | None:
if not database_url.startswith("sqlite") or ":///" not in database_url:
return None
return database_url.split(":///", 1)[1]
def stamp_existing_sqlite_schema(
database_url: str, revision: str = CURRENT_REVISION
) -> bool:
"""Stamp a pre-Alembic SQLite DB when it already has the app tables.
Early local/dev runs could create tables through SQLAlchemy create_all()
before Alembic was applied. In that case, `alembic upgrade head` tries to
create tables that already exist. This marks that compatible schema as
revision 001 so normal migrations can continue.
"""
path = _sqlite_path(database_url)
if not path or not os.path.exists(path):
return False
with sqlite3.connect(path) as conn:
rows = conn.execute(
"SELECT name FROM sqlite_master WHERE type = 'table'"
).fetchall()
tables = {row[0] for row in rows}
if not REQUIRED_TABLES.issubset(tables):
return False
if "alembic_version" in tables:
versions = conn.execute(
"SELECT version_num FROM alembic_version"
).fetchall()
if versions:
return False
else:
conn.execute(
"CREATE TABLE alembic_version "
"(version_num VARCHAR(32) NOT NULL)"
)
conn.execute(
"INSERT INTO alembic_version (version_num) VALUES (?)",
(revision,),
)
conn.commit()
return True
def main() -> None:
load_dotenv()
database_url = os.getenv("DATABASE_URL", DEFAULT_DATABASE_URL)
stamped = stamp_existing_sqlite_schema(database_url)
if stamped:
print(f"Stamped existing SQLite schema as Alembic revision {CURRENT_REVISION}")
if __name__ == "__main__":
main()