from marvin import ai_fn

import pydantic
from pydantic import Field
from typing import List, Tuple
from rich.prompt import Prompt

from marvin_sw_text_adventure.console import console


@ai_fn
def get_star_wars_character() -> str:
    "return a brand new never before heard of star wars character"


@ai_fn
def create_backstory(name: str) -> str:
    "create a backstory for a character"


class Mission(pydantic.BaseModel):
    name: str = Field(..., description='The name of a brand new never before heard of star wars mission')
    place: str = Field(..., description='The place of the mission')
    leader: str = Field(..., description='The name of the leader of the mission')
    year: int = Field(..., description='The year of the mission')
    description: str = Field(
        ..., description='The description of the brand new never before heard of star wars mission')
    risk: int = Field(...,
                      description='The risk of the mission in health the character will loose')
    reward: int = Field(...,
                        description='The reward of the mission in imperial credits')
    image_prompt: str = Field(...,
                              description='An ai image generator prompt of the mission')


class MissionResult(pydantic.BaseModel):
    success: bool = Field(..., description='The success of the mission')
    imperial_credits_spent: int = Field(...,
                                        description='The imperial credits spent on the mission')
    imperial_credits_earned: int = Field(...,
                                         description='The imperial credits earned on the mission')
    health_lost: int = Field(..., description='The health lost on the mission')
    health_gained: int = Field(..., description='The health gained on the mission')
    story: str = Field(..., description='The story of the mission')
    image_prompt: str = Field(...,
                              description='An ai image generator prompt of the mission')
    fuel_used: int = Field(..., description='The fuel used on the mission')


class Ship(pydantic.BaseModel):
    name: str = Field(..., description='The name of the ship')
    ship_type: str = Field(..., description='The type of the ship')
    year_built: int = Field(..., description='The year built of the ship')
    capacity: int = Field(..., description='The capacity of the ship')
    fuel_type: str = Field(..., description='The fuel type of the ship')
    fuel_level: int = Field(..., description='The fuel level of the ship')
    image_prompt: str = Field(...,
                              description='An ai image generator prompt of the mission')


class StarWarsCharacter(pydantic.BaseModel):
    name: str = Field(...,
                      description='The name of a brand new never before heard of star wars character')
    imperial_credits: int = Field(...,
                                  description='The imperial credits of the character')
    health: int = Field(..., description='The health of the character')
    home_planet: str = Field(...,
                             description='The name of the planet where the character was born')
    age: int = Field(..., description='The age of the character')
    birth_year_bby: int = Field(..., description='The birth year of the character')
    backstory: str = Field(..., description='The backstory of the character')
    ship: Ship = Field(...,
                       description='The name of the ship where the character operates')
    side: str = Field(..., description='The side of the character')
    city: str = Field(..., description='The name of the city where the character lives')
    friends: List[str] = Field(...,
                               description='The names of the friends of the character')
    team: str = Field(...,
                      description='The name of the team that the character belongs to')
    enemies: List[str] = Field(...,
                               description='The names of the enemies of the character')
    mission: Mission = Field(..., description='The current mission of the character')
    role: str = Field(..., description='The role of the character')
    image_prompt: str = Field(...,
                              description='An ai image generator prompt of the mission')
    previous_missions: List[Tuple[Mission, MissionResult]] = Field(
        [], description='The previous missions of the character')


@ai_fn
def did_complete_mission(character: StarWarsCharacter, action: str) -> MissionResult:
    "check if a character completed the mission or if they failed"


@ai_fn
def get_next_mission(character: StarWarsCharacter, action: str, last_mission_success: bool) -> Mission:
    """given a character, their action and the last mission success, return the next mission"""


@ai_fn
def create_character() -> StarWarsCharacter:
    "create a new character"


def game():
    console.print('generating your character')

    prompt = Prompt()

    character = create_character()
    console.print(character)

    while character.health > 0 and character.imperial_credits > 0 and character.ship.fuel_level > 0:
        console.print('Your character status is'.center(50, '-'))
        console.print(f'health: {character.health}')
        console.print(f'imperial credits: {character.imperial_credits}')
        console.print(f'fuel level: {character.ship.fuel_level}')
        console.print()
        console.print('Your current mission is'.center(50, '-'))
        console.print(character.mission)
        action = prompt.ask('What would you like to do? ')
        result = did_complete_mission(character, action)
        character.previous_missions.append((character.mission, result))
        character.imperial_credits -= result.imperial_credits_spent
        character.imperial_credits += result.imperial_credits_earned
        character.health -= result.health_lost
        character.health += result.health_gained
        character.ship.fuel_level -= result.fuel_used
        console.print()
        console.print('Your mission has been completed')
        console.print(result.story)
        console.print(f'You earned {result.imperial_credits_earned} imperial credits')
        console.print(f'You spent {result.imperial_credits_spent} imperial credits')
        console.print(f'You gained {result.health_gained} health')
        console.print(f'You lost {result.health_lost} health')
        character.mission = get_next_mission(character, action, result.success)

    if character.health <= 0:
        console.print('You are dead')
    if character.imperial_credits <= 0:
        console.print('You lost all your imperial credits')
    if character.ship.fuel_level <= 0:
        console.print('You lost all your fuel')


if __name__ == '__main__':
    game()