from pathlib import Path
from proxywhirl.config import discover_config, load_config

config_path = discover_config()
config = load_config(config_path)
config = load_config(Path("./custom-config.toml"))  # explicit

[tool.proxywhirl]
rotation_strategy = "round-robin"
timeout = 30
max_retries = 3
cache_enabled = true
cache_default_ttl = 3600
encrypt_credentials = true

[[tool.proxywhirl.proxies]]
url = "http://proxy1.example.com:8080"
username = "user1"
password = "pass1"

[tool.proxywhirl.data_storage]
pool_size = 5
persist_geo_data = false

export PROXYWHIRL_KEY=$(python -c "from cryptography.fernet import Fernet; print(Fernet.generate_key().decode())")

from pathlib import Path
from pydantic import SecretStr
from proxywhirl.config import CLIConfig, ProxyConfig, DataStorageConfig, load_config, save_config

config = CLIConfig(
    proxies=[ProxyConfig(url="http://proxy.example.com:8080", username=SecretStr("u"), password=SecretStr("p"))],
    rotation_strategy="weighted",
    timeout=45,
    data_storage=DataStorageConfig(pool_size=10, persist_geo_data=False),
)

save_config(config, Path("./pyproject.toml"))

[tool.proxywhirl]
rotation_strategy = "round-robin"
[[tool.proxywhirl.proxies]]
url = "http://proxy.example.com:8080"

config_path = discover_config()
print("No config" if config_path is None else f"Using {config_path}")

# Wrong: [tool.proxywhirl.data_storage] = {...}
# Right:
[tool.proxywhirl.data_storage]
persist_geo_data = false