tenk 0.1.0

//! Stock data example.

use tenk::sources::{EastMoneySource, SinaSource, THSSource};
use tenk::{DataClient, KLineType};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    tracing_subscriber::fmt()
        .with_max_level(tracing::Level::INFO)
        .init();

    println!("=== tenk Stock Data Example ===\n");

    // Create client with all sources for fallback
    let client = DataClient::new()
        .with_source(EastMoneySource::default())
        .with_source(SinaSource::default())
        .with_source(THSSource::default());

    // 1. Get all stock codes
    println!("1. Fetching all stock codes...");
    let codes = client.get_all_codes().await?;
    println!("   Found {} stocks", codes.len());
    
    // Show stocks by exchange
    let sh_count = codes.iter().filter(|c| c.exchange.to_string() == "SH").count();
    let sz_count = codes.iter().filter(|c| c.exchange.to_string() == "SZ").count();
    println!("   - Shanghai (SH): {}", sh_count);
    println!("   - Shenzhen (SZ): {}", sz_count);
    println!();

    // 2. Get historical K-line data (Daily)
    println!("2. Fetching daily K-line for 300059 (东方财富)...");
    let daily = client
        .get_market("300059", Some("2025-01-01"), Some("2025-01-31"), KLineType::Daily)
        .await?;
    println!("   Fetched {} daily records", daily.len());
    if let Some(first) = daily.first() {
        println!("   First: {} O:{:.2} H:{:.2} L:{:.2} C:{:.2}", 
            first.trade_date, first.open, first.high, first.low, first.close);
    }
    println!();

    // 3. Get weekly K-line data
    println!("3. Fetching weekly K-line for 600519 (贵州茅台)...");
    let weekly = client
        .get_market("600519", Some("2025-01-01"), Some("2025-01-31"), KLineType::Weekly)
        .await?;
    println!("   Fetched {} weekly records", weekly.len());
    for record in weekly.iter().take(3) {
        println!("   {} - Close: {:.2}, Change: {:+.2}%", 
            record.trade_date, record.close, record.change_pct);
    }
    println!();

    // 4. Get current market prices
    println!("4. Fetching current prices for multiple stocks...");
    let stocks = ["300059", "600519"];
    let current = client.get_market_current(&stocks).await?;
    println!("   {:12} {:12} {:>10} {:>10}", "Code", "Name", "Price", "Change%");
    println!("   {}", "-".repeat(48));
    for data in &current {
        let arrow = if data.change_pct >= 0.0 { "↑" } else { "↓" };
        println!("   {:12} {:12} {:>10.2} {:>9.2}%{}", 
            data.stock_code, data.short_name, data.price, data.change_pct.abs(), arrow);
    }
    println!();

    // 5. Get minute data
    println!("5. Fetching minute data for 600519...");
    let minutes = client.get_market_min("600519").await?;
    println!("   Fetched {} minute records", minutes.len());
    if !minutes.is_empty() {
        if let Some(first) = minutes.first() {
            println!("   First: {} - Price: {:.2}", 
                first.trade_time.format("%H:%M"), first.price);
        }
        if let Some(last) = minutes.last() {
            println!("   Last:  {} - Price: {:.2}", 
                last.trade_time.format("%H:%M"), last.price);
        }
        
        // Calculate price range
        let prices: Vec<f64> = minutes.iter().map(|m| m.price).collect();
        let min_price = prices.iter().cloned().fold(f64::INFINITY, f64::min);
        let max_price = prices.iter().cloned().fold(f64::NEG_INFINITY, f64::max);
        println!("   Range: {:.2} - {:.2}", min_price, max_price);
    }
    println!();

    // 6. Get stock info
    println!("6. Fetching stock info from different sources...");
    let test_stocks = ["300059", "600519"];
    for code in test_stocks {
        match client.get_stock_info(code).await {
            Ok(info) => {
                println!("   {} ({}):", code, info.short_name);
                if let Some(shares) = info.total_shares {
                    println!("      Total shares: {:.0}", shares);
                }
                if let Some(circ) = info.circulating_shares {
                    println!("      Circulating: {:.0}", circ);
                }
            }
            Err(e) => println!("   {}: Error - {}", code, e),
        }
    }
    println!();

    // 7. Test THS source directly
    println!("7. Testing THS source directly...");
    let ths = THSSource::default();
    use tenk::traits::StockMarketSource;
    
    // Get K-line from THS
    match ths.get_market("600519", Some("2025-01-01"), Some("2025-01-31"), KLineType::Daily).await {
        Ok(data) => {
            println!("   THS K-line for 600519: {} records", data.len());
            if let Some(first) = data.first() {
                println!("   First: {} - Close: {:.2}", first.trade_date, first.close);
            }
        }
        Err(e) => println!("   THS K-line error: {}", e),
    }
    
    // Get current from THS
    match ths.get_market_current(&["600519", "300059"]).await {
        Ok(data) => {
            println!("   THS current quotes: {} records", data.len());
            for d in &data {
                println!("   {} ({}): {:.2}", d.stock_code, d.short_name, d.price);
            }
        }
        Err(e) => println!("   THS current error: {}", e),
    }
    
    // Get minute from THS
    match ths.get_market_min("600519").await {
        Ok(data) => {
            println!("   THS minute for 600519: {} records", data.len());
        }
        Err(e) => println!("   THS minute error: {}", e),
    }

    println!("\nDone!");
    Ok(())
}