import random
import string
import requests

class Requests():
    def __init__(self, baseUrl):
        self.baseUrl = baseUrl
    
    def post(self, path, json = {}):
        return requests.post(self.baseUrl + path, json=json)

class Auth():
    token: string

    def __init__(self, token):
        self.token = token

class User():
    id: string
    email: string
    name: string
    password: string

    def __init__(self, email, password, name, id="", token = ""):
        self.email = email
        self.password = password
        self.name = name
        self.token = token


class BackendApi():
    def __init__(self, httpClient):
        self.httpClient = httpClient

    def parse_response(self, response):
        if response.status != 200:
            raise AssertionError('something wrong')
        
        json = response.json()
        if json['status'] == 'success':
            if 'result' in json:
                return json['result']
            return None
        
        error = json['error']
        raise AssertionError(error['id'], error['message'])

    def user_create(self, user: User | None) -> User:
        if user == None:
            email = ''.join(random.choices(string.ascii_lowercase + string.digits, k=10)) + '@test.test'
            name = ''.join(random.choices(string.ascii_letters, k=10))
            password = 'Abcdef1!!1'
            user = User(email, password, name)
    
        res = self.parse_response(
            self.httpClient.post(
                "/v1/user/create", json={
                    "email": user.email,
                    "password": user.password,
                    "name": user.name,
                }
            )
        )

        return User(res['email'], res['password'], res['name'], res['id'])
        
    def user_login(self, user: User) -> Auth:        
        res = self.parse_response(
            self.httpClient.post(
                "/v1/user/login", json={
                    "email": user.email+"a",
                    "password": user.password,
                },
            )
        )
        
        return Auth(res['status'])        

    def dummy_get(self, auth: Auth):
        headers = {"X-Auth": auth.token}
        response = self.httpClient.get("/v1/dummy", headers=headers)
        if response.status_code != 200:
            raise AssertionError('something wrong')
        
    def health_get(self):
        response = self.httpClient.get("/health")
        if response.status_code != 200:
            raise AssertionError('something wrong')