#!/usr/bin/env python3
"""
백엔드 코드 전체 분석 후 누락된 모든 컬럼과 테이블을 한 번에 수정하는 스크립트
"""

import os
import sys
import psycopg2
from psycopg2 import sql

# 현재 디렉토리를 Python 경로에 추가
sys.path.insert(0, '/app')

# 환경 변수 로드
DB_HOST = os.getenv("DB_HOST", "postgres")
DB_PORT = os.getenv("DB_PORT", "5432")
DB_NAME = os.getenv("DB_NAME", "tk_mp_bom")
DB_USER = os.getenv("DB_USER", "tkmp_user")
DB_PASSWORD = os.getenv("DB_PASSWORD", "tkmp_password_2025")

def fix_all_missing_columns():
    """백엔드 코드 분석 결과를 바탕으로 모든 누락된 컬럼 추가"""
    print("🔍 백엔드 코드 분석 결과를 바탕으로 모든 누락된 컬럼 수정 시작...")
    
    try:
        conn = psycopg2.connect(
            host=DB_HOST,
            port=DB_PORT,
            database=DB_NAME,
            user=DB_USER,
            password=DB_PASSWORD
        )
        
        with conn.cursor() as cursor:
            # 1. materials 테이블 누락된 컬럼들
            print("📝 materials 테이블 컬럼 추가 중...")
            materials_columns = [
                "ALTER TABLE materials ADD COLUMN IF NOT EXISTS brand VARCHAR(100);",
                "ALTER TABLE materials ADD COLUMN IF NOT EXISTS user_requirement TEXT;",
                "ALTER TABLE materials ADD COLUMN IF NOT EXISTS purchase_confirmed BOOLEAN DEFAULT FALSE;",
                "ALTER TABLE materials ADD COLUMN IF NOT EXISTS confirmed_quantity NUMERIC(10,3);",
                "ALTER TABLE materials ADD COLUMN IF NOT EXISTS purchase_status VARCHAR(20);",
                "ALTER TABLE materials ADD COLUMN IF NOT EXISTS purchase_confirmed_by VARCHAR(100);",
                "ALTER TABLE materials ADD COLUMN IF NOT EXISTS purchase_confirmed_at TIMESTAMP;",
                "ALTER TABLE materials ADD COLUMN IF NOT EXISTS material_hash VARCHAR(64);",
                "ALTER TABLE materials ADD COLUMN IF NOT EXISTS normalized_description TEXT;",
                "ALTER TABLE materials ADD COLUMN IF NOT EXISTS revision_status VARCHAR(20);",
            ]
            
            for sql_cmd in materials_columns:
                cursor.execute(sql_cmd)
            print("✅ materials 테이블 컬럼 추가 완료")
            
            # 2. material_purchase_tracking 테이블 누락된 컬럼들 (백엔드 코드에서 사용됨)
            print("📝 material_purchase_tracking 테이블 컬럼 추가 중...")
            mpt_columns = [
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS job_no VARCHAR(50);",
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS material_hash VARCHAR(64);",
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS revision VARCHAR(20);",
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS description TEXT;",
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS size_spec VARCHAR(50);",
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS unit VARCHAR(10);",
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS bom_quantity NUMERIC(10,3);",
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS calculated_quantity NUMERIC(10,3);",
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS supplier_name VARCHAR(100);",
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS unit_price NUMERIC(12,2);",
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS total_price NUMERIC(15,2);",
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS order_date DATE;",
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS delivery_date DATE;",
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS confirmed_by VARCHAR(100);",
                "ALTER TABLE material_purchase_tracking ADD COLUMN IF NOT EXISTS confirmed_at TIMESTAMP;",
            ]
            
            for sql_cmd in mpt_columns:
                cursor.execute(sql_cmd)
            print("✅ material_purchase_tracking 테이블 컬럼 추가 완료")
            
            # 3. files 테이블 누락된 컬럼들
            print("📝 files 테이블 컬럼 추가 중...")
            files_columns = [
                "ALTER TABLE files ADD COLUMN IF NOT EXISTS updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP;",
                "ALTER TABLE files ADD COLUMN IF NOT EXISTS project_type VARCHAR(50);",
            ]
            
            for sql_cmd in files_columns:
                cursor.execute(sql_cmd)
            print("✅ files 테이블 컬럼 추가 완료")
            
            # 4. pipe_details 테이블 누락된 컬럼들 (백엔드 코드에서 사용됨)
            print("📝 pipe_details 테이블 컬럼 추가 중...")
            pipe_details_columns = [
                "ALTER TABLE pipe_details ADD COLUMN IF NOT EXISTS material_id INTEGER REFERENCES materials(id);",
                "ALTER TABLE pipe_details ADD COLUMN IF NOT EXISTS outer_diameter VARCHAR(50);",
                "ALTER TABLE pipe_details ADD COLUMN IF NOT EXISTS material_spec VARCHAR(100);",
                "ALTER TABLE pipe_details ADD COLUMN IF NOT EXISTS classification_confidence NUMERIC(3,2);",
            ]
            
            for sql_cmd in pipe_details_columns:
                cursor.execute(sql_cmd)
            print("✅ pipe_details 테이블 컬럼 추가 완료")
            
            # 5. 기타 누락된 테이블들의 컬럼 추가
            print("📝 기타 테이블들 컬럼 추가 중...")
            
            # fitting_details 테이블 컬럼 추가
            cursor.execute("ALTER TABLE fitting_details ADD COLUMN IF NOT EXISTS main_size VARCHAR(50);")
            cursor.execute("ALTER TABLE fitting_details ADD COLUMN IF NOT EXISTS reduced_size VARCHAR(50);")
            cursor.execute("ALTER TABLE fitting_details ADD COLUMN IF NOT EXISTS length_mm NUMERIC(10,3);")
            
            # gasket_details 테이블 컬럼 추가
            cursor.execute("ALTER TABLE gasket_details ADD COLUMN IF NOT EXISTS gasket_subtype VARCHAR(50);")
            cursor.execute("ALTER TABLE gasket_details ADD COLUMN IF NOT EXISTS material_type VARCHAR(50);")
            cursor.execute("ALTER TABLE gasket_details ADD COLUMN IF NOT EXISTS size_inches VARCHAR(50);")
            cursor.execute("ALTER TABLE gasket_details ADD COLUMN IF NOT EXISTS temperature_range VARCHAR(50);")
            cursor.execute("ALTER TABLE gasket_details ADD COLUMN IF NOT EXISTS fire_safe BOOLEAN DEFAULT FALSE;")
            
            print("✅ 기타 테이블들 컬럼 추가 완료")
            
            conn.commit()
            print("✅ 모든 누락된 컬럼 추가 완료")
            
    except Exception as e:
        print(f"❌ 컬럼 추가 실패: {e}")
        return False
    finally:
        if conn:
            conn.close()
    
    return True

if __name__ == "__main__":
    print("🚀 백엔드 코드 분석 기반 스키마 수정 시작")
    success = fix_all_missing_columns()
    if success:
        print("✅ 모든 스키마 수정 완료")
    else:
        print("❌ 스키마 수정 실패")
        sys.exit(1)