#!/usr/bin/env python3

"""MODULE: Provides functions to call various APIs to retrieve IP/prefix information."""

from typing import Union

import ipaddress
import requests


def get_ip_information(ipv4_address: ipaddress.IPv4Address) -> dict:
    """Retrieves information about a given IPv4 address from IP-API.com."""
    api_endpoint = f"http://ip-api.com/json/{ipv4_address}"
    try:
        resp = requests.get(api_endpoint)
        resp.raise_for_status()
        ret = resp.json() if resp.json().get("status") == "success" else None
    except (requests.exceptions.JSONDecodeError, requests.exceptions.HTTPError):
        ret = None
    return ret


def get_autonomous_system_number(as_info: str) -> str:
    """Parses AS number from provided AS information."""
    as_number = as_info.split(" ")[0]
    return as_number


def get_prefix_information(autonomous_system: int) -> Union[list, None]:
    """Retrieves prefix information about a given autonomous system."""
    api_endpoint = f"https://api.hackertarget.com/aslookup/?q={str(autonomous_system)}"
    try:
        resp = requests.get(api_endpoint)
        resp.raise_for_status()
    except requests.exceptions.HTTPError:
        return None
    prefixes = resp.text.split("\n")
    prefixes.pop(0)
    return prefixes