imexp 0.1.0__py3-none-win_amd64.whl

imexp/__main__.py

@@ -0,0 +1,7 @@
+"""Module entrypoint for python -m imexp."""
+
+from imexp.cli.main import main
+
+
+if __name__ == "__main__":
+    main()

imexp/cli/config.py

@@ -0,0 +1,180 @@
+"""Configuration loading and management."""
+
+import os
+import logging
+import configparser
+from pathlib import Path
+from dataclasses import dataclass
+
+logger = logging.getLogger("imexp")
+
+IOS_BACKUP_ROOT = Path("~/Library/Application Support/MobileSync/Backup").expanduser()
+CONFIG_FILE = "cli/config.ini"
+
+
+@dataclass(frozen=True)
+class ExportDefaults:
+    """User-configurable export defaults from config.ini."""
+
+    platform: str
+    format: str
+    copy_method: str
+    conversation_filter: str
+    use_caller_id: bool
+    output_dir: str
+
+
+@dataclass(frozen=True)
+class CLIConfig:
+    """Resolved CLI configuration."""
+
+    export: ExportDefaults
+    path: Path
+
+
+def _get_project_root() -> Path:
+    """Find project root by looking for pyproject.toml or .git."""
+    current = Path(__file__).resolve()
+    for parent in current.parents:
+        if (parent / "pyproject.toml").exists():
+            return parent
+        if (parent / ".git").exists():
+            return parent
+    return Path.cwd()
+
+
+def _get_data_dir() -> Path:
+    """Get the data directory."""
+    env_value = os.getenv("IMEXP_DATA_DIR")
+    if env_value:
+        return Path(env_value)
+    return _get_project_root() / "data"
+
+
+def _get_config_dir() -> Path:
+    """Get the config directory."""
+    env_value = os.getenv("IMEXP_CONFIG_DIR")
+    if env_value:
+        return Path(env_value)
+    return _get_data_dir() / "config"
+
+
+def _resolve_config_path(config_path: Path | None = None) -> Path:
+    """Resolve the config file path."""
+    if config_path is not None:
+        return config_path.expanduser().resolve()
+
+    env_path = os.getenv("IMEXP_CONFIG_FILE")
+    if env_path:
+        return Path(env_path).expanduser().resolve()
+
+    return _get_config_dir() / CONFIG_FILE
+
+
+def _ensure_config_file(path: Path) -> None:
+    """Create the config file with defaults if it doesn't exist."""
+    if path.exists():
+        return
+
+    path.parent.mkdir(parents=True, exist_ok=True)
+    path.write_text(_default_config_template(), encoding="utf-8")
+
+
+def _get_value(
+    parser: configparser.ConfigParser,
+    section: str,
+    key: str,
+) -> str | None:
+    """Read a string value from the config parser."""
+    if not parser.has_section(section):
+        return None
+
+    if not parser.has_option(section, key):
+        return None
+
+    value = parser.get(section, key).strip()
+    if not value:
+        return None
+
+    return value
+
+
+def _get_bool_value(
+    parser: configparser.ConfigParser,
+    section: str,
+    key: str,
+) -> bool | None:
+    """Read a boolean value from the config parser."""
+    if not parser.has_section(section):
+        return None
+
+    if not parser.has_option(section, key):
+        return None
+
+    raw_value = parser.get(section, key).strip()
+    if not raw_value:
+        return None
+
+    return parser.getboolean(section, key)
+
+
+def load_config(config_path: Path | None = None) -> CLIConfig:
+    """Load configuration from the config.ini file."""
+    parser = configparser.ConfigParser()
+    resolved_path = _resolve_config_path(config_path)
+    _ensure_config_file(resolved_path)
+    parser.read(resolved_path)
+
+    output_dir = _get_value(parser, "export", "output_dir") or ""
+    output_dir = os.environ.get("IMEXP_BASE_OUTPUT_DIR", output_dir) or "./data/messages/sms"
+
+    return CLIConfig(
+        export=ExportDefaults(
+            platform=_get_value(parser, "export", "platform") or "",
+            format=_get_value(parser, "export", "format") or "txt",
+            copy_method=_get_value(parser, "export", "copy_method") or "full",
+            conversation_filter=_get_value(parser, "export", "conversation_filter") or "",
+            use_caller_id=_get_bool_value(parser, "export", "use_caller_id") or False,
+            output_dir=output_dir,
+        ),
+        path=resolved_path,
+    )
+
+
+def base_output_dir(cli_config: CLIConfig | None = None) -> Path:
+    """Return the base output directory for exports."""
+    if cli_config:
+        return Path(cli_config.export.output_dir)
+    value = os.environ.get("IMEXP_BASE_OUTPUT_DIR", "./data/messages/sms")
+    return Path(value)
+
+
+def _default_config_template() -> str:
+    return """# imexp CLI configuration
+# This file is auto-generated on first run.
+# Values here serve as defaults; CLI flags always override.
+
+[export]
+# Source platform (macOS or iOS). Leave empty to prompt interactively.
+platform =
+
+# Output format for exported messages.
+# Options: txt, html
+format = txt
+
+# Attachment copy method.
+# Options: disabled, clone, basic, full
+copy_method = full
+
+# Default conversation filter (comma-separated).
+# This is the filter passed to imessage-exporter --conversation-filter.
+# Leave empty to export all conversations.
+conversation_filter =
+
+# Use caller ID instead of "Me" in exports.
+use_caller_id = true
+
+# Base output directory for exports.
+# Can also be set via IMEXP_BASE_OUTPUT_DIR environment variable.
+output_dir = ./data/messages/sms
+"""