59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import argparse
|
|
from getpass import getpass
|
|
from pathlib import Path
|
|
|
|
from civilplan_mcp.secure_store import default_key_store_path, save_api_keys
|
|
|
|
|
|
def _parse_env_file(path: Path) -> dict[str, str]:
|
|
parsed: dict[str, str] = {}
|
|
for line in path.read_text(encoding="utf-8").splitlines():
|
|
stripped = line.strip()
|
|
if not stripped or stripped.startswith("#") or "=" not in stripped:
|
|
continue
|
|
key, value = stripped.split("=", 1)
|
|
parsed[key.strip()] = value.strip().strip("'").strip('"')
|
|
return parsed
|
|
|
|
|
|
def _prompt_value(name: str, current: str = "") -> str:
|
|
prompt = f"{name}"
|
|
if current:
|
|
prompt += " [press Enter to keep imported value]"
|
|
prompt += ": "
|
|
entered = getpass(prompt).strip()
|
|
return entered or current
|
|
|
|
|
|
def main(argv: list[str] | None = None) -> int:
|
|
parser = argparse.ArgumentParser(description="Store CivilPlan API keys in Windows DPAPI encrypted local storage.")
|
|
parser.add_argument(
|
|
"--from-env-file",
|
|
type=Path,
|
|
help="Import API keys from an existing .env file before prompting.",
|
|
)
|
|
args = parser.parse_args(argv)
|
|
|
|
imported: dict[str, str] = {}
|
|
if args.from_env_file:
|
|
imported = _parse_env_file(args.from_env_file)
|
|
|
|
data_go_kr_api_key = _prompt_value("DATA_GO_KR_API_KEY", imported.get("DATA_GO_KR_API_KEY", ""))
|
|
vworld_api_key = _prompt_value("VWORLD_API_KEY", imported.get("VWORLD_API_KEY", ""))
|
|
gemini_api_key = _prompt_value("GEMINI_API_KEY", imported.get("GEMINI_API_KEY", ""))
|
|
|
|
target = save_api_keys(
|
|
{
|
|
"DATA_GO_KR_API_KEY": data_go_kr_api_key,
|
|
"VWORLD_API_KEY": vworld_api_key,
|
|
"GEMINI_API_KEY": gemini_api_key,
|
|
}
|
|
)
|
|
|
|
print("CivilPlan API keys were saved to encrypted local storage.")
|
|
print(f"Path: {target}")
|
|
print("The file is protected with Windows DPAPI and can only be read by the same Windows user on this machine.")
|
|
return 0
|