import click
import requests

from .cli_main import main, auth_header, base_url


@main.group("pillar")
def pillar():
    pass


@pillar.command("get")
@click.argument("fqdn")
def pillar_get(fqdn):
    """Get pillar data for a given FQDN."""
    try:
        response = requests.get(
            f"{base_url()}/pillar/{fqdn}",
            headers=auth_header(),
        )
        response.raise_for_status()
        pillar_data = response.json()
        click.echo(pillar_data)
    except requests.exceptions.RequestException as e:
        click.echo(f"Error: {e}")