from flask import Flask, jsonify, request, send_from_directory
from datetime import datetime
import re
import urllib.request
from bs4 import BeautifulSoup
import os
from dotenv import load_dotenv

load_dotenv()

app = Flask(__name__, static_folder='../frontend', static_url_path='')

def get_schedule_data(url, target_date):
    try:
        with urllib.request.urlopen(url) as response:
            html_content = response.read().decode('utf-8', 'ignore')
    except Exception as e:
        return {"error": f"Failed to fetch data: {str(e)}"}

    try:
        soup = BeautifulSoup(html_content, 'lxml')
        table = soup.find('table', id='listtable')
        if not table:
            return {"error": "Table with id 'listtable' not found."}

        rows = table.find_all('tr')
        header_row = rows[0]
        header_cells = header_row.find_all('td')
        headers = [cell.get_text(strip=True) for cell in header_cells[1:]]

        tables = []
        for row in rows[1:]:
            cells = [cell.get_text(strip=True) for cell in row.find_all('td')]
            if not cells:
                continue

            date_cell_text = cells[0]
            date_match = re.match(r'(\d{1,2}/\d{1,2})', date_cell_text)
            if date_match:
                row_date = date_match.group(1)
                if row_date == target_date:
                    statuses = cells[1:]

                    maru_members = []
                    sankaku_members = []
                    for i, status in enumerate(statuses):
                        if i < len(headers):
                            member_name = headers[i]
                            if status == '○':
                                maru_members.append(member_name)
                            elif status == '△':
                                sankaku_members.append(member_name)

                    final_rows = []
                    if maru_members:
                        final_rows.append(['○'] + maru_members)
                    if sankaku_members:
                        final_rows.append(['△'] + sankaku_members)

                    if final_rows:
                        tables.append({
                            "title": date_cell_text,
                            "header": ["Status", "Members"],
                            "rows": final_rows
                        })

        return tables

    except Exception as e:
        import traceback
        return {"error": f"Failed to parse data: {traceback.format_exc()}"}

@app.route('/api')
def api():
    date_str = request.args.get('date', default=datetime.now().strftime('%Y-%m-%d'))
    try:
        target_date = datetime.strptime(date_str, '%Y-%m-%d').strftime('%-m/%-d')

        urls = [
            "https://densuke.biz/list?cd=uQYExVaQLJ4Dd2vB",
            "https://densuke.biz/list?cd=tfXDcX2hEfTkdW6t"
        ]

        all_tables = []
        for url in urls:
            result = get_schedule_data(url, target_date)
            if isinstance(result, dict) and "error" in result:
                return jsonify(result), 500
            all_tables.extend(result)

        return jsonify(all_tables)
    except Exception as e:
        return jsonify({"error": str(e)}), 400

@app.route('/')
def index():
    return send_from_directory(app.static_folder, 'index.html')

if __name__ == '__main__':
    port = int(os.environ.get('PORT', 8000))
    app.run(host='0.0.0.0', port=port)