# tools.py
import httpx
from langchain_core.tools import tool
import yfinance as yf
import json

@tool
def get_crypto_price(ticker: str) -> str:
    """Get the current price of a cryptocurrency (e.g., 'bitcoin', 'ethereum')."""
    try:
        url = f"https://api.coingecko.com/api/v3/simple/price?ids={ticker.lower()}&vs_currencies=usd"
        response = httpx.get(url, timeout=10)
        data = response.json()
        if ticker.lower() in data:
            price = data[ticker.lower()]["usd"]
            return f"The current price of {ticker} is ${price:,} USD."
        return f"Could not find price for {ticker}. Please check the coin name."
    except Exception as e:
        return f"Error fetching price: {str(e)}"

@tool
def get_thai_stock_price(symbol: str) -> str:
    """
    Get the current stock price of a Thailand stock (SET) by its symbol.
    Example input: 'PTT', 'CPALL', 'BDMS', 'ADVANC'
    """
    try:
        # จัดการชื่อหุ้นให้อยู่ในฟอร์แมตของ Yahoo Finance (เช่น PTT.BK)
        clean_symbol = symbol.strip().upper()
        if not clean_symbol.endswith(".BK"):
            yf_symbol = f"{clean_symbol}.BK"
        else:
            yf_symbol = clean_symbol

        # ดึงข้อมูลผ่าน Ticker
        stock = yf.Ticker(yf_symbol)
        
        # 💡 วิธีที่ชัวร์ที่สุด: ดึงประวัติราคาย้อนหลังสั้นที่สุด (1 วัน ระดับนาที) 
        # วิธีนี้จะได้ราคาตลาดล่าสุด (Close) และราคาปิดวันก่อนหน้า (Open ของแท่งวัน หรือดึงประวัติเพิ่ม)
        hist = stock.history(period="1d")
        
        # ถ้าไม่มีข้อมูลใน Dataframe แปลว่าชื่อหุ้นผิด หรือดึงข้อมูลไม่ได้
        if hist.empty:
            return f"ไม่พบข้อมูลราคาหุ้นของ {clean_symbol} กรุณาตรวจสอบชื่อย่อหุ้นอีกครั้ง"
            
        # ดึงราคาล่าสุดจากแถวสุดท้ายของคอลัมน์ 'Close'
        current_price = float(hist['Close'].iloc[-1])
        
        # ดึงข้อมูลเพิ่มเติมเพื่อหาเปอร์เซ็นต์การเปลี่ยนแปลง
        # โหลดเพิ่มอีกเล็กน้อยเพื่อความแม่นยำในการคำนวณราคาปิดวันก่อนหน้า
        hist_5d = stock.history(period="5d")
        if len(hist_5d) >= 2:
            # ราคาปิดของวันทำการก่อนหน้า (แถวก่อนแถวสุดท้าย)
            prev_close = float(hist_5d['Close'].iloc[-2])
        else:
            prev_close = current_price

        # คำนวณส่วนต่าง
        change = current_price - prev_close
        change_percent = (change / prev_close) * 100 if prev_close else 0

        # จัดฟอร์แมตข้อความแสดงผล
        direction = "📈" if change >= 0 else "📉"
        return (
            f"ราคาหุ้น {clean_symbol} ล่าสุดในตลาด SET อยู่ที่ {current_price:,.2f} บาท "
            f"({direction} {change:+,.2f} / {change_percent:+.2f}%)"
        )
        
    except Exception as e:
        return f"เกิดข้อผิดพลาดในการดึงข้อมูลหุ้น {symbol}: {str(e)}"

