log/src/commands/post.rs

use anyhow::{Context, Result};
use rand::Rng;
use serde::{Deserialize, Serialize};
use serde_json::Value;
use std::fs;

use super::auth;
use crate::lexicons::{self, com_atproto_repo, com_atproto_identity};

#[derive(Debug, Serialize)]
struct PutRecordRequest {
    repo: String,
    collection: String,
    rkey: String,
    record: Value,
}

#[derive(Debug, Serialize)]
struct DeleteRecordRequest {
    repo: String,
    collection: String,
    rkey: String,
}

#[derive(Debug, Deserialize)]
struct PutRecordResponse {
    uri: String,
    cid: String,
}

#[derive(Debug, Deserialize)]
struct ListRecordsResponse {
    records: Vec<Record>,
    #[serde(default)]
    #[allow(dead_code)]
    cursor: Option<String>,
}

#[derive(Debug, Deserialize)]
struct Record {
    uri: String,
    cid: String,
    value: Value,
}

/// Generate TID (timestamp-based ID)
fn generate_tid() -> String {
    const CHARSET: &[u8] = b"234567abcdefghijklmnopqrstuvwxyz";
    let mut rng = rand::thread_rng();
    (0..13)
        .map(|_| {
            let idx = rng.gen_range(0..CHARSET.len());
            CHARSET[idx] as char
        })
        .collect()
}

/// Put a record to ATProto
pub async fn put_record(file: &str, collection: &str, rkey: Option<&str>) -> Result<()> {
    let session = auth::refresh_session().await?;
    let pds = session.pds.as_deref().unwrap_or("bsky.social");

    let content = fs::read_to_string(file)
        .with_context(|| format!("Failed to read file: {}", file))?;
    let record: Value = serde_json::from_str(&content)?;

    let rkey = rkey.map(|s| s.to_string()).unwrap_or_else(generate_tid);

    let client = reqwest::Client::new();
    let url = lexicons::url(pds, &com_atproto_repo::PUT_RECORD);

    let req = PutRecordRequest {
        repo: session.did.clone(),
        collection: collection.to_string(),
        rkey: rkey.clone(),
        record,
    };

    println!("Posting to {} with rkey: {}", collection, rkey);
    println!("{}", serde_json::to_string_pretty(&req)?);

    let res = client
        .post(&url)
        .header("Authorization", format!("Bearer {}", session.access_jwt))
        .json(&req)
        .send()
        .await?;

    if !res.status().is_success() {
        let status = res.status();
        let body = res.text().await.unwrap_or_default();
        anyhow::bail!("Put record failed: {} - {}", status, body);
    }

    let result: PutRecordResponse = res.json().await?;
    println!("Success!");
    println!("  URI: {}", result.uri);
    println!("  CID: {}", result.cid);

    Ok(())
}

/// Put a lexicon schema
pub async fn put_lexicon(file: &str) -> Result<()> {
    let session = auth::refresh_session().await?;
    let pds = session.pds.as_deref().unwrap_or("bsky.social");

    let content = fs::read_to_string(file)
        .with_context(|| format!("Failed to read file: {}", file))?;
    let lexicon: Value = serde_json::from_str(&content)?;

    let lexicon_id = lexicon["id"]
        .as_str()
        .context("Lexicon file must have 'id' field")?
        .to_string();

    let client = reqwest::Client::new();
    let url = lexicons::url(pds, &com_atproto_repo::PUT_RECORD);

    let req = PutRecordRequest {
        repo: session.did.clone(),
        collection: "com.atproto.lexicon.schema".to_string(),
        rkey: lexicon_id.clone(),
        record: lexicon,
    };

    println!("Putting lexicon: {}", lexicon_id);
    println!("{}", serde_json::to_string_pretty(&req)?);

    let res = client
        .post(&url)
        .header("Authorization", format!("Bearer {}", session.access_jwt))
        .json(&req)
        .send()
        .await?;

    if !res.status().is_success() {
        let status = res.status();
        let body = res.text().await.unwrap_or_default();
        anyhow::bail!("Put lexicon failed: {} - {}", status, body);
    }

    let result: PutRecordResponse = res.json().await?;
    println!("Success!");
    println!("  URI: {}", result.uri);
    println!("  CID: {}", result.cid);

    Ok(())
}

/// Get records from a collection
pub async fn get_records(collection: &str, limit: u32) -> Result<()> {
    let session = auth::refresh_session().await?;
    let pds = session.pds.as_deref().unwrap_or("bsky.social");

    let client = reqwest::Client::new();
    let base_url = lexicons::url(pds, &com_atproto_repo::LIST_RECORDS);
    let url = format!(
        "{}?repo={}&collection={}&limit={}",
        base_url, session.did, collection, limit
    );

    let res = client
        .get(&url)
        .header("Authorization", format!("Bearer {}", session.access_jwt))
        .send()
        .await?;

    if !res.status().is_success() {
        let status = res.status();
        let body = res.text().await.unwrap_or_default();
        anyhow::bail!("Get records failed: {} - {}", status, body);
    }

    let result: ListRecordsResponse = res.json().await?;

    println!("Found {} records in {}", result.records.len(), collection);
    for record in &result.records {
        println!("---");
        println!("URI: {}", record.uri);
        println!("CID: {}", record.cid);
        println!("{}", serde_json::to_string_pretty(&record.value)?);
    }

    Ok(())
}

