import os
from bareunpy import Tagger

# ── 설정값 ───────────────────────────────────────────────
# API_KEY : bareun.ai에서 발급받은 본인 API 키. 소스에 직접 적지 말고
#           환경변수(os.environ["BAREUN_API_KEY"]) 등으로 주입하길 권장합니다.
# HOST    : 바른 서버 호스트. 로컬 설치본이면 "localhost".
# PORT    : 바른 서버 포트. 기본값은 5656.
# SRC_DIR : Git으로 관리하는 '내 단어 목록' 디렉토리(품사별 txt가 들어 있음).
# DOMAIN  : 바른에 등록·갱신할 사용자 사전 이름(요청 시 custom_dict_names로 골라 씀).
API_KEY = "koba-..."
HOST, PORT = "localhost", 5656
SRC_DIR = "custom-dict-src/legal"
DOMAIN = "legal"

def read_words(path: str) -> set:
    """품사별 txt 파일을 읽어 표제어 집합으로 돌려줍니다.

    txt 파일은 '한 줄에 한 단어' 구조입니다. 각 줄의 앞뒤 공백을 없애고,
    빈 줄과 '#'으로 시작하는 주석 줄은 건너뜁니다. 같은 단어가 여러 번
    나와도 집합(set)이라 자동으로 중복이 제거됩니다.

    Args:
        path (str): 읽어들일 txt 파일 경로(예: "custom-dict-src/legal/np.txt").

    Returns:
        set: 표제어 문자열의 집합. 파일이 없으면 빈 집합을 돌려줍니다
             (그 품사 사전을 비워서 등록하는 효과).
    """
    # 파일이 아직 없는 품사(예: va.txt 미작성)는 빈 집합으로 처리합니다.
    if not os.path.exists(path):
        return set()
    # UTF-8로 읽고, 공백 제거 후 빈 줄·주석(#)을 뺀 단어만 모읍니다.
    with open(path, encoding="utf-8") as fp:
        return {ln.strip() for ln in fp
                if ln.strip() and not ln.startswith("#")}

# 1) 품사별 txt 목록을 각각 집합으로 읽어들입니다.
np_set       = read_words(f"{SRC_DIR}/np.txt")        # 고유명사(NP)
cp_set       = read_words(f"{SRC_DIR}/cp.txt")        # 복합명사(CP)
cp_caret_set = read_words(f"{SRC_DIR}/cp-caret.txt")  # 복합명사 분리(CP^), ^로 분리 지점 표시
vv_set       = read_words(f"{SRC_DIR}/vv.txt")        # 동사(VV)
va_set       = read_words(f"{SRC_DIR}/va.txt")        # 형용사(VA)

# 2) 서버에 접속해 대상 사용자 사전(DOMAIN)을 열고, 품사별 집합으로 채웁니다.
#    copy_*_set은 해당 품사 사전의 '전체 내용을 이 집합으로 교체'합니다.
#    (누적이 아니라 대체이므로, txt가 곧 그 사전의 최종 상태가 됩니다.)
tagger = Tagger(API_KEY, HOST, PORT)
cust = tagger.custom_dict(DOMAIN)
cust.copy_np_set(np_set)
cust.copy_cp_set(cp_set)
cust.copy_cp_caret_set(cp_caret_set)
cust.copy_vv_set(vv_set)
cust.copy_va_set(va_set)

# 3) UpdateCustomDictionary API를 호출해 서버에 반영합니다.
#    호출 즉시 그 서버에 적용되고(무중단), 여러 서버가 사전 디렉토리를
#    공유하면 나머지 서버에도 곧바로 전파됩니다.
cust.update()
print(f"'{DOMAIN}' 사전 반영 완료: "
      f"NP {len(np_set)}, CP {len(cp_set)}, CP^ {len(cp_caret_set)}, "
      f"VV {len(vv_set)}, VA {len(va_set)}")

'legal' 사전 반영 완료: NP 3, CP 1, CP^ 1, VV 1, VA 0