import argparse
import json
from baserow_api import BaserowAPI
from data_processor import DataProcessor
from matrix_builder import MatrixBuilder
import view

def handle_output(quiet, data):
    if not quiet:
        view.print_json_output(data)

def main():
    args = parse_arguments()
    api = BaserowAPI(args.base_url, args.api_key, args.verbose)
    data_processor = DataProcessor(api, args.verbose)
    matrix_builder = MatrixBuilder(data_processor, args.verbose)
    
    if args.table_ids:
        tables_data = fetch_table_data(data_processor, args.table_ids)
        
        if "linked_fields" in args.output:
            linked_fields_data = data_processor.get_link_fields_for_all_tables(tables_data)
            handle_output(args.quiet, linked_fields_data)
            
        if "tables" in args.output:
            handle_output(args.quiet, tables_data)
        
        if "matrix" in args.output:
            matrix_data = matrix_builder.build_multitable_matrix(tables_data)
            handle_output(args.quiet, matrix_data)
    
    if args.database_id:
        all_data = data_processor.get_all_data_from_database(args.database_id)
        handle_output(args.quiet, all_data)
        
def parse_arguments():
    parser = argparse.ArgumentParser(description="Fetch all data from a Baserow database.")
    parser.add_argument("base_url", help="Base URL of your Baserow instance, e.g., https://YOUR_BASEROW_INSTANCE_URL/api/")
    parser.add_argument("api_key", help="Your Baserow API key.")
    parser.add_argument("--database_id", help="ID of the Baserow database you want to fetch data from.", default=None)
    parser.add_argument("--table_ids", help="IDs of the Baserow tables you want to fetch data from, separated by commas.", default=None)
    parser.add_argument("-v", "--verbose", action="store_true", help="Enable verbose mode for debugging.")
    parser.add_argument("--output", choices=["linked_fields", "tables", "matrix"], default=[], nargs='+', help="Specify the type(s) of output: linked_fields, matrix, content or both")
    parser.add_argument("--quiet", action="store_true", help="Suppress output of json")
    return parser.parse_args()

def fetch_table_data(data_processor, table_ids_str):
    table_ids = table_ids_str.split(',')
    return data_processor.get_tables(table_ids)

if __name__ == "__main__":
    main()