import json
import os

def fix_swagger_responses(swagger_data):
    """
    Modifies Swagger 2.0 responses to include 'application/json' content type
    in an OpenAPI 3.0.x style, by wrapping the existing schema.
    It also removes the '$schema' key from the response schemas.

    Args:
        swagger_data (dict): The Swagger data as a Python dictionary.

    Returns:
        dict: The modified Swagger data.
    """
    if not isinstance(swagger_data, dict):
        return swagger_data

    paths = swagger_data.get("paths", {})
    for path_item_key, path_item_value in paths.items(): # Iterate with keys for better debugging if needed
        if not isinstance(path_item_value, dict):
            continue
        for operation_method, operation_details in path_item_value.items():
            # Valid HTTP methods according to OpenAPI spec.
            # We are interested in actual operations like get, post, put, delete, etc.
            if operation_method.lower() not in ["get", "post", "put", "delete", "patch", "options", "head", "trace"]:
                continue

            if not isinstance(operation_details, dict):
                continue
                
            responses = operation_details.get("responses")
            if not isinstance(responses, dict):
                continue

            for status_code, response_definition in responses.items():
                if not isinstance(response_definition, dict):
                    continue
                
                if "schema" in response_definition:
                    # Preserve the original schema and remove it from its current location
                    original_schema = response_definition.pop("schema")
                    
                    # Remove "$schema" key from the original schema if it exists and schema is a dict
                    if isinstance(original_schema, dict) and "$schema" in original_schema:
                        del original_schema["$schema"]
                    
                    # Create the new content structure
                    response_definition["content"] = {
                        "application/json": {
                            "schema": original_schema
                        }
                    }
    return swagger_data

def process_swagger_file(file_path):
    """
    Reads a Swagger JSON file, applies fixes, and prints the modified JSON.

    Args:
        file_path (str): The path to the Swagger JSON file.
    """
    try:
        # Expand user's home directory if ~ is used
        expanded_path = os.path.expanduser(file_path)

        with open(expanded_path, 'r', encoding='utf-8') as f:
            swagger_data = json.load(f)
        
        print(f"Successfully read file: {expanded_path}")
        
        modified_swagger_data = fix_swagger_responses(swagger_data)
        
        # Print the modified JSON data
        # indent=2 makes the output pretty-printed and readable
        # ensure_ascii=False allows non-ASCII characters (like Chinese) to be printed as is
        print("\nModified Swagger JSON:")
        print(json.dumps(modified_swagger_data, indent=2, ensure_ascii=False))

        # If you want to save it back to a new file or overwrite:
        output_file_path = os.path.join(os.path.dirname(expanded_path), "井筒API示例swagger_fixed.json")
        with open(output_file_path, 'w', encoding='utf-8') as f:
            json.dump(modified_swagger_data, f, indent=2, ensure_ascii=False)
        print(f"\nModified Swagger JSON saved to: {output_file_path}")

    except FileNotFoundError:
        print(f"Error: File not found at {file_path} (Expanded: {expanded_path})")
    except json.JSONDecodeError:
        print(f"Error: Could not decode JSON from file {file_path}. Make sure it's a valid JSON.")
    except Exception as e:
        print(f"An unexpected error occurred: {e}")

# --- Main execution ---
if __name__ == "__main__":
    # 使用您提供的文件路径
    # 注意：在Python字符串中，反斜杠 '\' 是转义字符。
    # 在Windows路径中，您可能需要使用双反斜杠 '\\' 或者原始字符串 r"..."
    # 在macOS或Linux上，普通正斜杠 '/' 就可以。
    # 您提供的路径是macOS/Linux风格，所以可以直接使用。
    swagger_file_path = "/Users/zpc01/workspace/zzlh/compliance/assets/doc/井筒API示例swagger.json"
    
    process_swagger_file(swagger_file_path)