/// Delete a record
pub async fn delete_record(collection: &str, rkey: &str) -> Result<()> {
    let session = auth::refresh_session().await?;
    let pds = session.pds.as_deref().unwrap_or("bsky.social");

    let client = reqwest::Client::new();
    let url = lexicons::url(pds, &com_atproto_repo::DELETE_RECORD);

    let req = DeleteRecordRequest {
        repo: session.did.clone(),
        collection: collection.to_string(),
        rkey: rkey.to_string(),
    };

    println!("Deleting {} from {}", rkey, collection);

    let res = client
        .post(&url)
        .header("Authorization", format!("Bearer {}", session.access_jwt))
        .json(&req)
        .send()
        .await?;

    if !res.status().is_success() {
        let status = res.status();
        let body = res.text().await.unwrap_or_default();
        anyhow::bail!("Delete failed: {} - {}", status, body);
    }

    println!("Deleted successfully");

    Ok(())
}

#[derive(Debug, Deserialize)]
struct Config {
    handle: String,
    #[serde(default)]
    collection: Option<String>,
}

#[derive(Debug, Deserialize)]
struct DescribeRepoResponse {
    did: String,
    handle: String,
    collections: Vec<String>,
}

/// Sync PDS data to local content directory
pub async fn sync_to_local(output: &str) -> Result<()> {
    let config_content = fs::read_to_string("public/config.json")
        .context("config.json not found")?;
    let config: Config = serde_json::from_str(&config_content)?;

    println!("Syncing data for {}", config.handle);

    let client = reqwest::Client::new();

    // Resolve handle to DID
    let resolve_url = format!(
        "{}?handle={}",
        lexicons::url("public.api.bsky.app", &com_atproto_identity::RESOLVE_HANDLE),
        config.handle
    );
    let res = client.get(&resolve_url).send().await?;
    let resolve: serde_json::Value = res.json().await?;
    let did = resolve["did"].as_str().context("Could not resolve handle")?;

    println!("DID: {}", did);

    // Get PDS from DID document
    let plc_url = format!("https://plc.directory/{}", did);
    let res = client.get(&plc_url).send().await?;
    let did_doc: serde_json::Value = res.json().await?;
    let pds = did_doc["service"]
        .as_array()
        .and_then(|services| {
            services.iter().find(|s| s["type"] == "AtprotoPersonalDataServer")
        })
        .and_then(|s| s["serviceEndpoint"].as_str())
        .context("Could not find PDS")?;

    println!("PDS: {}", pds);

    // Remove https:// prefix for lexicons::url
    let pds_host = pds.trim_start_matches("https://");

    // Create output directory
    let did_dir = format!("{}/{}", output, did);
    fs::create_dir_all(&did_dir)?;

    // 1. Sync describeRepo
    let describe_url = format!(
        "{}?repo={}",
        lexicons::url(pds_host, &com_atproto_repo::DESCRIBE_REPO),
        did
    );
    let res = client.get(&describe_url).send().await?;
    let describe: DescribeRepoResponse = res.json().await?;

    let describe_path = format!("{}/describe.json", did_dir);
    let describe_json = serde_json::to_string_pretty(&serde_json::json!({
        "did": describe.did,
        "handle": describe.handle,
        "collections": describe.collections,
    }))?;
    fs::write(&describe_path, &describe_json)?;
    println!("Saved: {}", describe_path);

    // 2. Sync profile
    let profile_url = format!(
        "{}?repo={}&collection=app.bsky.actor.profile&rkey=self",
        lexicons::url(pds_host, &com_atproto_repo::GET_RECORD),
        did
    );
    let res = client.get(&profile_url).send().await?;
    if res.status().is_success() {
        let profile: serde_json::Value = res.json().await?;
        let profile_dir = format!("{}/app.bsky.actor.profile", did_dir);
        fs::create_dir_all(&profile_dir)?;
        let profile_path = format!("{}/self.json", profile_dir);
        fs::write(&profile_path, serde_json::to_string_pretty(&profile)?)?;
        println!("Saved: {}", profile_path);
    }

    // 3. Sync collection records
    let collection = config.collection.as_deref().unwrap_or("ai.syui.log.post");
    let records_url = format!(
        "{}?repo={}&collection={}&limit=100",
        lexicons::url(pds_host, &com_atproto_repo::LIST_RECORDS),
        did, collection
    );
    let res = client.get(&records_url).send().await?;
    if res.status().is_success() {
        let list: ListRecordsResponse = res.json().await?;
        let collection_dir = format!("{}/{}", did_dir, collection);
        fs::create_dir_all(&collection_dir)?;

        for record in &list.records {
            let rkey = record.uri.split('/').last().unwrap_or("unknown");
            let record_path = format!("{}/{}.json", collection_dir, rkey);
            let record_json = serde_json::json!({
                "uri": record.uri,
                "cid": record.cid,
                "value": record.value,
            });
            fs::write(&record_path, serde_json::to_string_pretty(&record_json)?)?;
            println!("Saved: {}", record_path);
        }
        println!("Synced {} records from {}", list.records.len(), collection);
    }

    println!("Sync complete!");

    Ok(())
}