markitect-main/markitect/matter_frontmatter/commands.py

"""
CLI commands for frontmatter operations.
"""

import click
import json
from pathlib import Path
from .parser import FrontmatterParser


@click.command('frontmatter-get')
@click.argument('key')
@click.option('--file', 'file_path', required=True, type=click.Path(exists=True),
              help='Path to markdown file')
@click.option('--format', 'output_format', default='raw', type=click.Choice(['raw', 'json']),
              help='Output format (raw or json)')
def frontmatter_get(key, file_path, output_format):
    """Get specific frontmatter value by key (supports dot notation for nested values)."""
    try:
        file_path = Path(file_path)
        with open(file_path, 'r', encoding='utf-8') as f:
            text = f.read()

        parser = FrontmatterParser()
        frontmatter = parser.extract_frontmatter(text)

        if not frontmatter:
            click.echo("No frontmatter found in document", err=True)
            return

        # Get value using dot notation if needed
        value = parser.get_nested_value(frontmatter, key)

        if value is None:
            click.echo(f"Key '{key}' not found in frontmatter", err=True)
            return

        if output_format == 'json':
            click.echo(json.dumps(value, indent=2))
        else:
            if isinstance(value, (dict, list)):
                click.echo(json.dumps(value, indent=2))
            else:
                click.echo(str(value))

    except Exception as e:
        click.echo(f"Error: {e}", err=True)
        raise click.ClickException(f"Failed to get frontmatter value from {file_path}")


@click.command('frontmatter-set')
@click.argument('key_value')
@click.option('--file', 'file_path', required=True, type=click.Path(exists=True),
              help='Path to markdown file')
@click.option('--backup', is_flag=True, help='Create backup of original file')
def frontmatter_set(key_value, file_path, backup):
    """Set frontmatter value (format: key=value, supports dot notation for nested)."""
    try:
        if '=' not in key_value:
            raise click.ClickException("Key-value must be in format 'key=value'")

        key, value = key_value.split('=', 1)
        key = key.strip()
        value = value.strip()

        # Try to parse value as JSON for complex types
        try:
            # Handle boolean and number values
            if value.lower() in ['true', 'false']:
                value = value.lower() == 'true'
            elif value.replace('.', '').replace('-', '').isdigit():
                value = float(value) if '.' in value else int(value)
            elif value.startswith('[') or value.startswith('{'):
                value = json.loads(value)
        except (json.JSONDecodeError, ValueError):
            # Keep as string if parsing fails
            pass

        file_path = Path(file_path)

        # Create backup if requested
        if backup:
            backup_path = file_path.with_suffix(f"{file_path.suffix}.bak")
            backup_path.write_text(file_path.read_text())
            click.echo(f"Backup created: {backup_path}")

        with open(file_path, 'r', encoding='utf-8') as f:
            text = f.read()

        parser = FrontmatterParser()
        new_text = parser.set_frontmatter_value(text, key, value)

        with open(file_path, 'w', encoding='utf-8') as f:
            f.write(new_text)

        click.echo(f"Set {key}={value} in {file_path}")

    except Exception as e:
        click.echo(f"Error: {e}", err=True)
        raise click.ClickException(f"Failed to set frontmatter value in {file_path}")


@click.command('frontmatter-keys')
@click.option('--file', 'file_path', required=True, type=click.Path(exists=True),
              help='Path to markdown file')
@click.option('--nested', is_flag=True, help='Include nested keys with dot notation')
@click.option('--format', 'output_format', default='list', type=click.Choice(['list', 'json']),
              help='Output format (list or json)')
def frontmatter_keys(file_path, nested, output_format):
    """List all frontmatter keys."""
    try:
        file_path = Path(file_path)
        with open(file_path, 'r', encoding='utf-8') as f:
            text = f.read()

        parser = FrontmatterParser()
        keys = parser.get_frontmatter_keys(text, include_nested=nested)

        if not keys:
            click.echo("No frontmatter keys found")
            return

        if output_format == 'json':
            click.echo(json.dumps(keys, indent=2))
        else:
            for key in sorted(keys):
                click.echo(key)

    except Exception as e:
        click.echo(f"Error: {e}", err=True)
        raise click.ClickException(f"Failed to list frontmatter keys from {file_path}")


@click.command('frontmatter-stats')
@click.option('--file', 'file_path', required=True, type=click.Path(exists=True),
              help='Path to markdown file')
@click.option('--format', 'output_format', default='json', type=click.Choice(['json', 'text']),
              help='Output format (json or text)')
def frontmatter_stats(file_path, output_format):
    """Calculate frontmatter statistics."""
    try:
        file_path = Path(file_path)
        with open(file_path, 'r', encoding='utf-8') as f:
            text = f.read()

        parser = FrontmatterParser()
        stats = parser.calculate_frontmatter_stats(text)

        if output_format == 'json':
            click.echo(json.dumps(stats.to_dict(), indent=2))
        else:
            click.echo(f"Has frontmatter: {stats.has_frontmatter}")
            click.echo(f"Total fields: {stats.total_fields}")
            click.echo(f"Nested fields: {stats.nested_fields}")
            click.echo(f"Format: {stats.format or 'N/A'}")

            if stats.field_types:
                click.echo("Field types:")
                for field_type, count in stats.field_types.items():
                    click.echo(f"  {field_type}: {count}")

    except Exception as e:
        click.echo(f"Error: {e}", err=True)
        raise click.ClickException(f"Failed to calculate frontmatter stats for {file_path}")