import os
import re
import subprocess
from dotenv import load_dotenv
from concurrent.futures import ThreadPoolExecutor, as_completed

# .env 파일에서 사용자 이름 및 비밀번호 설정
load_dotenv()
IDRAC_USER = os.getenv("IDRAC_USER")
IDRAC_PASS = os.getenv("IDRAC_PASS")

def fetch_idrac_info(idrac_ip, output_dir):
    try:
        # 서비스 태그 가져오기 (get 제외)
        cmd_getsysinfo = f"racadm -r {idrac_ip} -u {IDRAC_USER} -p {IDRAC_PASS} getsysinfo"
        getsysinfo = subprocess.getoutput(cmd_getsysinfo)
        svc_tag_match = re.search(r"SVC Tag\s*=\s*(\S+)", getsysinfo)
        svc_tag = svc_tag_match.group(1) if svc_tag_match else None

        if not svc_tag:
            print(f"Failed to retrieve SVC Tag for IP: {idrac_ip}")
            return

        # InfiniBand.VndrConfigPage 목록 가져오기
        cmd_list = f"racadm -r {idrac_ip} -u {IDRAC_USER} -p {IDRAC_PASS} get InfiniBand.VndrConfigPage"
        output_list = subprocess.getoutput(cmd_list)

        # InfiniBand.VndrConfigPage.<숫자> 및 Key 값 추출
        matches = re.findall(r"InfiniBand\.VndrConfigPage\.(\d+)\s+\[Key=InfiniBand\.Slot\.(\d+)-\d+#VndrConfigPage]", output_list)

        # 결과를 저장할 파일 생성
        output_file = os.path.join(output_dir, f"{svc_tag}.txt")
        os.makedirs(output_dir, exist_ok=True)

        with open(output_file, "w") as f:
            # 서비스 태그 저장
            f.write(f"{svc_tag}\n")

            # 모든 PortGUID를 저장할 리스트
            hex_guid_list = []

            # 각 InfiniBand.VndrConfigPage.<숫자> 처리
            for number, slot in matches:
                cmd_detail = f"racadm -r {idrac_ip} -u {IDRAC_USER} -p {IDRAC_PASS} get InfiniBand.VndrConfigPage.{number}"
                output_detail = subprocess.getoutput(cmd_detail)

                # PortGUID 값 추출
                match_guid = re.search(r"PortGUID=(\S+)", output_detail)
                port_guid = match_guid.group(1) if match_guid else "Not Found"

                # Slot.<숫자>: <PortGUID> 형식으로 저장
                f.write(f"Slot.{slot}: {port_guid}\n")

                # PortGUID를 0x 형식으로 변환하여 리스트에 추가
                if port_guid != "Not Found":
                    hex_guid_list.append(f"0x{port_guid.replace(':', '').upper()}")

            # 모든 PortGUID를 "GUID: 0x<GUID1>;0x<GUID2>" 형식으로 저장
            if hex_guid_list:
                f.write(f"GUID: {';'.join(hex_guid_list)}\n")

    except Exception as e:
        print(f"Error processing IP {idrac_ip}: {e}")

def main(ip_file):
    if not os.path.isfile(ip_file):
        print(f"IP file {ip_file} does not exist.")
        return

    output_dir = "/app/idrac_info/idrac_info"
    os.makedirs(output_dir, exist_ok=True)

    with open(ip_file, "r") as file:
        ip_addresses = [line.strip() for line in file.readlines()]

    with ThreadPoolExecutor(max_workers=100) as executor:
        future_to_ip = {executor.submit(fetch_idrac_info, ip, output_dir): ip for ip in ip_addresses}

        for future in as_completed(future_to_ip):
            try:
                future.result()
            except Exception as e:
                print(f"Error processing task: {e}")

if __name__ == "__main__":
    import sys
    if len(sys.argv) != 2:
        print("Usage: python script.py <ip_file>")
        sys.exit(1)

    ip_file = sys.argv[1]
    main(ip_file)
