#!/usr/bin/env python3
"""
Script for managing SFOSR concepts (export/import).
"""

import argparse
import sys
import os

# Ensure the script can find the sfosr_core module
script_dir = os.path.dirname(os.path.abspath(__file__))
project_root = script_dir # Assumes script is in the root
sys.path.insert(0, project_root)

try:
    from sfosr_core.sfosr_database import SFOSRDatabase
except ImportError as e:
    print(f"Error: Could not import SFOSRDatabase. Make sure sfosr_core is accessible.")
    print(f"PYTHONPATH: {sys.path}")
    sys.exit(1)

DEFAULT_DB_PATH = "sfosr.db"
DEFAULT_JSON_PATH_EXPORT = "concepts_export.json" # Default for export
DEFAULT_JSON_PATH_IMPORT = "concepts_to_import.json" # Default for import

def main():
    parser = argparse.ArgumentParser(description="Manage SFOSR concepts.")
    subparsers = parser.add_subparsers(dest='action', help='Action to perform', required=True)

    # Export command
    parser_export = subparsers.add_parser('export', help='Export concepts to JSON')
    parser_export.add_argument(
        '--db-path', 
        default=DEFAULT_DB_PATH, 
        help=f"Path to the SFOSR database file (default: {DEFAULT_DB_PATH})"
    )
    parser_export.add_argument(
        '--json-path', 
        default=DEFAULT_JSON_PATH_EXPORT, 
        help=f"Path to the output JSON file (default: {DEFAULT_JSON_PATH_EXPORT})"
    )

    # Import command
    parser_import = subparsers.add_parser('import', help='Import concepts from JSON (adds only new concepts)')
    parser_import.add_argument(
        '--db-path', 
        default=DEFAULT_DB_PATH, 
        help=f"Path to the SFOSR database file (default: {DEFAULT_DB_PATH})"
    )
    parser_import.add_argument(
        '--json-path', 
        default=DEFAULT_JSON_PATH_IMPORT, 
        help=f"Path to the input JSON file (default: {DEFAULT_JSON_PATH_IMPORT})"
    )

    args = parser.parse_args()

    if not os.path.exists(args.db_path):
        print(f"Error: Database file not found at {args.db_path}")
        sys.exit(1)
        
    db = SFOSRDatabase(db_path=args.db_path)

    if args.action == 'export':
        print(f"Exporting concepts from {args.db_path} to {args.json_path}...")
        db.export_concepts_to_json(args.json_path)
    
    elif args.action == 'import':
        if not os.path.exists(args.json_path):
             print(f"Error: Import JSON file not found at {args.json_path}")
             sys.exit(1)
             
        print(f"Importing concepts from {args.json_path} to {args.db_path}...")
        try:
            added, skipped = db.import_concepts_from_json(args.json_path)
            # Optional: Add more feedback based on counts
        except AttributeError:
            print(f"Error: import_concepts_from_json method not found in SFOSRDatabase class in {db.__class__.__module__}.")
            sys.exit(1)
        except Exception as e:
            print(f"An unexpected error occurred during import: {e}")
            import traceback
            traceback.print_exc() # Print stack trace for unexpected errors
            sys.exit(1)

if __name__ == "__main__":
    main()