@tool
def generate_comparison_flex_message(title: str, items_json: str) -> str:
    """
    Generate a LINE Flex Message JSON string for comparing multiple stocks or cryptos in a table format.
    Use this tool ONLY when the user wants to compare, list, or summarize multiple items (stocks/cryptos) together.
    
    Parameters:
    - title: The title of the comparison (e.g., 'เปรียบเทียบหุ้นกลุ่มธนาคาร', 'สรุปราคาคริปโตยอดฮิต')
    - items_json: A JSON string containing a list of dictionaries with keys: 'name', 'price', and 'change'.
      Example: '[{"name": "PTT", "price": "34.25", "change": "+0.50"}, {"name": "CPALL", "price": "65.00", "change": "-0.25"}]'
    """
    try:
        # แปลงข้อมูล String JSON ที่ LLM ส่งมาให้กลายเป็น Python List
        items = json.loads(items_json)
        
        # 1. สร้างแถวของตาราง (Table Rows) จากข้อมูลที่ได้
        table_rows = []
        for item in items:
            name = item.get("name", "-")
            price = item.get("price", "-")
            change = item.get("change", "-")
            
            # 🚨 [เพิ่มจุดนี้] ดักจับป้องกันกรณีที่ LLM ส่งค่าว่าง "" มาให้เปลี่ยนเป็น "-" แทน
            if not change or change.strip() == "":
                change = "-"
            
            # กำหนดสีตามการเปลี่ยนแปลง (บวก=เขียว, ลบ=แดง, เท่าเดิม/ไม่มีข้อมูล=เทา)
            color = "#27ae60" if "+" in change else ("#c0392b" if "-" in change and change != "-" else "#555555")
            
            row_box = {
                "type": "box",
                "layout": "horizontal",
                "margin": "md",
                "contents": [
                    {"type": "text", "text": name, "size": "sm", "color": "#111111", "flex": 3, "weight": "bold"},
                    {"type": "text", "text": f"{price}", "size": "sm", "color": "#444444", "flex": 4, "align": "end"},
                    {"type": "text", "text": change, "size": "sm", "color": color, "flex": 3, "align": "end", "weight": "bold"}
                ]
            }
            table_rows.append(row_box)

        # 2. ประกอบโครงสร้าง Flex Message (Bubble) ตามดีไซน์ของ LINE
        flex_bubble = {
            "type": "bubble",
            "body": {
                "type": "box",
                "layout": "vertical",
                "contents": [
                    # หัวข้อ (Header)
                    {"type": "text", "text": title, "weight": "bold", "size": "md", "color": "#1ca7ec"},
                    {"type": "separator", "margin": "md"},
                    
                    # หัวตาราง (Table Header)
                    {
                        "type": "box",
                        "layout": "horizontal",
                        "margin": "md",
                        "contents": [
                            {"type": "text", "text": "ชื่อ", "size": "xs", "color": "#aaaaaa", "flex": 3, "weight": "bold"},
                            {"type": "text", "text": "ราคาล่าสุด", "size": "xs", "color": "#aaaaaa", "flex": 4, "align": "end", "weight": "bold"},
                            {"type": "text", "text": "เปลี่ยนแปลง", "size": "xs", "color": "#aaaaaa", "flex": 3, "align": "end", "weight": "bold"}
                        ]
                    },
                    {"type": "separator", "margin": "sm"}
                ] + table_rows # นำข้อมูลแถวที่สร้างไว้มาต่อท้ายหัวตาราง
            }
        }
        
        # ส่งโครงสร้าง Flex Container กลับไปในรูปแบบ String JSON พิเศษเพื่อระบุว่าเป็น Flex
        # เราจะครอบด้วยคำว่า FLEX_JSON: เพื่อให้ไฟล์ app.py ตรวจจับได้ง่าย
        return f"FLEX_JSON:{json.dumps(flex_bubble, ensure_ascii=False)}"
        
    except Exception as e:
        return f"Error generating Flex Message: {str(e)}"

# รวบรวม Tools ทั้งหมดไว้ใน List เพื่อให้ Agent รู้จัก
tools_list = [get_crypto_price, get_thai_stock_price, generate_comparison_flex_message]