import requests

class Api:
    def __init__(self, api_key):
        self.api_url = 'https://roicmedya.com/api/v2'
        self.api_key = api_key

    # Add order
    def order(self, data):
        post = {**data, 'key': self.api_key, 'action': 'add'}
        return self.connect(post)

    # Get order status
    def status(self, order_id):
        return self.connect({
            'key': self.api_key,
            'action': 'status',
            'order': order_id,
        })

    # Get multiple orders status
    def multi_status(self, order_ids):
        return self.connect({
            'key': self.api_key,
            'action': 'status',
            'orders': ','.join(map(str, order_ids)),
        })

    # Get services
    def services(self):
        return self.connect({
            'key': self.api_key,
            'action': 'services',
        })

    # Refill order
    def refill(self, order_id):
        return self.connect({
            'key': self.api_key,
            'order': order_id,
        })

    # Refill multiple orders
    def multi_refill(self, order_ids):
        return self.connect({
            'key': self.api_key,
            'orders': ','.join(map(str, order_ids)),
        })

    # Get balance
    def balance(self):
        return self.connect({
            'key': self.api_key,
            'action': 'balance',
        })

    # Helper method to connect to API
    def connect(self, post):
        try:
            response = requests.post(self.api_url, data=post)
            response.raise_for_status()
            return response.json()
        except requests.RequestException as e:
            print(f'Error connecting to API: {e}')
            return None

# Example usage
if __name__ == '__main__':
    api = Api('your-api-key')
    services = api.services()
    print(services)

    balance = api.balance()
    print(balance)

    order = api.order({
        'service': 1,
        'link': 'http://example.com/test',
        'quantity': 100,
        'runs': 2,
        'interval': 5,
    })
    print(order)

    status = api.status(order['order'])
    print(status)