from .base import Base
from time import time
from nullptr.util import *
from nullptr.models import Waypoint

class Ship(Base):  
  def define(self):
    self.cargo:dict = {}
    self.mission_state:dict = {}
    self.status:str = ''
    self.cargo_capacity:int = 0
    self.cargo_units:int = 0
    self.location = None
    self.cooldown:int = 0
    self.arrival:int = 0
    self.fuel_current:int = 0
    self.fuel_capacity:int = 0
    self.mission:str = None
    self.mission_status:str = 'init'
    self.role = None
  
  @classmethod
  def ext(self):
    return 'shp'
  
  def range(self):
    if self.fuel_capacity == 0:
      return 100000
    return self.fuel_capacity
    
  def update(self, d):
    self.seta('status', d, 'nav.status')
    getter = self.store.getter(Waypoint, create=True)
    self.seta('location', d, 'nav.waypointSymbol', interp=getter)
    self.seta('cargo_capacity', d, 'cargo.capacity')
    self.seta('cargo_units', d, 'cargo.units')
    self.seta('fuel_capacity', d, 'fuel.capacity')
    self.seta('fuel_current', d,'fuel.current')
    cargo = sg(d, 'cargo.inventory')
    if cargo is not None:
      self.load_cargo(cargo)
    self.seta('cooldown', d, 'cooldown.expiration', parse_timestamp)
    self.seta('arrival', d, 'nav.route.arrival', parse_timestamp)
    
  def tick(self):
    if self.status == 'IN_TRANSIT' and self.arrival < time():
      self.status = 'IN_ORBIT'
  
  def is_cooldown(self):
    return self.cooldown > time()
  
  def is_travelling(self):
    return self.status == 'IN_TRANSIT'
  
  def set_mission_state(self, nm, val):
    self.mission_state[nm] = val
    self.store.dirty(self)

  def get_cargo(self, typ):
    if typ not in self.cargo:
      return 0
    return self.cargo[typ]
  
  def take_cargo(self, typ, amt):
    if typ not in self.cargo:
      return
    if self.cargo[typ] <= amt:
      del self.cargo[typ]
    else:
      self.cargo[typ] -= amt
    
    self.cargo_units = sum(self.cargo.values())
    
  def load_cargo(self, cargo):
    result = {}
    total = 0
    for i in cargo:
      symbol = must_get(i, 'symbol')
      units = must_get(i, 'units')
      result[symbol] = units
      total += units
    self.cargo_units = total
    self.cargo = result
    
  def deliverable_cargo(self, contract):
    result = []
    for d in contract.deliveries:
      if self.get_cargo(d['trade_symbol']) > 0:
        result.append(d['trade_symbol'])
    return result
    
  def nondeliverable_cargo(self, contract):
    cargo = self.cargo.keys()
    deliveries = [d['trade_symbol'] for d in contract.deliveries]
    garbage = [c for c in cargo if c not in deliveries]
    return garbage
  
  def update_timers(self):
    if self.status == 'IN_TRANSIT' and self.arrival < time():
      self.status = 'IN_ORBIT'
    
    
  def f(self, detail=1):
    self.update_timers()
    arrival = int(self.arrival - time())
    cooldown = int(self.cooldown - time())
    r = self.symbol
    if detail > 1:
      if self.role is not None:
        r += f' {self.role}'
      r += ' ' + self.status
      r += f' [{self.fuel_current}/{self.fuel_capacity}]'
      r += ' ' + str(self.location)
      if self.is_travelling():
        r += f' [A: {arrival}]'
      if self.is_cooldown():
        r += f' [C: {cooldown}]'
    return r