first commit
This commit is contained in:
commit
93b8b89639
15 changed files with 1967 additions and 0 deletions
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
|
|
@ -0,0 +1 @@
|
|||
__pycache__/
|
||||
18
Dockerfile
Normal file
18
Dockerfile
Normal file
|
|
@ -0,0 +1,18 @@
|
|||
FROM python:3.13-slim AS base
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||
gcc libpq-dev \
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
COPY app/requirements.txt .
|
||||
RUN pip install --no-cache-dir -r requirements.txt
|
||||
|
||||
COPY app/ ./app/
|
||||
COPY alembic.ini ./alembic.ini
|
||||
COPY alembic/ ./alembic/
|
||||
|
||||
EXPOSE 8000
|
||||
ENV PYTHONPATH=/app
|
||||
CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000"]
|
||||
37
alembic.ini
Normal file
37
alembic.ini
Normal file
|
|
@ -0,0 +1,37 @@
|
|||
[alembic]
|
||||
script_location = alembic
|
||||
# Override at runtime via DATABASE_URL environment variable (set in ConfigMap/Deployment)
|
||||
sqlalchemy.url = driver://user:pass@localhost/dbname
|
||||
|
||||
[loggers]
|
||||
keys = root,sqlalchemy,alembic
|
||||
|
||||
[handlers]
|
||||
keys = console
|
||||
|
||||
[formatters]
|
||||
keys = generic
|
||||
|
||||
[logger_root]
|
||||
level = WARN
|
||||
handlers = console
|
||||
|
||||
[logger_sqlalchemy]
|
||||
level = WARN
|
||||
handlers =
|
||||
qualname = sqlalchemy.engine
|
||||
|
||||
[logger_alembic]
|
||||
level = INFO
|
||||
handlers =
|
||||
qualname = alembic
|
||||
|
||||
[handler_console]
|
||||
class = StreamHandler
|
||||
args = (sys.stderr,)
|
||||
level = NOTSET
|
||||
formatter = generic
|
||||
|
||||
[formatter_generic]
|
||||
format = %(levelname)-5.5s [%(name)s] %(message)s
|
||||
datefmt = %H:%M:%S
|
||||
55
alembic/env.py
Normal file
55
alembic/env.py
Normal file
|
|
@ -0,0 +1,55 @@
|
|||
import sys
|
||||
from logging.config import fileConfig
|
||||
from pathlib import Path
|
||||
|
||||
from sqlalchemy import pool
|
||||
from sqlalchemy.ext.asyncio import async_engine_from_config
|
||||
|
||||
from alembic import context
|
||||
|
||||
# Add app directory to path so we can import models/database
|
||||
sys.path.insert(0, str(Path(__file__).parent.parent / "app"))
|
||||
|
||||
from database import metadata, DATABASE_URL
|
||||
|
||||
config = context.config
|
||||
if config.config_file_name is not None:
|
||||
fileConfig(config.config_file_name)
|
||||
target_metadata = metadata
|
||||
|
||||
|
||||
def run_migrations_offline() -> None:
|
||||
"""Run migrations in 'offline' mode."""
|
||||
url = config.get_main_option("sqlalchemy.url")
|
||||
context.configure(url=url, target_metadata=target_metadata, literal_binds=True)
|
||||
with context.begin_transaction():
|
||||
context.run_migrations()
|
||||
|
||||
|
||||
def do_run_migrations(connection):
|
||||
context.configure(connection=connection, target_metadata=target_metadata)
|
||||
with context.begin_transaction():
|
||||
context.run_migrations()
|
||||
|
||||
|
||||
async def run_async_migrations():
|
||||
connectable = async_engine_from_config(
|
||||
config.get_section(config.config_ini_section, {}),
|
||||
prefix="sqlalchemy.",
|
||||
poolclass=pool.NullPool,
|
||||
)
|
||||
async with connectable.connect() as connection:
|
||||
await connection.run_sync(do_run_migrations)
|
||||
await connectable.dispose()
|
||||
|
||||
|
||||
def run_migrations_online() -> None:
|
||||
"""Run migrations in 'online' mode."""
|
||||
import asyncio
|
||||
asyncio.run(run_async_migrations())
|
||||
|
||||
|
||||
if context.is_offline_mode():
|
||||
run_migrations_offline()
|
||||
else:
|
||||
run_migrations_online()
|
||||
28
alembic/script.py.mako
Normal file
28
alembic/script.py.mako
Normal file
|
|
@ -0,0 +1,28 @@
|
|||
<%%doc>Template for rendering a Multiple Migration Revision Identifier.</%%doc>
|
||||
<%%-
|
||||
from alembic import context
|
||||
context.configure()
|
||||
-%>
|
||||
"""${message}
|
||||
|
||||
Revision ID: ${up_revision}
|
||||
Revises: ${down_revision | comma_n, trim}
|
||||
Create Date: ${create_date}
|
||||
"""
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
${imports if imports else ""}
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision = ${repr(up_revision)}
|
||||
down_revision = ${repr(down_revision)}
|
||||
branch_labels = ${repr(branch_labels)}
|
||||
depends_on = ${repr(depends_on)}
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
${upgrades if upgrades else "pass"}
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
${downgrades if downgrades else "pass"}
|
||||
149
alembic/versions/001_initial.py
Normal file
149
alembic/versions/001_initial.py
Normal file
|
|
@ -0,0 +1,149 @@
|
|||
"""initial schema
|
||||
|
||||
Revision ID: 001_initial
|
||||
Revises:
|
||||
Create Date: 2026-05-18
|
||||
"""
|
||||
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
from sqlalchemy.dialects.postgresql import UUID, TSVECTOR, ENUM
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision = '001_initial'
|
||||
down_revision = None
|
||||
branch_labels = None
|
||||
depends_on = None
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
# Enums
|
||||
op.execute("CREATE TYPE feed_source_type AS ENUM ('rss', 'gdel-t2', 'social', 'earthquake', 'disaster', 'weather', 'fire', 'satellite')")
|
||||
op.execute("CREATE TYPE event_source_type AS ENUM ('rss', 'gdel-t2', 'social', 'earthquake', 'disaster', 'weather', 'fire', 'satellite')")
|
||||
op.execute("CREATE TYPE sentiment_label AS ENUM ('positive', 'neutral', 'negative')")
|
||||
op.execute("CREATE TYPE entity_type AS ENUM ('person', 'organization', 'location', 'topic', 'asset')")
|
||||
op.execute("CREATE TYPE alert_type AS ENUM ('entity_mention', 'sentiment_shift', 'geo_proximity', 'keyword_match', 'threshold', 'anomaly')")
|
||||
op.execute("CREATE TYPE alert_severity AS ENUM ('low', 'medium', 'high', 'critical')")
|
||||
|
||||
# Extensions
|
||||
op.execute("CREATE EXTENSION IF NOT EXISTS postgis")
|
||||
op.execute("CREATE EXTENSION IF NOT EXISTS timescaledb")
|
||||
|
||||
# feed_sources
|
||||
op.create_table(
|
||||
'feed_sources',
|
||||
sa.Column('id', UUID(as_uuid=True), primary_key=True),
|
||||
sa.Column('name', sa.String(256), nullable=False),
|
||||
sa.Column('source_type', sa.Enum('rss', 'gdel-t2', 'social', 'earthquake', 'disaster', 'weather', 'fire', 'satellite', name='feed_source_type'), nullable=False),
|
||||
sa.Column('url', sa.Text()),
|
||||
sa.Column('config', sa.JSON()),
|
||||
sa.Column('enabled', sa.Integer, server_default='1', nullable=False),
|
||||
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
||||
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
||||
)
|
||||
|
||||
# events (will become hypertable)
|
||||
op.create_table(
|
||||
'events',
|
||||
sa.Column('id', UUID(as_uuid=True), primary_key=True),
|
||||
sa.Column('source_type', sa.Enum('rss', 'gdel-t2', 'social', 'earthquake', 'disaster', 'weather', 'fire', 'satellite', name='event_source_type'), nullable=False, index=True),
|
||||
sa.Column('source_id', UUID(as_uuid=True)),
|
||||
sa.Column('title', sa.Text()),
|
||||
sa.Column('body', sa.Text()),
|
||||
sa.Column('url', sa.Text()),
|
||||
sa.Column('sentiment_score', sa.Float()),
|
||||
sa.Column('sentiment_label', sa.Enum('positive', 'neutral', 'negative', name='sentiment_label')),
|
||||
sa.Column('location_lat', sa.Float()),
|
||||
sa.Column('location_lon', sa.Float()),
|
||||
sa.Column('location_name', sa.String(512)),
|
||||
sa.Column('entities', sa.JSON()),
|
||||
sa.Column('tags', sa.JSON()),
|
||||
sa.Column('raw', sa.JSON()),
|
||||
sa.Column('ingested_at', sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
||||
sa.Column('source_timestamp', sa.DateTime(timezone=True), nullable=False),
|
||||
sa.Column('search_vector', TSVECTOR),
|
||||
)
|
||||
|
||||
# Convert events to TimescaleDB hypertable
|
||||
op.execute("SELECT create_hypertable('events', 'ingested_at', if_not_exists => TRUE)")
|
||||
|
||||
# GIN index for full-text search
|
||||
op.create_index('ix_events_search_vector', 'events', ['search_vector'], postgresql_using='gin')
|
||||
# Spatial index
|
||||
op.create_index('ix_events_location', 'events', ['location_lat', 'location_lon'])
|
||||
|
||||
# entities
|
||||
op.create_table(
|
||||
'entities',
|
||||
sa.Column('id', UUID(as_uuid=True), primary_key=True),
|
||||
sa.Column('name', sa.String(512), nullable=False, index=True),
|
||||
sa.Column('entity_type', sa.Enum('person', 'organization', 'location', 'topic', 'asset', name='entity_type'), nullable=False),
|
||||
sa.Column('aliases', sa.JSON()),
|
||||
sa.Column('description', sa.Text()),
|
||||
sa.Column('metadata', sa.JSON()),
|
||||
sa.Column('location_lat', sa.Float()),
|
||||
sa.Column('location_lon', sa.Float()),
|
||||
sa.Column('event_count', sa.Integer, server_default='0'),
|
||||
sa.Column('first_seen', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
||||
sa.Column('last_seen', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
||||
)
|
||||
|
||||
# entity_events
|
||||
op.create_table(
|
||||
'entity_events',
|
||||
sa.Column('entity_id', UUID(as_uuid=True), primary_key=True),
|
||||
sa.Column('event_id', UUID(as_uuid=True), primary_key=True),
|
||||
sa.Column('relevance_score', sa.Float()),
|
||||
sa.Column('linked_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
||||
)
|
||||
|
||||
# alerts
|
||||
op.create_table(
|
||||
'alerts',
|
||||
sa.Column('id', UUID(as_uuid=True), primary_key=True),
|
||||
sa.Column('alert_type', sa.Enum('entity_mention', 'sentiment_shift', 'geo_proximity', 'keyword_match', 'threshold', 'anomaly', name='alert_type'), nullable=False),
|
||||
sa.Column('entity_id', UUID(as_uuid=True)),
|
||||
sa.Column('event_id', UUID(as_uuid=True)),
|
||||
sa.Column('severity', sa.Enum('low', 'medium', 'high', 'critical', name='alert_severity'), nullable=False),
|
||||
sa.Column('title', sa.Text(), nullable=False),
|
||||
sa.Column('message', sa.Text()),
|
||||
sa.Column('context', sa.JSON()),
|
||||
sa.Column('acknowledged', sa.Integer, server_default='0'),
|
||||
sa.Column('acknowledged_by', sa.String(256)),
|
||||
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
||||
sa.Column('resolved_at', sa.DateTime(timezone=True)),
|
||||
)
|
||||
op.create_index('ix_alerts_severity_created', 'alerts', ['severity', 'created_at'])
|
||||
op.create_index('ix_alerts_entity', 'alerts', ['entity_id'])
|
||||
|
||||
# documents
|
||||
op.create_table(
|
||||
'documents',
|
||||
sa.Column('id', UUID(as_uuid=True), primary_key=True),
|
||||
sa.Column('bucket', sa.String(256), nullable=False),
|
||||
sa.Column('object_key', sa.String(1024), nullable=False),
|
||||
sa.Column('content_type', sa.String(256)),
|
||||
sa.Column('size_bytes', sa.Integer()),
|
||||
sa.Column('description', sa.Text()),
|
||||
sa.Column('tags', sa.JSON()),
|
||||
sa.Column('event_id', UUID(as_uuid=True)),
|
||||
sa.Column('uploaded_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
||||
)
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
op.drop_table('documents')
|
||||
op.drop_table('alerts')
|
||||
op.drop_table('entity_events')
|
||||
op.drop_table('entities')
|
||||
op.execute("SELECT drop_hypertable('events', cascade => TRUE)")
|
||||
op.drop_table('events')
|
||||
op.drop_table('feed_sources')
|
||||
|
||||
# Drop enums
|
||||
op.execute("DROP TYPE IF EXISTS feed_source_type")
|
||||
op.execute("DROP TYPE IF EXISTS event_source_type")
|
||||
op.execute("DROP TYPE IF EXISTS sentiment_label")
|
||||
op.execute("DROP TYPE IF EXISTS entity_type")
|
||||
op.execute("DROP TYPE IF EXISTS alert_type")
|
||||
op.execute("DROP TYPE IF EXISTS alert_severity")
|
||||
28
app/database.py
Normal file
28
app/database.py
Normal file
|
|
@ -0,0 +1,28 @@
|
|||
import os
|
||||
|
||||
from sqlalchemy import MetaData, event, text
|
||||
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
|
||||
|
||||
DB_USER = os.getenv("DB_USER", "osint")
|
||||
DB_PASS = os.getenv("DB_PASSWORD", "")
|
||||
DB_HOST = os.getenv("DB_HOST", "osint-pgdb-rw.customer1.svc.cluster.local")
|
||||
DB_PORT = os.getenv("DB_PORT", "5432")
|
||||
DB_NAME = os.getenv("DB_NAME", "osint_data")
|
||||
|
||||
DATABASE_URL = f"postgresql+asyncpg://{DB_USER}:{DB_PASS}@{DB_HOST}:{DB_PORT}/{DB_NAME}"
|
||||
|
||||
engine = create_async_engine(
|
||||
DATABASE_URL, echo=False, pool_size=5, max_overflow=10, pool_recycle=300
|
||||
)
|
||||
async_session = async_sessionmaker(
|
||||
engine, class_=AsyncSession, expire_on_commit=False
|
||||
)
|
||||
metadata = MetaData()
|
||||
|
||||
|
||||
async def init_extensions():
|
||||
"""Initialize PostGIS and TimescaleDB extensions on first connection."""
|
||||
async with engine.connect() as conn:
|
||||
await conn.execute(text("CREATE EXTENSION IF NOT EXISTS postgis"))
|
||||
await conn.execute(text("CREATE EXTENSION IF NOT EXISTS timescaledb"))
|
||||
await conn.commit()
|
||||
46
app/ingest_cron.py
Normal file
46
app/ingest_cron.py
Normal file
|
|
@ -0,0 +1,46 @@
|
|||
"""CronJob entry point for scheduled ingestion."""
|
||||
|
||||
import asyncio
|
||||
import os
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
# Add app dir to path
|
||||
sys.path.insert(0, str(Path(__file__).parent))
|
||||
|
||||
from sources import ingest_rss_feed, ingest_gdelt, ingest_earthquakes
|
||||
from ingestor import fetch_and_process
|
||||
|
||||
INGESTOR_TYPE = os.getenv("INGESTOR_TYPE", "rss")
|
||||
RSS_URL = os.getenv("RSS_URL", "")
|
||||
|
||||
|
||||
async def main():
|
||||
print(f"Starting ingester: {INGESTOR_TYPE}")
|
||||
|
||||
if INGESTOR_TYPE == "rss":
|
||||
if not RSS_URL:
|
||||
print("No RSS_URL set, skipping")
|
||||
return
|
||||
count = await ingest_rss_feed(RSS_URL)
|
||||
print(f"RSS: ingested {count} items")
|
||||
|
||||
elif INGESTOR_TYPE == "gdelt":
|
||||
count = await ingest_gdelt(max_articles=50)
|
||||
print(f"GDELT: ingested {count} articles")
|
||||
|
||||
elif INGESTOR_TYPE == "earthquake":
|
||||
count = await ingest_earthquakes()
|
||||
print(f"Earthquakes: ingested {count} events")
|
||||
|
||||
elif INGESTOR_TYPE == "nats":
|
||||
count = await fetch_and_process(batch_size=500)
|
||||
print(f"NATS: processed {count} messages")
|
||||
|
||||
else:
|
||||
print(f"Unknown ingestor type: {INGESTOR_TYPE}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
107
app/ingestor.py
Normal file
107
app/ingestor.py
Normal file
|
|
@ -0,0 +1,107 @@
|
|||
"""NATS JetStream consumer — ingests OSINT events from NATS streams."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import logging
|
||||
from datetime import datetime, timezone
|
||||
|
||||
import nats
|
||||
from nats.errors import TimeoutError
|
||||
|
||||
from database import async_session
|
||||
from models import events as events_table
|
||||
|
||||
logger = logging.getLogger("osint.ingestor")
|
||||
|
||||
# NATS connection settings
|
||||
NATS_URLS = "nats://osint-nats.customer1.svc.cluster.local:4222"
|
||||
NATS_STREAM = "events"
|
||||
NATS_DURABLE = "osint-ingestor"
|
||||
|
||||
# Redis cache settings
|
||||
REDIS_URL = "redis://osint-redis-sentinel.customer1.svc.cluster.local:26379/0"
|
||||
|
||||
|
||||
async def ingest_event(msg: dict):
|
||||
"""Ingest a single event from NATS into PostgreSQL."""
|
||||
event_row = {
|
||||
"source_type": msg.get("source_type", "rss"),
|
||||
"source_id": msg.get("source_id"),
|
||||
"title": msg.get("title"),
|
||||
"body": msg.get("body"),
|
||||
"url": msg.get("url"),
|
||||
"sentiment_score": msg.get("sentiment_score"),
|
||||
"sentiment_label": msg.get("sentiment_label"),
|
||||
"location_lat": msg.get("location_lat"),
|
||||
"location_lon": msg.get("location_lon"),
|
||||
"location_name": msg.get("location_name"),
|
||||
"entities": msg.get("entities", []),
|
||||
"tags": msg.get("tags", []),
|
||||
"raw": msg.get("raw"),
|
||||
"source_timestamp": msg.get("source_timestamp", datetime.now(timezone.utc).isoformat()),
|
||||
}
|
||||
|
||||
# Parse timestamp if string
|
||||
if isinstance(event_row["source_timestamp"], str):
|
||||
event_row["source_timestamp"] = datetime.fromisoformat(event_row["source_timestamp"])
|
||||
|
||||
async with async_session() as session:
|
||||
result = await session.execute(events_table.insert().values(**event_row))
|
||||
await session.commit()
|
||||
event_id = result.inserted_primary_key[0] # type: ignore[union-attr]
|
||||
logger.info("Ingested event %s from source %s", event_id, msg.get("source_type"))
|
||||
return event_id
|
||||
|
||||
|
||||
async def start_nats_consumer():
|
||||
"""Start NATS JetStream consumer for OSINT events."""
|
||||
nc = await nats.connect(NATS_URLS)
|
||||
js = nc.jetstream()
|
||||
|
||||
# Create stream if not exists
|
||||
try:
|
||||
await js.add_stream(
|
||||
name=NATS_STREAM,
|
||||
subjects=[
|
||||
"events.gdelt", "events.rss", "events.social",
|
||||
"events.earthquake", "events.disaster", "events.weather",
|
||||
"events.fire", "events.satellite", "events.new", "events.alert",
|
||||
],
|
||||
retention=nats.js.api.RetentionPolicy.INTERESTS,
|
||||
max_msgs=1_000_000,
|
||||
)
|
||||
logger.info("Created NATS stream %s", NATS_STREAM)
|
||||
except Exception:
|
||||
logger.debug("Stream %s already exists", NATS_STREAM)
|
||||
|
||||
# Create durable consumer
|
||||
sub = await js.pull_subscribe(
|
||||
subject="events.>",
|
||||
durable_name=NATS_DURABLE,
|
||||
)
|
||||
|
||||
logger.info("NATS consumer started, durable=%s", NATS_DURABLE)
|
||||
return nc, sub
|
||||
|
||||
|
||||
async def fetch_and_process(batch_size: int = 100):
|
||||
"""Fetch a batch of messages and process them."""
|
||||
nc, sub = await start_nats_consumer()
|
||||
js = nc.jetstream()
|
||||
|
||||
msgs = await sub.fetch(batch_size, timeout=5)
|
||||
processed = 0
|
||||
|
||||
for msg in msgs:
|
||||
try:
|
||||
data = json.loads(msg.data)
|
||||
await ingest_event(data)
|
||||
await msg.ack()
|
||||
processed += 1
|
||||
except Exception:
|
||||
logger.error("Failed to process message: %s", msg.data, exc_info=True)
|
||||
|
||||
await nc.close()
|
||||
logger.info("Processed %d messages in batch", processed)
|
||||
return processed
|
||||
603
app/main.py
Normal file
603
app/main.py
Normal file
|
|
@ -0,0 +1,603 @@
|
|||
"""OSINT Dashboard — FastAPI backend.
|
||||
|
||||
Real-time geospatial OSINT dashboard API:
|
||||
- Event ingestion via NATS JetStream consumers
|
||||
- Full-text search across events (PostgreSQL tsvector)
|
||||
- Entity tracking and relationship mapping
|
||||
- Alert management
|
||||
- Document storage (MinIO-backed)
|
||||
- Sentiment aggregation and timeline analytics
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import logging
|
||||
from datetime import datetime, timedelta, timezone
|
||||
from decimal import Decimal
|
||||
from pathlib import Path
|
||||
from uuid import UUID
|
||||
|
||||
import structlog
|
||||
from fastapi import FastAPI, HTTPException, Query
|
||||
from fastapi.responses import FileResponse, HTMLResponse
|
||||
from sqlalchemy import and_, func, select, text
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from database import async_session, init_extensions
|
||||
from models import (
|
||||
alerts, documents, entities, entity_events, events, feed_sources
|
||||
)
|
||||
from schemas import (
|
||||
AlertCreate, AlertOut, AlertSeverity, AlertType, AlertUpdate,
|
||||
DashboardSummary, EntityCreate, EntityKind, EntityOut,
|
||||
EventCreate, EventOut,
|
||||
FeedSourceCreate, FeedSourceOut,
|
||||
SearchResult, SentimentSummary, SourceType,
|
||||
SearchQuery, TimelinePoint,
|
||||
)
|
||||
from ingestor import ingest_event, fetch_and_process
|
||||
from sources import ingest_rss_feed, ingest_gdelt, ingest_earthquakes, ingest_social_signals
|
||||
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
logger = structlog.get_logger("osint.dashboard")
|
||||
|
||||
app = FastAPI(
|
||||
title="OSINT Dashboard",
|
||||
description="Real-time geospatial OSINT intelligence dashboard",
|
||||
version="0.1.0",
|
||||
)
|
||||
|
||||
STATIC_DIR = Path(__file__).parent / "static"
|
||||
|
||||
|
||||
# ── Helpers ───────────────────────────────────────────────────────────────
|
||||
|
||||
def event_to_out(row: dict) -> EventOut:
|
||||
"""Convert DB row dict to EventOut schema."""
|
||||
return EventOut(
|
||||
id=row["id"],
|
||||
source_type=row["source_type"],
|
||||
source_id=row["source_id"],
|
||||
title=row["title"],
|
||||
body=row["body"],
|
||||
url=row["url"],
|
||||
sentiment_score=row["sentiment_score"],
|
||||
sentiment_label=row["sentiment_label"],
|
||||
location_lat=row["location_lat"],
|
||||
location_lon=row["location_lon"],
|
||||
location_name=row["location_name"],
|
||||
entities=row["entities"],
|
||||
tags=row["tags"],
|
||||
ingested_at=row["ingested_at"],
|
||||
source_timestamp=row["source_timestamp"],
|
||||
)
|
||||
|
||||
|
||||
def entity_to_out(row: dict) -> EntityOut:
|
||||
"""Convert DB row dict to EntityOut schema."""
|
||||
return EntityOut(
|
||||
id=row["id"],
|
||||
name=row["name"],
|
||||
entity_type=row["entity_type"],
|
||||
aliases=row["aliases"],
|
||||
description=row["description"],
|
||||
metadata=row["metadata"],
|
||||
location_lat=row["location_lat"],
|
||||
location_lon=row["location_lon"],
|
||||
event_count=row["event_count"],
|
||||
first_seen=row["first_seen"],
|
||||
last_seen=row["last_seen"],
|
||||
)
|
||||
|
||||
|
||||
def alert_to_out(row: dict) -> AlertOut:
|
||||
"""Convert DB row dict to AlertOut schema."""
|
||||
return AlertOut(
|
||||
id=row["id"],
|
||||
alert_type=row["alert_type"],
|
||||
entity_id=row["entity_id"],
|
||||
event_id=row["event_id"],
|
||||
severity=row["severity"],
|
||||
title=row["title"],
|
||||
message=row["message"],
|
||||
context=row["context"],
|
||||
acknowledged=bool(row["acknowledged"]),
|
||||
acknowledged_by=row["acknowledged_by"],
|
||||
created_at=row["created_at"],
|
||||
resolved_at=row["resolved_at"],
|
||||
)
|
||||
|
||||
|
||||
# ── Health ────────────────────────────────────────────────────────────────
|
||||
|
||||
@app.get("/api/health")
|
||||
async def health():
|
||||
"""Health check with database connectivity."""
|
||||
async with async_session() as session:
|
||||
result = await session.execute(select(func.now()))
|
||||
db_time = result.scalar()
|
||||
return {"status": "ok", "db_time": db_time.isoformat() if db_time else None}
|
||||
|
||||
|
||||
# ── Startup ───────────────────────────────────────────────────────────────
|
||||
|
||||
@app.on_event("startup")
|
||||
async def startup():
|
||||
"""Initialize extensions and run migrations."""
|
||||
await init_extensions()
|
||||
from alembic import command
|
||||
from alembic.config import Config
|
||||
alembic_cfg = Config(str(Path(__file__).parent.parent / "alembic.ini"))
|
||||
command.upgrade(alembic_cfg, "head")
|
||||
|
||||
|
||||
# ── Feed Sources ──────────────────────────────────────────────────────────
|
||||
|
||||
@app.get("/api/sources", response_model=list[FeedSourceOut])
|
||||
async def list_sources(enabled_only: bool = Query(True)):
|
||||
"""List all configured feed sources."""
|
||||
async with async_session() as session:
|
||||
stmt = select(feed_sources).order_by(feed_sources.c.name)
|
||||
if enabled_only:
|
||||
stmt = stmt.where(feed_sources.c.enabled == 1)
|
||||
rows = (await session.execute(stmt)).mappings().all()
|
||||
return [FeedSourceOut(
|
||||
id=r["id"], name=r["name"], source_type=r["source_type"],
|
||||
url=r["url"], config=r["config"], enabled=bool(r["enabled"]),
|
||||
created_at=r["created_at"],
|
||||
) for r in rows]
|
||||
|
||||
|
||||
@app.post("/api/sources", status_code=201)
|
||||
async def create_source(payload: FeedSourceCreate):
|
||||
"""Add a new feed source."""
|
||||
async with async_session() as session:
|
||||
values = payload.model_dump()
|
||||
result = await session.execute(feed_sources.insert().values(**values))
|
||||
await session.commit()
|
||||
pk = result.inserted_primary_key[0] # type: ignore
|
||||
return {"id": str(pk)}
|
||||
|
||||
|
||||
@app.patch("/api/sources/{source_id}")
|
||||
async def update_source(source_id: UUID, payload: dict):
|
||||
"""Update a feed source (e.g., toggle enabled)."""
|
||||
async with async_session() as session:
|
||||
row = (await session.execute(
|
||||
select(feed_sources).where(feed_sources.c.id == source_id)
|
||||
)).mappings().one_or_none()
|
||||
if not row:
|
||||
raise HTTPException(404, "Source not found")
|
||||
await session.execute(
|
||||
feed_sources.update()
|
||||
.where(feed_sources.c.id == source_id)
|
||||
.values(**payload)
|
||||
)
|
||||
await session.commit()
|
||||
return {"ok": True}
|
||||
|
||||
|
||||
# ── Events ────────────────────────────────────────────────────────────────
|
||||
|
||||
@app.get("/api/events", response_model=list[EventOut])
|
||||
async def list_events(
|
||||
source_type: SourceType | None = Query(None),
|
||||
limit: int = Query(50, ge=1, le=500),
|
||||
offset: int = Query(0, ge=0),
|
||||
):
|
||||
"""List recent ingested events."""
|
||||
async with async_session() as session:
|
||||
stmt = select(events).order_by(events.c.ingested_at.desc())
|
||||
if source_type:
|
||||
stmt = stmt.where(events.c.source_type == source_type.value)
|
||||
stmt = stmt.limit(limit).offset(offset)
|
||||
rows = (await session.execute(stmt)).mappings().all()
|
||||
return [event_to_out(r) for r in rows]
|
||||
|
||||
|
||||
@app.get("/api/events/{event_id}", response_model=EventOut)
|
||||
async def get_event(event_id: UUID):
|
||||
"""Get a single event by ID."""
|
||||
async with async_session() as session:
|
||||
row = (await session.execute(
|
||||
select(events).where(events.c.id == event_id)
|
||||
)).mappings().one_or_none()
|
||||
if not row:
|
||||
raise HTTPException(404, "Event not found")
|
||||
return event_to_out(row)
|
||||
|
||||
|
||||
@app.post("/api/events", status_code=201)
|
||||
async def create_event(payload: EventCreate):
|
||||
"""Manually ingest an event (bypasses NATS)."""
|
||||
values = payload.model_dump(exclude_unset=True)
|
||||
if not values.get("source_timestamp"):
|
||||
values["source_timestamp"] = datetime.now(timezone.utc)
|
||||
event_id = await ingest_event(values)
|
||||
return {"id": str(event_id)}
|
||||
|
||||
|
||||
# ── Search ────────────────────────────────────────────────────────────────
|
||||
|
||||
@app.post("/api/search", response_model=SearchResult)
|
||||
async def search_events(query: SearchQuery):
|
||||
"""Full-text search across events with optional filters."""
|
||||
async with async_session() as session:
|
||||
# Build query with tsvector full-text search (parameterized to avoid SQL injection)
|
||||
tsquery_param = text("plainto_tsquery('english', :q)")
|
||||
|
||||
base_stmt = select(
|
||||
events,
|
||||
func.count().over().label("total")
|
||||
).where(
|
||||
events.c.search_vector.op("@@")(tsquery_param)
|
||||
)
|
||||
|
||||
# Apply filters
|
||||
if query.source_type:
|
||||
base_stmt = base_stmt.where(events.c.source_type == query.source_type.value)
|
||||
if query.entity_id:
|
||||
base_stmt = base_stmt.join(
|
||||
entity_events, entity_events.c.event_id == events.c.id
|
||||
).where(entity_events.c.entity_id == query.entity_id)
|
||||
if query.sentiment:
|
||||
base_stmt = base_stmt.where(events.c.sentiment_label == query.sentiment.value)
|
||||
if query.min_date:
|
||||
base_stmt = base_stmt.where(events.c.source_timestamp >= query.min_date)
|
||||
if query.max_date:
|
||||
base_stmt = base_stmt.where(events.c.source_timestamp <= query.max_date)
|
||||
if query.min_lat is not None and query.max_lat is not None:
|
||||
base_stmt = base_stmt.where(
|
||||
and_(
|
||||
events.c.location_lat >= query.min_lat,
|
||||
events.c.location_lat <= query.max_lat,
|
||||
)
|
||||
)
|
||||
if query.min_lon is not None and query.max_lon is not None:
|
||||
base_stmt = base_stmt.where(
|
||||
and_(
|
||||
events.c.location_lon >= query.min_lon,
|
||||
events.c.location_lon <= query.max_lon,
|
||||
)
|
||||
)
|
||||
|
||||
base_stmt = base_stmt.order_by(events.c.ingested_at.desc())
|
||||
base_stmt = base_stmt.limit(query.limit).offset(query.offset)
|
||||
|
||||
result = (await session.execute(base_stmt, {"q": query.q})).mappings().all()
|
||||
if result:
|
||||
total = result[0]["total"]
|
||||
else:
|
||||
total = 0
|
||||
evts = [event_to_out(r) for r in result]
|
||||
|
||||
return SearchResult(
|
||||
events=evts,
|
||||
total=total,
|
||||
has_more=query.offset + len(evts) < total,
|
||||
)
|
||||
|
||||
|
||||
# ── Entities ──────────────────────────────────────────────────────────────
|
||||
|
||||
@app.get("/api/entities", response_model=list[EntityOut])
|
||||
async def list_entities(
|
||||
entity_type: EntityKind | None = Query(None),
|
||||
limit: int = Query(50, ge=1, le=500),
|
||||
):
|
||||
"""List tracked entities."""
|
||||
async with async_session() as session:
|
||||
stmt = select(entities).order_by(entities.c.event_count.desc())
|
||||
if entity_type:
|
||||
stmt = stmt.where(entities.c.entity_type == entity_type.value)
|
||||
stmt = stmt.limit(limit)
|
||||
rows = (await session.execute(stmt)).mappings().all()
|
||||
return [entity_to_out(r) for r in rows]
|
||||
|
||||
|
||||
@app.get("/api/entities/{entity_id}", response_model=EntityOut)
|
||||
async def get_entity(entity_id: UUID):
|
||||
"""Get entity details with recent events."""
|
||||
async with async_session() as session:
|
||||
row = (await session.execute(
|
||||
select(entities).where(entities.c.id == entity_id)
|
||||
)).mappings().one_or_none()
|
||||
if not row:
|
||||
raise HTTPException(404, "Entity not found")
|
||||
return entity_to_out(row)
|
||||
|
||||
|
||||
@app.post("/api/entities", status_code=201)
|
||||
async def create_entity(payload: EntityCreate):
|
||||
"""Create or update a tracked entity."""
|
||||
async with async_session() as session:
|
||||
# Check if entity already exists by name
|
||||
existing = (await session.execute(
|
||||
select(entities).where(entities.c.name == payload.name)
|
||||
)).mappings().one_or_none()
|
||||
|
||||
if existing:
|
||||
# Update
|
||||
updates = payload.model_dump(exclude_unset=True)
|
||||
updates["last_seen"] = datetime.now(timezone.utc)
|
||||
await session.execute(
|
||||
entities.update()
|
||||
.where(entities.c.id == existing["id"])
|
||||
.values(**updates)
|
||||
)
|
||||
await session.commit()
|
||||
return {"id": str(existing["id"]), "created": False}
|
||||
|
||||
# Create
|
||||
values = payload.model_dump()
|
||||
result = await session.execute(entities.insert().values(**values))
|
||||
await session.commit()
|
||||
pk = result.inserted_primary_key[0] # type: ignore
|
||||
return {"id": str(pk), "created": True}
|
||||
|
||||
|
||||
@app.get("/api/entities/{entity_id}/events", response_model=list[EventOut])
|
||||
async def get_entity_events(
|
||||
entity_id: UUID,
|
||||
limit: int = Query(50, ge=1, le=500),
|
||||
):
|
||||
"""Get events linked to a specific entity."""
|
||||
async with async_session() as session:
|
||||
stmt = (
|
||||
select(events)
|
||||
.join(entity_events, entity_events.c.event_id == events.c.id)
|
||||
.where(entity_events.c.entity_id == entity_id)
|
||||
.order_by(events.c.source_timestamp.desc())
|
||||
.limit(limit)
|
||||
)
|
||||
rows = (await session.execute(stmt)).mappings().all()
|
||||
return [event_to_out(r) for r in rows]
|
||||
|
||||
|
||||
# ── Alerts ────────────────────────────────────────────────────────────────
|
||||
|
||||
@app.get("/api/alerts", response_model=list[AlertOut])
|
||||
async def list_alerts(
|
||||
severity: AlertSeverity | None = Query(None),
|
||||
acknowledged: bool | None = Query(None),
|
||||
entity_id: UUID | None = Query(None),
|
||||
limit: int = Query(50, ge=1, le=500),
|
||||
):
|
||||
"""List alerts with optional filters."""
|
||||
async with async_session() as session:
|
||||
stmt = select(alerts).order_by(
|
||||
alerts.c.severity.desc(), alerts.c.created_at.desc()
|
||||
)
|
||||
if severity:
|
||||
stmt = stmt.where(alerts.c.severity == severity.value)
|
||||
if acknowledged is not None:
|
||||
stmt = stmt.where(alerts.c.acknowledged == int(acknowledged))
|
||||
if entity_id:
|
||||
stmt = stmt.where(alerts.c.entity_id == entity_id)
|
||||
stmt = stmt.limit(limit)
|
||||
rows = (await session.execute(stmt)).mappings().all()
|
||||
return [alert_to_out(r) for r in rows]
|
||||
|
||||
|
||||
@app.post("/api/alerts", status_code=201)
|
||||
async def create_alert(payload: AlertCreate):
|
||||
"""Create a new alert."""
|
||||
async with async_session() as session:
|
||||
values = payload.model_dump()
|
||||
result = await session.execute(alerts.insert().values(**values))
|
||||
await session.commit()
|
||||
pk = result.inserted_primary_key[0] # type: ignore
|
||||
return {"id": str(pk)}
|
||||
|
||||
|
||||
@app.patch("/api/alerts/{alert_id}")
|
||||
async def update_alert(alert_id: UUID, payload: AlertUpdate):
|
||||
"""Update alert (acknowledge, resolve)."""
|
||||
async with async_session() as session:
|
||||
row = (await session.execute(
|
||||
select(alerts).where(alerts.c.id == alert_id)
|
||||
)).mappings().one_or_none()
|
||||
if not row:
|
||||
raise HTTPException(404, "Alert not found")
|
||||
updates = payload.model_dump(exclude_unset=True)
|
||||
if "acknowledged" in updates:
|
||||
updates["acknowledged"] = int(updates["acknowledged"])
|
||||
await session.execute(
|
||||
alerts.update().where(alerts.c.id == alert_id).values(**updates)
|
||||
)
|
||||
await session.commit()
|
||||
return {"ok": True}
|
||||
|
||||
|
||||
# ── Documents ─────────────────────────────────────────────────────────────
|
||||
|
||||
@app.get("/api/documents", response_model=dict)
|
||||
async def list_documents(
|
||||
limit: int = Query(50, ge=1, le=500),
|
||||
offset: int = Query(0, ge=0),
|
||||
):
|
||||
"""List documents indexed in MinIO."""
|
||||
async with async_session() as session:
|
||||
stmt = select(documents).order_by(documents.c.uploaded_at.desc()).limit(limit).offset(offset)
|
||||
rows = (await session.execute(stmt)).mappings().all()
|
||||
return {
|
||||
"documents": [{
|
||||
"id": str(r["id"]), "bucket": r["bucket"], "object_key": r["object_key"],
|
||||
"content_type": r["content_type"], "size_bytes": r["size_bytes"],
|
||||
"description": r["description"], "tags": r["tags"],
|
||||
"event_id": str(r["event_id"]) if r["event_id"] else None,
|
||||
"uploaded_at": r["uploaded_at"].isoformat() if r["uploaded_at"] else None,
|
||||
} for r in rows],
|
||||
}
|
||||
|
||||
|
||||
# ── Ingestion Triggers ───────────────────────────────────────────────────
|
||||
|
||||
@app.post("/api/ingest/rss")
|
||||
async def trigger_rss_ingest(feed_url: str, source_id: str | None = None):
|
||||
"""Trigger RSS feed ingestion."""
|
||||
count = await ingest_rss_feed(feed_url, source_id)
|
||||
return {"status": "ok", "items_ingested": count}
|
||||
|
||||
|
||||
@app.post("/api/ingest/gdelt")
|
||||
async def trigger_gdelt_ingest(query: str = "", max_articles: int = 50):
|
||||
"""Trigger GDELT data ingestion."""
|
||||
count = await ingest_gdelt(query, max_articles)
|
||||
return {"status": "ok", "articles_ingested": count}
|
||||
|
||||
|
||||
@app.post("/api/ingest/earthquakes")
|
||||
async def trigger_earthquake_ingest():
|
||||
"""Trigger USGS earthquake ingestion."""
|
||||
count = await ingest_earthquakes()
|
||||
return {"status": "ok", "events_ingested": count}
|
||||
|
||||
|
||||
@app.post("/api/ingest/social")
|
||||
async def trigger_social_ingest(query: str = "", max_items: int = 50):
|
||||
"""Trigger social signals ingestion."""
|
||||
count = await ingest_social_signals(query, max_items)
|
||||
return {"status": "ok", "signals_ingested": count}
|
||||
|
||||
|
||||
@app.post("/api/ingest/process")
|
||||
async def trigger_nats_processing(batch_size: int = 100):
|
||||
"""Process pending NATS JetStream messages."""
|
||||
count = await fetch_and_process(batch_size)
|
||||
return {"status": "ok", "processed": count}
|
||||
|
||||
|
||||
# ── Analytics / Aggregation ───────────────────────────────────────────────
|
||||
|
||||
@app.get("/api/analytics/summary", response_model=DashboardSummary)
|
||||
async def get_dashboard_summary():
|
||||
"""Dashboard overview: event counts, sentiment, top entities, alerts."""
|
||||
async with async_session() as session:
|
||||
now = datetime.now(timezone.utc)
|
||||
yesterday = now - timedelta(hours=24)
|
||||
|
||||
# Total events
|
||||
total = (await session.execute(
|
||||
select(func.count()).select_from(events)
|
||||
)).scalar() or 0
|
||||
|
||||
# Events in last 24h
|
||||
events_24h = (await session.execute(
|
||||
select(func.count()).where(events.c.ingested_at >= yesterday)
|
||||
)).scalar() or 0
|
||||
|
||||
# Active sources
|
||||
active = (await session.execute(
|
||||
select(func.count()).where(feed_sources.c.enabled == 1)
|
||||
)).scalar() or 0
|
||||
|
||||
# Open alerts
|
||||
open_alerts = (await session.execute(
|
||||
select(func.count()).where(alerts.c.acknowledged == 0)
|
||||
)).scalar() or 0
|
||||
|
||||
# Tracked entities
|
||||
ent_count = (await session.execute(
|
||||
select(func.count()).select_from(entities)
|
||||
)).scalar() or 0
|
||||
|
||||
# Sentiment breakdown (last 24h)
|
||||
def sentiment_query():
|
||||
return select(
|
||||
func.count().where(events.c.sentiment_label == "positive").label("pos"),
|
||||
func.count().where(events.c.sentiment_label == "neutral").label("neu"),
|
||||
func.count().where(events.c.sentiment_label == "negative").label("neg"),
|
||||
func.avg(events.c.sentiment_score).label("avg"),
|
||||
).where(events.c.ingested_at >= yesterday)
|
||||
|
||||
sent_row = (await session.execute(sentiment_query())).mappings().one()
|
||||
sentiment = SentimentSummary(
|
||||
period="24h",
|
||||
positive_count=sent_row["pos"] or 0,
|
||||
neutral_count=sent_row["neu"] or 0,
|
||||
negative_count=sent_row["neg"] or 0,
|
||||
avg_score=float(sent_row["avg"] or 0),
|
||||
)
|
||||
|
||||
# Top entities by event count
|
||||
top_ent = (await session.execute(
|
||||
select(entities).order_by(entities.c.event_count.desc()).limit(10)
|
||||
)).mappings().all()
|
||||
|
||||
return DashboardSummary(
|
||||
total_events=total,
|
||||
events_last_24h=events_24h,
|
||||
active_sources=active,
|
||||
open_alerts=open_alerts,
|
||||
tracked_entities=ent_count,
|
||||
sentiment=sentiment,
|
||||
top_entities=[entity_to_out(r) for r in top_ent],
|
||||
)
|
||||
|
||||
|
||||
@app.get("/api/analytics/timeline")
|
||||
async def get_timeline(
|
||||
hours: int = Query(24, ge=1, le=168),
|
||||
bucket_hours: int = Query(1, ge=1, le=24),
|
||||
):
|
||||
"""Event timeline: counts and avg sentiment per time bucket."""
|
||||
async with async_session() as session:
|
||||
cutoff = datetime.now(timezone.utc) - timedelta(hours=hours)
|
||||
# Use date_trunc for bucketing
|
||||
buckets = await session.execute(text(f"""
|
||||
SELECT
|
||||
date_trunc('hour', source_timestamp) AS ts,
|
||||
COUNT(*) AS event_count,
|
||||
COALESCE(AVG(sentiment_score), 0) AS avg_sentiment
|
||||
FROM events
|
||||
WHERE source_timestamp >= :cutoff
|
||||
GROUP BY ts
|
||||
ORDER BY ts
|
||||
"""), {"cutoff": cutoff})
|
||||
rows = buckets.mappings().all()
|
||||
|
||||
return [TimelinePoint(timestamp=r["ts"], event_count=r["event_count"],
|
||||
avg_sentiment=float(r["avg_sentiment"])) for r in rows]
|
||||
|
||||
|
||||
@app.get("/api/analytics/sentiment/by-source")
|
||||
async def sentiment_by_source(hours: int = 24):
|
||||
"""Sentiment breakdown grouped by source type."""
|
||||
async with async_session() as session:
|
||||
cutoff = datetime.now(timezone.utc) - timedelta(hours=hours)
|
||||
result = await session.execute(text(f"""
|
||||
SELECT
|
||||
source_type,
|
||||
COUNT(*) AS total,
|
||||
COUNT(*) FILTER (WHERE sentiment_label = 'positive') AS positive,
|
||||
COUNT(*) FILTER (WHERE sentiment_label = 'neutral') AS neutral,
|
||||
COUNT(*) FILTER (WHERE sentiment_label = 'negative') AS negative,
|
||||
COALESCE(AVG(sentiment_score), 0) AS avg_score
|
||||
FROM events
|
||||
WHERE ingested_at >= :cutoff
|
||||
GROUP BY source_type
|
||||
ORDER BY total DESC
|
||||
"""), {"cutoff": cutoff})
|
||||
rows = result.mappings().all()
|
||||
return [{
|
||||
"source_type": r["source_type"],
|
||||
"total": r["total"],
|
||||
"positive": r["positive"],
|
||||
"neutral": r["neutral"],
|
||||
"negative": r["negative"],
|
||||
"avg_score": float(r["avg_score"]),
|
||||
} for r in rows]
|
||||
|
||||
|
||||
# ── Frontend ──────────────────────────────────────────────────────────────
|
||||
|
||||
@app.get("/", response_class=HTMLResponse)
|
||||
async def index():
|
||||
return FileResponse(str(STATIC_DIR / "index.html"))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
import uvicorn
|
||||
uvicorn.run(app, host="0.0.0.0", port=8000)
|
||||
147
app/models.py
Normal file
147
app/models.py
Normal file
|
|
@ -0,0 +1,147 @@
|
|||
"""OSINT Dashboard — SQLAlchemy models (async, declarative)."""
|
||||
|
||||
from datetime import datetime, timezone
|
||||
from uuid import uuid4
|
||||
|
||||
from sqlalchemy import (
|
||||
Column, Enum, Float, Index, Integer, String, Text,
|
||||
DateTime, JSON, func, Table,
|
||||
)
|
||||
from sqlalchemy.dialects.postgresql import UUID, TSVECTOR
|
||||
|
||||
from database import metadata
|
||||
|
||||
|
||||
# ── Feed Sources ──────────────────────────────────────────────────────────
|
||||
|
||||
feed_sources = Table(
|
||||
"feed_sources",
|
||||
metadata,
|
||||
Column("id", UUID(as_uuid=True), primary_key=True, default=uuid4),
|
||||
Column("name", String(256), nullable=False),
|
||||
Column("source_type", Enum(
|
||||
"rss", "gdel-t2", "social", "earthquake", "disaster",
|
||||
"weather", "fire", "satellite", name="feed_source_type"
|
||||
), nullable=False),
|
||||
Column("url", Text),
|
||||
Column("config", JSON),
|
||||
Column("enabled", Integer, server_default="1", nullable=False),
|
||||
Column("created_at", DateTime(timezone=True), server_default=func.now(), nullable=False),
|
||||
Column("updated_at", DateTime(timezone=True), server_default=func.now(), onupdate=func.now()),
|
||||
)
|
||||
|
||||
|
||||
# ── Events (hypertable via TimescaleDB) ──────────────────────────────────
|
||||
|
||||
events = Table(
|
||||
"events",
|
||||
metadata,
|
||||
Column("id", UUID(as_uuid=True), primary_key=True, default=uuid4),
|
||||
Column("source_type", Enum(
|
||||
"rss", "gdel-t2", "social", "earthquake", "disaster",
|
||||
"weather", "fire", "satellite", name="event_source_type"
|
||||
), nullable=False, index=True),
|
||||
Column("source_id", UUID(as_uuid=True)),
|
||||
Column("title", Text),
|
||||
Column("body", Text),
|
||||
Column("url", Text),
|
||||
Column("sentiment_score", Float),
|
||||
Column("sentiment_label", Enum("positive", "neutral", "negative", name="sentiment_label")),
|
||||
Column("location_lat", Float),
|
||||
Column("location_lon", Float),
|
||||
Column("location_name", String(512)),
|
||||
Column("entities", JSON),
|
||||
Column("tags", JSON),
|
||||
Column("raw", JSON),
|
||||
Column("ingested_at", DateTime(timezone=True), server_default=func.now(), nullable=False),
|
||||
Column("source_timestamp", DateTime(timezone=True), nullable=False),
|
||||
# Full-text search vector
|
||||
Column(
|
||||
"search_vector",
|
||||
TSVECTOR,
|
||||
nullable=True,
|
||||
),
|
||||
)
|
||||
|
||||
# GIN index for full-text search
|
||||
Index("ix_events_search_vector", events.c.search_vector, postgresql_using="gin")
|
||||
# Spatial index on location
|
||||
Index("ix_events_location", events.c.location_lat, events.c.location_lon)
|
||||
|
||||
|
||||
# ── Entities (people, organizations, locations of interest) ──────────────
|
||||
|
||||
entities = Table(
|
||||
"entities",
|
||||
metadata,
|
||||
Column("id", UUID(as_uuid=True), primary_key=True, default=uuid4),
|
||||
Column("name", String(512), nullable=False, index=True),
|
||||
Column("entity_type", Enum(
|
||||
"person", "organization", "location", "topic", "asset",
|
||||
name="entity_type"
|
||||
), nullable=False),
|
||||
Column("aliases", JSON),
|
||||
Column("description", Text),
|
||||
Column("metadata", JSON),
|
||||
Column("location_lat", Float),
|
||||
Column("location_lon", Float),
|
||||
Column("event_count", Integer, server_default="0"),
|
||||
Column("first_seen", DateTime(timezone=True), server_default=func.now()),
|
||||
Column("last_seen", DateTime(timezone=True), server_default=func.now()),
|
||||
)
|
||||
|
||||
|
||||
# ── Entity-Event Link ────────────────────────────────────────────────────
|
||||
|
||||
entity_events = Table(
|
||||
"entity_events",
|
||||
metadata,
|
||||
Column("entity_id", UUID(as_uuid=True), primary_key=True),
|
||||
Column("event_id", UUID(as_uuid=True), primary_key=True),
|
||||
Column("relevance_score", Float),
|
||||
Column("linked_at", DateTime(timezone=True), server_default=func.now()),
|
||||
)
|
||||
|
||||
|
||||
# ── Alerts ───────────────────────────────────────────────────────────────
|
||||
|
||||
alerts = Table(
|
||||
"alerts",
|
||||
metadata,
|
||||
Column("id", UUID(as_uuid=True), primary_key=True, default=uuid4),
|
||||
Column("alert_type", Enum(
|
||||
"entity_mention", "sentiment_shift", "geo_proximity",
|
||||
"keyword_match", "threshold", "anomaly",
|
||||
name="alert_type"
|
||||
), nullable=False),
|
||||
Column("entity_id", UUID(as_uuid=True)),
|
||||
Column("event_id", UUID(as_uuid=True)),
|
||||
Column("severity", Enum("low", "medium", "high", "critical", name="alert_severity"), nullable=False),
|
||||
Column("title", Text, nullable=False),
|
||||
Column("message", Text),
|
||||
Column("context", JSON),
|
||||
Column("acknowledged", Integer, server_default="0"),
|
||||
Column("acknowledged_by", String(256)),
|
||||
Column("created_at", DateTime(timezone=True), server_default=func.now(), nullable=False),
|
||||
Column("resolved_at", DateTime(timezone=True)),
|
||||
)
|
||||
|
||||
Index("ix_alerts_severity_created", alerts.c.severity, alerts.c.created_at.desc())
|
||||
Index("ix_alerts_entity", alerts.c.entity_id)
|
||||
|
||||
|
||||
# ── Documents (stored in MinIO, indexed here) ────────────────────────────
|
||||
|
||||
documents = Table(
|
||||
"documents",
|
||||
metadata,
|
||||
Column("id", UUID(as_uuid=True), primary_key=True, default=uuid4),
|
||||
Column("bucket", String(256), nullable=False),
|
||||
Column("object_key", String(1024), nullable=False),
|
||||
Column("content_type", String(256)),
|
||||
Column("size_bytes", Integer),
|
||||
Column("description", Text),
|
||||
Column("tags", JSON),
|
||||
Column("event_id", UUID(as_uuid=True)),
|
||||
Column("uploaded_at", DateTime(timezone=True), server_default=func.now()),
|
||||
)
|
||||
13
app/requirements.txt
Normal file
13
app/requirements.txt
Normal file
|
|
@ -0,0 +1,13 @@
|
|||
fastapi>=0.115
|
||||
uvicorn[standard]>=0.34
|
||||
sqlalchemy[asyncio]>=2.0
|
||||
asyncpg>=0.30
|
||||
nats-py>=2.9
|
||||
redis[hiredis]>=5.2
|
||||
minio>=7.2
|
||||
pydantic>=2.10
|
||||
alembic>=1.14
|
||||
httpx>=0.28
|
||||
feedparser>=6.0
|
||||
python-dateutil>=2.9
|
||||
structlog>=24.4
|
||||
242
app/schemas.py
Normal file
242
app/schemas.py
Normal file
|
|
@ -0,0 +1,242 @@
|
|||
"""OSINT Dashboard — Pydantic schemas."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from datetime import datetime
|
||||
from enum import Enum
|
||||
from typing import Optional
|
||||
from uuid import UUID
|
||||
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
|
||||
# ─── Enums ───────────────────────────────────────────────────────────────
|
||||
|
||||
class SourceType(str, Enum):
|
||||
rss = "rss"
|
||||
gdel_t2 = "gdel-t2"
|
||||
social = "social"
|
||||
earthquake = "earthquake"
|
||||
disaster = "disaster"
|
||||
weather = "weather"
|
||||
fire = "fire"
|
||||
satellite = "satellite"
|
||||
|
||||
|
||||
class EntityKind(str, Enum):
|
||||
person = "person"
|
||||
organization = "organization"
|
||||
location = "location"
|
||||
topic = "topic"
|
||||
asset = "asset"
|
||||
|
||||
|
||||
class Sentiment(str, Enum):
|
||||
positive = "positive"
|
||||
neutral = "neutral"
|
||||
negative = "negative"
|
||||
|
||||
|
||||
class AlertType(str, Enum):
|
||||
entity_mention = "entity_mention"
|
||||
sentiment_shift = "sentiment_shift"
|
||||
geo_proximity = "geo_proximity"
|
||||
keyword_match = "keyword_match"
|
||||
threshold = "threshold"
|
||||
anomaly = "anomaly"
|
||||
|
||||
|
||||
class AlertSeverity(str, Enum):
|
||||
low = "low"
|
||||
medium = "medium"
|
||||
high = "high"
|
||||
critical = "critical"
|
||||
|
||||
|
||||
# ─── Feed Sources ───────────────────────────────────────────────────────
|
||||
|
||||
class FeedSourceCreate(BaseModel):
|
||||
name: str
|
||||
source_type: SourceType
|
||||
url: Optional[str] = None
|
||||
config: Optional[dict] = None
|
||||
|
||||
|
||||
class FeedSourceOut(BaseModel):
|
||||
id: UUID
|
||||
name: str
|
||||
source_type: SourceType
|
||||
url: Optional[str]
|
||||
config: Optional[dict]
|
||||
enabled: bool
|
||||
created_at: datetime
|
||||
|
||||
|
||||
# ─── Events ─────────────────────────────────────────────────────────────
|
||||
|
||||
class EventCreate(BaseModel):
|
||||
source_type: SourceType
|
||||
source_id: Optional[UUID] = None
|
||||
title: Optional[str] = None
|
||||
body: Optional[str] = None
|
||||
url: Optional[str] = None
|
||||
sentiment_score: Optional[float] = None
|
||||
sentiment_label: Optional[Sentiment] = None
|
||||
location_lat: Optional[float] = None
|
||||
location_lon: Optional[float] = None
|
||||
location_name: Optional[str] = None
|
||||
entities: Optional[list[dict]] = None
|
||||
tags: Optional[list[str]] = None
|
||||
raw: Optional[dict] = None
|
||||
source_timestamp: Optional[datetime] = None
|
||||
|
||||
|
||||
class EventOut(BaseModel):
|
||||
id: UUID
|
||||
source_type: SourceType
|
||||
source_id: Optional[UUID]
|
||||
title: Optional[str]
|
||||
body: Optional[str]
|
||||
url: Optional[str]
|
||||
sentiment_score: Optional[float]
|
||||
sentiment_label: Optional[Sentiment]
|
||||
location_lat: Optional[float]
|
||||
location_lon: Optional[float]
|
||||
location_name: Optional[str]
|
||||
entities: Optional[list[dict]]
|
||||
tags: Optional[list[str]]
|
||||
ingested_at: datetime
|
||||
source_timestamp: datetime
|
||||
|
||||
|
||||
# ─── Search ──────────────────────────────────────────────────────────────
|
||||
|
||||
class SearchQuery(BaseModel):
|
||||
q: str = Field(..., min_length=1, max_length=500)
|
||||
source_type: Optional[SourceType] = None
|
||||
entity_id: Optional[UUID] = None
|
||||
sentiment: Optional[Sentiment] = None
|
||||
min_date: Optional[datetime] = None
|
||||
max_date: Optional[datetime] = None
|
||||
min_lat: Optional[float] = None
|
||||
max_lat: Optional[float] = None
|
||||
min_lon: Optional[float] = None
|
||||
max_lon: Optional[float] = None
|
||||
limit: int = Field(50, ge=1, le=500)
|
||||
offset: int = Field(0, ge=0)
|
||||
|
||||
|
||||
class SearchResult(BaseModel):
|
||||
events: list[EventOut]
|
||||
total: int
|
||||
has_more: bool
|
||||
|
||||
|
||||
# ─── Entities ────────────────────────────────────────────────────────────
|
||||
|
||||
class EntityCreate(BaseModel):
|
||||
name: str
|
||||
entity_type: EntityKind
|
||||
aliases: Optional[list[str]] = None
|
||||
description: Optional[str] = None
|
||||
metadata: Optional[dict] = None
|
||||
location_lat: Optional[float] = None
|
||||
location_lon: Optional[float] = None
|
||||
|
||||
|
||||
class EntityOut(BaseModel):
|
||||
id: UUID
|
||||
name: str
|
||||
entity_type: EntityKind
|
||||
aliases: Optional[list[str]]
|
||||
description: Optional[str]
|
||||
metadata: Optional[dict]
|
||||
location_lat: Optional[float]
|
||||
location_lon: Optional[float]
|
||||
event_count: int
|
||||
first_seen: datetime
|
||||
last_seen: datetime
|
||||
|
||||
|
||||
# ─── Alerts ──────────────────────────────────────────────────────────────
|
||||
|
||||
class AlertCreate(BaseModel):
|
||||
alert_type: AlertType
|
||||
entity_id: Optional[UUID] = None
|
||||
event_id: Optional[UUID] = None
|
||||
severity: AlertSeverity
|
||||
title: str
|
||||
message: Optional[str] = None
|
||||
context: Optional[dict] = None
|
||||
|
||||
|
||||
class AlertUpdate(BaseModel):
|
||||
acknowledged: Optional[bool] = None
|
||||
acknowledged_by: Optional[str] = None
|
||||
resolved_at: Optional[datetime] = None
|
||||
|
||||
|
||||
class AlertOut(BaseModel):
|
||||
id: UUID
|
||||
alert_type: AlertType
|
||||
entity_id: Optional[UUID]
|
||||
event_id: Optional[UUID]
|
||||
severity: AlertSeverity
|
||||
title: str
|
||||
message: Optional[str]
|
||||
context: Optional[dict]
|
||||
acknowledged: bool
|
||||
acknowledged_by: Optional[str]
|
||||
created_at: datetime
|
||||
resolved_at: Optional[datetime]
|
||||
|
||||
|
||||
# ─── Documents ───────────────────────────────────────────────────────────
|
||||
|
||||
class DocumentCreate(BaseModel):
|
||||
bucket: str
|
||||
object_key: str
|
||||
content_type: Optional[str] = None
|
||||
size_bytes: Optional[int] = None
|
||||
description: Optional[str] = None
|
||||
tags: Optional[list[str]] = None
|
||||
event_id: Optional[UUID] = None
|
||||
|
||||
|
||||
class DocumentOut(BaseModel):
|
||||
id: UUID
|
||||
bucket: str
|
||||
object_key: str
|
||||
content_type: Optional[str]
|
||||
size_bytes: Optional[int]
|
||||
description: Optional[str]
|
||||
tags: Optional[list[str]]
|
||||
event_id: Optional[UUID]
|
||||
uploaded_at: datetime
|
||||
|
||||
|
||||
# ─── Aggregations ────────────────────────────────────────────────────────
|
||||
|
||||
class SentimentSummary(BaseModel):
|
||||
period: str
|
||||
positive_count: int
|
||||
neutral_count: int
|
||||
negative_count: int
|
||||
avg_score: float
|
||||
|
||||
|
||||
class TimelinePoint(BaseModel):
|
||||
timestamp: datetime
|
||||
event_count: int
|
||||
avg_sentiment: float
|
||||
|
||||
|
||||
class DashboardSummary(BaseModel):
|
||||
total_events: int
|
||||
events_last_24h: int
|
||||
active_sources: int
|
||||
open_alerts: int
|
||||
tracked_entities: int
|
||||
sentiment: SentimentSummary
|
||||
top_entities: list[EntityOut]
|
||||
|
||||
186
app/sources.py
Normal file
186
app/sources.py
Normal file
|
|
@ -0,0 +1,186 @@
|
|||
"""Data source ingestors — fetch from external APIs and push to NATS."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import logging
|
||||
from datetime import datetime, timezone
|
||||
from email.utils import parsedate_to_datetime
|
||||
|
||||
import httpx
|
||||
import feedparser
|
||||
import nats
|
||||
|
||||
logger = logging.getLogger("osint.sources")
|
||||
|
||||
|
||||
def _parse_rfc822(date_str: object) -> str | None:
|
||||
"""Parse RFC-822 date string from feedparser entries."""
|
||||
if not isinstance(date_str, str):
|
||||
return None
|
||||
try:
|
||||
return parsedate_to_datetime(date_str).astimezone(timezone.utc).isoformat()
|
||||
except (ValueError, TypeError):
|
||||
return None
|
||||
|
||||
# NATS connection
|
||||
NATS_URLS = "nats://osint-nats.customer1.svc.cluster.local:4222"
|
||||
|
||||
|
||||
async def publish_event(subject: str, event: dict):
|
||||
"""Publish an event to NATS JetStream."""
|
||||
nc = await nats.connect(NATS_URLS)
|
||||
js = nc.jetstream()
|
||||
await js.publish(subject, json.dumps(event).encode())
|
||||
await nc.close()
|
||||
logger.debug("Published event to %s", subject)
|
||||
|
||||
|
||||
# ─── RSS Feed Ingestor ──────────────────────────────────────────────────
|
||||
|
||||
async def ingest_rss_feed(feed_url: str, source_id: str = None):
|
||||
"""Fetch and parse an RSS feed, publish items to NATS."""
|
||||
async with httpx.AsyncClient(timeout=30) as client:
|
||||
resp = await client.get(feed_url)
|
||||
resp.raise_for_status()
|
||||
feed = feedparser.parse(resp.text)
|
||||
|
||||
count = 0
|
||||
for entry in feed.entries[:100]: # max 100 per run
|
||||
event = {
|
||||
"source_type": "rss",
|
||||
"source_id": source_id,
|
||||
"title": entry.get("title"),
|
||||
"body": entry.get("summary") or entry.get("description"),
|
||||
"url": entry.get("link"),
|
||||
"source_timestamp": _parse_rfc822(entry.get("published"))
|
||||
or datetime.now(timezone.utc).isoformat(),
|
||||
"tags": [t.get("term") for t in entry.get("tags", []) if t.get("term")],
|
||||
"raw": {
|
||||
"feed_title": feed.feed.get("title"),
|
||||
"author": entry.get("author"),
|
||||
"categories": [c.get("term") for c in entry.get("categories", [])],
|
||||
},
|
||||
}
|
||||
await publish_event("events.rss", event)
|
||||
count += 1
|
||||
|
||||
logger.info("Ingested %d items from RSS feed %s", count, feed_url)
|
||||
return count
|
||||
|
||||
|
||||
# ─── GDELT 2.0 Ingestor ─────────────────────────────────────────────────
|
||||
|
||||
GDELT_API = "https://api.gdeltproject.org/gdeltv2"
|
||||
|
||||
|
||||
async def ingest_gdelt(query: str = "", max_articles: int = 50):
|
||||
"""Fetch articles from GDELT 2.0 API."""
|
||||
params = {
|
||||
"mode": "artlist",
|
||||
"format": "json",
|
||||
"maxrecords": max_articles,
|
||||
"mode": "artlist",
|
||||
}
|
||||
if query:
|
||||
params["search"] = query
|
||||
|
||||
async with httpx.AsyncClient(timeout=60) as client:
|
||||
resp = await client.get(GDELT_API, params=params)
|
||||
resp.raise_for_status()
|
||||
data = resp.json()
|
||||
|
||||
count = 0
|
||||
for article in data.get("articles", []):
|
||||
event = {
|
||||
"source_type": "gdel-t2",
|
||||
"title": article.get("title"),
|
||||
"body": article.get("articleBody"),
|
||||
"url": article.get("url"),
|
||||
"sentiment_score": _parse_gdelt_tone(article.get("Tone", "0")),
|
||||
"location_lat": article.get("Latitude"),
|
||||
"location_lon": article.get("Longitude"),
|
||||
"location_name": article.get("Location"),
|
||||
"source_timestamp": article.get("FirstCreated"),
|
||||
"entities": [
|
||||
{"name": e.get("Topic"), "type": "topic"}
|
||||
for e in article.get("Mentions", [])
|
||||
if e.get("Topic")
|
||||
],
|
||||
"raw": article,
|
||||
}
|
||||
await publish_event("events.gdelt", event)
|
||||
count += 1
|
||||
|
||||
logger.info("Ingested %d articles from GDELT", count)
|
||||
return count
|
||||
|
||||
|
||||
def _parse_gdelt_tone(tone: str) -> float | None:
|
||||
"""Parse GDELT tone string to a -1..1 sentiment score."""
|
||||
try:
|
||||
tone_float = float(tone)
|
||||
return max(-1.0, min(1.0, tone_float / 4249.0)) # GDELT tone range
|
||||
except (ValueError, TypeError):
|
||||
return None
|
||||
|
||||
|
||||
# ─── Earthquake Ingestor (USGS) ─────────────────────────────────────────
|
||||
|
||||
USGS_API = "https://earthquake.usgs.gov/earthquakes/feed/v1.0/summary/all_hour.geojson"
|
||||
|
||||
|
||||
async def ingest_earthquakes():
|
||||
"""Fetch recent earthquakes from USGS."""
|
||||
async with httpx.AsyncClient(timeout=30) as client:
|
||||
resp = await client.get(USGS_API)
|
||||
resp.raise_for_status()
|
||||
data = resp.json()
|
||||
|
||||
count = 0
|
||||
for feature in data.get("features", []):
|
||||
props = feature.get("properties", {})
|
||||
geometry = feature.get("geometry", {}).get("coordinates", [])
|
||||
event = {
|
||||
"source_type": "earthquake",
|
||||
"title": props.get("title"),
|
||||
"body": props.get("description"),
|
||||
"url": props.get("url"),
|
||||
"location_lat": geometry[1] if len(geometry) > 1 else None,
|
||||
"location_lon": geometry[0] if len(geometry) > 0 else None,
|
||||
"location_name": props.get("place"),
|
||||
"sentiment_label": "neutral",
|
||||
"tags": [f"magnitude:{props.get('mag')}"] if props.get("mag") else [],
|
||||
"source_timestamp": (
|
||||
datetime.utcfromtimestamp(props.get("time", 0) / 1000)
|
||||
.replace(tzinfo=timezone.utc)
|
||||
.isoformat()
|
||||
),
|
||||
"raw": props,
|
||||
}
|
||||
await publish_event("events.earthquake", event)
|
||||
count += 1
|
||||
|
||||
logger.info("Ingested %d earthquake events", count)
|
||||
return count
|
||||
|
||||
|
||||
# ─── Social Signals (Twitter/X-like placeholder) ────────────────────────
|
||||
|
||||
async def ingest_social_signals(query: str = "", max_items: int = 50):
|
||||
"""Placeholder for social media signal ingestion.
|
||||
|
||||
In production, this would connect to Twitter API, Reddit, NewsAPI, etc.
|
||||
For now, it publishes a heartbeat to signal the pipeline is active.
|
||||
"""
|
||||
event = {
|
||||
"source_type": "social",
|
||||
"title": f"Social signal scan: {query}",
|
||||
"body": f"Scanned for '{query}' — placeholder connector",
|
||||
"source_timestamp": datetime.now(timezone.utc).isoformat(),
|
||||
"tags": [query] if query else [],
|
||||
"raw": {"query": query, "max_items": max_items, "connector": "placeholder"},
|
||||
}
|
||||
await publish_event("events.social", event)
|
||||
logger.info("Social signal scan complete for '%s'", query)
|
||||
return 1
|
||||
307
app/static/index.html
Normal file
307
app/static/index.html
Normal file
|
|
@ -0,0 +1,307 @@
|
|||
<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<title>OSINT Dashboard</title>
|
||||
<style>
|
||||
:root { --bg: #0f172a; --surface: #1e293b; --border: #334155; --text: #e2e8f0; --muted: #94a3b8; --accent: #38bdf8; --green: #4ade80; --red: #f87171; --yellow: #fbbf24; }
|
||||
* { margin: 0; padding: 0; box-sizing: border-box; }
|
||||
body { font-family: -apple-system, system-ui, sans-serif; background: var(--bg); color: var(--text); min-height: 100vh; }
|
||||
header { background: var(--surface); border-bottom: 1px solid var(--border); padding: 1rem 2rem; display: flex; justify-content: space-between; align-items: center; }
|
||||
header h1 { font-size: 1.25rem; color: var(--accent); }
|
||||
.status { font-size: 0.85rem; color: var(--muted); }
|
||||
.status .ok { color: var(--green); }
|
||||
.container { max-width: 1400px; margin: 0 auto; padding: 1.5rem; }
|
||||
.grid { display: grid; grid-template-columns: repeat(auto-fit, minmax(280px, 1fr)); gap: 1rem; margin-bottom: 1.5rem; }
|
||||
.card { background: var(--surface); border: 1px solid var(--border); border-radius: 8px; padding: 1.25rem; }
|
||||
.card h3 { font-size: 0.8rem; text-transform: uppercase; color: var(--muted); margin-bottom: 0.5rem; }
|
||||
.card .value { font-size: 2rem; font-weight: 700; }
|
||||
.card .sub { font-size: 0.85rem; color: var(--muted); margin-top: 0.25rem; }
|
||||
.section { margin-bottom: 1.5rem; }
|
||||
.section h2 { font-size: 1rem; margin-bottom: 0.75rem; color: var(--accent); }
|
||||
table { width: 100%; border-collapse: collapse; font-size: 0.9rem; }
|
||||
th, td { text-align: left; padding: 0.6rem 0.75rem; border-bottom: 1px solid var(--border); }
|
||||
th { color: var(--muted); font-weight: 500; font-size: 0.8rem; }
|
||||
.badge { display: inline-block; padding: 0.15rem 0.5rem; border-radius: 9999px; font-size: 0.75rem; font-weight: 500; }
|
||||
.badge-positive { background: #052e16; color: var(--green); }
|
||||
.badge-negative { background: #450a0a; color: var(--red); }
|
||||
.badge-neutral { background: #1e293b; color: var(--muted); }
|
||||
.badge-high { background: #450a0a; color: var(--red); }
|
||||
.badge-medium { background: #451a03; color: var(--yellow); }
|
||||
.badge-low { background: #052e16; color: var(--green); }
|
||||
.search-box { display: flex; gap: 0.5rem; margin-bottom: 1rem; }
|
||||
.search-box input { flex: 1; background: var(--surface); border: 1px solid var(--border); border-radius: 6px; padding: 0.6rem 1rem; color: var(--text); font-size: 0.9rem; }
|
||||
.search-box input:focus { outline: none; border-color: var(--accent); }
|
||||
.search-box button { background: var(--accent); color: #0f172a; border: none; border-radius: 6px; padding: 0.6rem 1.25rem; font-weight: 500; cursor: pointer; }
|
||||
.btn { background: var(--surface); border: 1px solid var(--border); color: var(--text); border-radius: 6px; padding: 0.5rem 1rem; cursor: pointer; font-size: 0.85rem; }
|
||||
.btn:hover { border-color: var(--accent); }
|
||||
.sentiment-bar { display: flex; height: 24px; border-radius: 4px; overflow: hidden; margin-top: 0.5rem; }
|
||||
.sentiment-bar div { transition: width 0.3s; }
|
||||
.tab-bar { display: flex; gap: 0.5rem; margin-bottom: 1rem; }
|
||||
.tab-bar .btn.active { background: var(--accent); color: #0f172a; }
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<header>
|
||||
<h1>OSINT Dashboard</h1>
|
||||
<div class="status">
|
||||
Status: <span id="health" class="ok">checking...</span>
|
||||
| Last update: <span id="last-update">-</span>
|
||||
</div>
|
||||
</header>
|
||||
|
||||
<div class="container">
|
||||
<!-- Summary Cards -->
|
||||
<div class="grid" id="summary-cards">
|
||||
<div class="card"><h3>Total Events</h3><div class="value" id="total-events">-</div></div>
|
||||
<div class="card"><h3>Events (24h)</h3><div class="value" id="events-24h">-</div><div class="sub">last 24 hours</div></div>
|
||||
<div class="card"><h3>Active Sources</h3><div class="value" id="active-sources">-</div></div>
|
||||
<div class="card"><h3>Open Alerts</h3><div class="value" id="open-alerts" style="color:var(--red)">-</div></div>
|
||||
<div class="card"><h3>Tracked Entities</h3><div class="value" id="tracked-entities">-</div></div>
|
||||
<div class="card">
|
||||
<h3>Sentiment (24h)</h3>
|
||||
<div class="sentiment-bar">
|
||||
<div id="sent-pos" style="background:var(--green)"></div>
|
||||
<div id="sent-neu" style="background:var(--muted)"></div>
|
||||
<div id="sent-neg" style="background:var(--red)"></div>
|
||||
</div>
|
||||
<div class="sub" id="sent-detail"></div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<!-- Search -->
|
||||
<div class="section">
|
||||
<h2>Search Events</h2>
|
||||
<div class="search-box">
|
||||
<input type="text" id="search-q" placeholder="Search events by keyword..." />
|
||||
<button onclick="searchEvents()">Search</button>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<!-- Tabs -->
|
||||
<div class="tab-bar">
|
||||
<button class="btn active" onclick="showTab('recent')">Recent Events</button>
|
||||
<button class="btn" onclick="showTab('alerts')">Alerts</button>
|
||||
<button class="btn" onclick="showTab('entities')">Entities</button>
|
||||
<button class="btn" onclick="showTab('ingest')">Ingest</button>
|
||||
</div>
|
||||
|
||||
<!-- Recent Events -->
|
||||
<div class="section" id="tab-recent">
|
||||
<h2>Recent Events</h2>
|
||||
<table>
|
||||
<thead><tr><th>Time</th><th>Source</th><th>Title</th><th>Sentiment</th><th>Location</th></tr></thead>
|
||||
<tbody id="events-body"></tbody>
|
||||
</table>
|
||||
</div>
|
||||
|
||||
<!-- Alerts -->
|
||||
<div class="section" id="tab-alerts" style="display:none">
|
||||
<h2>Open Alerts</h2>
|
||||
<table>
|
||||
<thead><tr><th>Time</th><th>Severity</th><th>Type</th><th>Title</th></tr></thead>
|
||||
<tbody id="alerts-body"></tbody>
|
||||
</table>
|
||||
</div>
|
||||
|
||||
<!-- Entities -->
|
||||
<div class="section" id="tab-entities" style="display:none">
|
||||
<h2>Tracked Entities</h2>
|
||||
<table>
|
||||
<thead><tr><th>Name</th><th>Type</th><th>Events</th><th>Last Seen</th></tr></thead>
|
||||
<tbody id="entities-body"></tbody>
|
||||
</table>
|
||||
</div>
|
||||
|
||||
<!-- Ingest -->
|
||||
<div class="section" id="tab-ingest" style="display:none">
|
||||
<h2>Data Ingestion</h2>
|
||||
<div class="grid">
|
||||
<div class="card">
|
||||
<h3>RSS Feed</h3>
|
||||
<div style="margin-top:0.5rem;display:flex;gap:0.5rem">
|
||||
<input type="text" id="rss-url" placeholder="https://example.com/feed" style="flex:1;background:var(--bg);border:1px solid var(--border);border-radius:4px;padding:0.4rem;color:var(--text);font-size:0.85rem">
|
||||
<button class="btn" onclick="ingestRSS()">Fetch</button>
|
||||
</div>
|
||||
</div>
|
||||
<div class="card">
|
||||
<h3>GDELT Articles</h3>
|
||||
<p class="sub" style="margin:0.5rem 0">Global news monitoring</p>
|
||||
<button class="btn" onclick="ingestGDELT()">Fetch Latest</button>
|
||||
</div>
|
||||
<div class="card">
|
||||
<h3>Earthquakes (USGS)</h3>
|
||||
<p class="sub" style="margin:0.5rem 0">Last hour of seismic data</p>
|
||||
<button class="btn" onclick="ingestEarthquakes()">Fetch Latest</button>
|
||||
</div>
|
||||
<div class="card">
|
||||
<h3>NATS Consumer</h3>
|
||||
<p class="sub" style="margin:0.5rem 0">Process pending messages</p>
|
||||
<button class="btn" onclick="processNATS()">Process</button>
|
||||
</div>
|
||||
</div>
|
||||
<div id="ingest-result" style="margin-top:1rem;color:var(--muted);font-size:0.9rem"></div>
|
||||
</div>
|
||||
|
||||
<!-- Search Results -->
|
||||
<div class="section" id="search-results" style="display:none">
|
||||
<h2>Search Results (<span id="search-total">0</span>)</h2>
|
||||
<table>
|
||||
<thead><tr><th>Time</th><th>Source</th><th>Title</th><th>Sentiment</th><th>Location</th></tr></thead>
|
||||
<tbody id="search-body"></tbody>
|
||||
</table>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<script>
|
||||
const API = '';
|
||||
|
||||
async function loadSummary() {
|
||||
try {
|
||||
const r = await fetch(`${API}/api/analytics/summary`);
|
||||
const d = await r.json();
|
||||
document.getElementById('total-events').textContent = d.total_events;
|
||||
document.getElementById('events-24h').textContent = d.events_last_24h;
|
||||
document.getElementById('active-sources').textContent = d.active_sources;
|
||||
document.getElementById('open-alerts').textContent = d.open_alerts;
|
||||
document.getElementById('tracked-entities').textContent = d.tracked_entities;
|
||||
const s = d.sentiment;
|
||||
const total = s.positive_count + s.neutral_count + s.negative_count || 1;
|
||||
document.getElementById('sent-pos').style.width = ((s.positive_count/total)*100)+'%';
|
||||
document.getElementById('sent-neu').style.width = ((s.neutral_count/total)*100)+'%';
|
||||
document.getElementById('sent-neg').style.width = ((s.negative_count/total)*100)+'%';
|
||||
document.getElementById('sent-detail').textContent = `+${s.positive_count} | ~${s.neutral_count} | -${s.negative_count} (avg: ${s.avg_score.toFixed(3)})`;
|
||||
} catch(e) { console.error('Summary load failed', e); }
|
||||
}
|
||||
|
||||
async function loadEvents() {
|
||||
try {
|
||||
const r = await fetch(`${API}/api/events?limit=30`);
|
||||
const d = await r.json();
|
||||
const tb = document.getElementById('events-body');
|
||||
tb.innerHTML = d.map(e => `<tr>
|
||||
<td>${new Date(e.ingested_at).toLocaleString()}</td>
|
||||
<td>${e.source_type}</td>
|
||||
<td>${(e.title||'').substring(0,80)}</td>
|
||||
<td>${sentimentBadge(e.sentiment_label)}</td>
|
||||
<td>${e.location_name || '-'}</td>
|
||||
</tr>`).join('');
|
||||
} catch(e) { console.error('Events load failed', e); }
|
||||
}
|
||||
|
||||
async function loadAlerts() {
|
||||
try {
|
||||
const r = await fetch(`${API}/api/alerts?limit=20`);
|
||||
const d = await r.json();
|
||||
const tb = document.getElementById('alerts-body');
|
||||
tb.innerHTML = d.map(a => `<tr>
|
||||
<td>${new Date(a.created_at).toLocaleString()}</td>
|
||||
<td><span class="badge badge-${a.severity}">${a.severity}</span></td>
|
||||
<td>${a.alert_type}</td>
|
||||
<td>${a.title}</td>
|
||||
</tr>`).join('');
|
||||
} catch(e) { console.error('Alerts load failed', e); }
|
||||
}
|
||||
|
||||
async function loadEntities() {
|
||||
try {
|
||||
const r = await fetch(`${API}/api/entities?limit=20`);
|
||||
const d = await r.json();
|
||||
const tb = document.getElementById('entities-body');
|
||||
tb.innerHTML = d.map(e => `<tr>
|
||||
<td>${e.name}</td>
|
||||
<td>${e.entity_type}</td>
|
||||
<td>${e.event_count}</td>
|
||||
<td>${new Date(e.last_seen).toLocaleString()}</td>
|
||||
</tr>`).join('');
|
||||
} catch(e) { console.error('Entities load failed', e); }
|
||||
}
|
||||
|
||||
async function searchEvents() {
|
||||
const q = document.getElementById('search-q').value.trim();
|
||||
if (!q) return;
|
||||
try {
|
||||
const r = await fetch(`${API}/api/search`, {
|
||||
method: 'POST',
|
||||
headers: {'Content-Type': 'application/json'},
|
||||
body: JSON.stringify({q, limit: 50})
|
||||
});
|
||||
const d = await r.json();
|
||||
document.getElementById('search-total').textContent = d.total;
|
||||
document.getElementById('search-results').style.display = 'block';
|
||||
const tb = document.getElementById('search-body');
|
||||
tb.innerHTML = d.events.map(e => `<tr>
|
||||
<td>${new Date(e.ingested_at).toLocaleString()}</td>
|
||||
<td>${e.source_type}</td>
|
||||
<td>${(e.title||'').substring(0,80)}</td>
|
||||
<td>${sentimentBadge(e.sentiment_label)}</td>
|
||||
<td>${e.location_name || '-'}</td>
|
||||
</tr>`).join('');
|
||||
} catch(e) { console.error('Search failed', e); }
|
||||
}
|
||||
|
||||
function sentimentBadge(label) {
|
||||
if (!label) return '-';
|
||||
const cls = {positive:'badge-positive',negative:'badge-negative',neutral:'badge-neutral'}[label]||'badge-neutral';
|
||||
return `<span class="badge ${cls}">${label}</span>`;
|
||||
}
|
||||
|
||||
function showTab(name) {
|
||||
['recent','alerts','entities','ingest'].forEach(t => {
|
||||
document.getElementById('tab-'+t).style.display = t===name?'block':'none';
|
||||
});
|
||||
document.querySelectorAll('.tab-bar .btn').forEach((b,i) => {
|
||||
b.classList.toggle('active', ['recent','alerts','entities','ingest'][i]===name);
|
||||
});
|
||||
if (name==='alerts') loadAlerts();
|
||||
if (name==='entities') loadEntities();
|
||||
}
|
||||
|
||||
async function ingestRSS() {
|
||||
const url = document.getElementById('rss-url').value;
|
||||
const r = await fetch(`${API}/api/ingest/rss?feed_url=${encodeURIComponent(url)}`, {method:'POST'});
|
||||
const d = await r.json();
|
||||
document.getElementById('ingest-result').textContent = `RSS: ${d.items_ingested} items ingested`;
|
||||
loadSummary(); loadEvents();
|
||||
}
|
||||
|
||||
async function ingestGDELT() {
|
||||
const r = await fetch(`${API}/api/ingest/gdelt?max_articles=50`, {method:'POST'});
|
||||
const d = await r.json();
|
||||
document.getElementById('ingest-result').textContent = `GDELT: ${d.articles_ingested} articles ingested`;
|
||||
loadSummary(); loadEvents();
|
||||
}
|
||||
|
||||
async function ingestEarthquakes() {
|
||||
const r = await fetch(`${API}/api/ingest/earthquakes`, {method:'POST'});
|
||||
const d = await r.json();
|
||||
document.getElementById('ingest-result').textContent = `USGS: ${d.events_ingested} events ingested`;
|
||||
loadSummary(); loadEvents();
|
||||
}
|
||||
|
||||
async function processNATS() {
|
||||
const r = await fetch(`${API}/api/ingest/process?batch_size=100`, {method:'POST'});
|
||||
const d = await r.json();
|
||||
document.getElementById('ingest-result').textContent = `NATS: ${d.processed} messages processed`;
|
||||
loadSummary(); loadEvents();
|
||||
}
|
||||
|
||||
async function checkHealth() {
|
||||
try {
|
||||
const r = await fetch(`${API}/api/health`);
|
||||
const d = await r.json();
|
||||
document.getElementById('health').textContent = 'healthy';
|
||||
document.getElementById('last-update').textContent = new Date().toLocaleTimeString();
|
||||
} catch(e) {
|
||||
document.getElementById('health').textContent = 'unreachable';
|
||||
document.getElementById('health').className = '';
|
||||
}
|
||||
}
|
||||
|
||||
// Initial load
|
||||
loadSummary(); loadEvents(); checkHealth();
|
||||
setInterval(() => { loadSummary(); loadEvents(); checkHealth(); }, 30000);
|
||||
</script>
|
||||
</body>
|
||||
</html>
|
||||
Loading…
Add table
Reference in a new issue