#!/usr/bin/python

"""Tracks and exports conman status information for e.g. ledmonitor."""

# This may seem over-engineered, but conman has enough loosely-coupled moving
# parts that it is worth being able to reason formally and separately about the
# state of the system.  Otherwise it would be very easy for new conman code to
# create subtle bugs in e.g. LED behavior.

import inspect
import logging
import os


class P(object):
  """Enumerate propositions about conman status.

  Using class attributes rather than just strings will help prevent typos.
  """

  TRYING_OPEN = 'TRYING_OPEN'
  TRYING_WLAN = 'TRYING_WLAN'
  WLAN_FAILED = 'WLAN_FAILED'
  CONNECTED_TO_OPEN = 'CONNECTED_TO_OPEN'
  CONNECTED_TO_WLAN = 'CONNECTED_TO_WLAN'
  HAVE_CONFIG = 'HAVE_CONFIG'
  HAVE_WORKING_CONFIG = 'HAVE_WORKING_CONFIG'
  CAN_REACH_ACS = 'CAN_REACH_ACS'
  # Were we able to connect to the ACS last time we expected to be able to?
  COULD_REACH_ACS = 'COULD_REACH_ACS'
  CAN_REACH_INTERNET = 'CAN_REACH_INTERNET'
  PROVISIONING_FAILED = 'PROVISIONING_FAILED'
  ATTACHED_TO_WPA_SUPPLICANT = 'ATTACHED_TO_WPA_SUPPLICANT'

  WAITING_FOR_PROVISIONING = 'WAITING_FOR_PROVISIONING'
  WAITING_FOR_DHCP = 'WAITING_FOR_DHCP'
  ACS_CONNECTION_CHECK = 'ACS_CONNECTION_CHECK'
  WAITING_FOR_CWMP_WAKEUP = 'WAITING_FOR_CWMP_WAKEUP'
  WAITING_FOR_ACS_SESSION = 'WAITING_FOR_ACS_SESSION'
  PROVISIONING_COMPLETED = 'PROVISIONING_COMPLETED'


# Format:  { proposition: (implications, counter-implications), ... }
# If you want to add a new proposition to the Status class, just edit this dict.
IMPLICATIONS = {
    P.TRYING_OPEN: (
        (),
        (P.CONNECTED_TO_OPEN, P.TRYING_WLAN, P.CONNECTED_TO_WLAN,
         P.WAITING_FOR_PROVISIONING)
    ),
    P.TRYING_WLAN: (
        (),
        (P.TRYING_OPEN, P.CONNECTED_TO_OPEN, P.CONNECTED_TO_WLAN,
         P.WAITING_FOR_PROVISIONING)
    ),
    P.WLAN_FAILED: (
        (),
        (P.TRYING_WLAN, P.CONNECTED_TO_WLAN)
    ),
    P.CONNECTED_TO_OPEN: (
        (),
        (P.CONNECTED_TO_WLAN, P.TRYING_OPEN, P.TRYING_WLAN)
    ),
    P.CONNECTED_TO_WLAN: (
        (P.HAVE_WORKING_CONFIG,),
        (P.CONNECTED_TO_OPEN, P.TRYING_OPEN, P.TRYING_WLAN,
         P.WAITING_FOR_PROVISIONING)
    ),
    P.CAN_REACH_ACS: (
        (P.COULD_REACH_ACS,),
        (P.TRYING_OPEN, P.TRYING_WLAN)
    ),
    P.COULD_REACH_ACS: (
        (),
        (),
    ),
    P.PROVISIONING_FAILED: (
        (),
        (P.WAITING_FOR_PROVISIONING, P.PROVISIONING_COMPLETED,),
    ),
    P.HAVE_WORKING_CONFIG: (
        (P.HAVE_CONFIG,),
        (),
    ),
    P.ATTACHED_TO_WPA_SUPPLICANT: (
        (),
        (),
    ),
    P.WAITING_FOR_PROVISIONING: (
        (P.CONNECTED_TO_OPEN,),
        (),
    ),
    P.WAITING_FOR_DHCP: (
        (P.WAITING_FOR_PROVISIONING,),
        (P.WAITING_FOR_CWMP_WAKEUP, P.WAITING_FOR_ACS_SESSION),
    ),
    P.ACS_CONNECTION_CHECK: (
        (P.WAITING_FOR_PROVISIONING,),
        (P.WAITING_FOR_DHCP, P.WAITING_FOR_CWMP_WAKEUP,
         P.WAITING_FOR_ACS_SESSION),
    ),
    P.WAITING_FOR_CWMP_WAKEUP: (
        (P.WAITING_FOR_PROVISIONING,),
        (P.WAITING_FOR_DHCP, P.ACS_CONNECTION_CHECK, P.WAITING_FOR_ACS_SESSION),
    ),
    P.WAITING_FOR_ACS_SESSION: (
        (P.WAITING_FOR_PROVISIONING,),
        (P.WAITING_FOR_DHCP, P.ACS_CONNECTION_CHECK, P.WAITING_FOR_CWMP_WAKEUP),
    ),
    P.PROVISIONING_COMPLETED: (
        (),
        (P.WAITING_FOR_PROVISIONING, P.PROVISIONING_FAILED,),
    ),
}


class ExportedValue(object):

  def __init__(self, name, export_path):
    self._name = name
    self._export_path = export_path

  def export(self):
    filepath = os.path.join(self._export_path, self._name)
    if self.value():
      if not os.path.exists(filepath):
        open(filepath, 'w')
    else:
      if os.path.exists(filepath):
        os.unlink(filepath)


class Proposition(ExportedValue):
  """Represents a proposition.

  May imply truth or falsity of other propositions.
  """

  def __init__(self, *args, **kwargs):
    super(Proposition, self).__init__(*args, **kwargs)
    self._value = None
    self._implications = set()
    self._counter_implications = set()
    self._impliers = set()
    self._counter_impliers = set()
    self.parents = set()

  def implies(self, implication):
    self._counter_implications.discard(implication)
    self._implications.add(implication)
    # pylint: disable=protected-access
    implication._implied_by(self)

  def implies_not(self, counter_implication):
    self._implications.discard(counter_implication)
    self._counter_implications.add(counter_implication)
    # pylint: disable=protected-access
    counter_implication._counter_implied_by(self)

  def _implied_by(self, implier):
    self._counter_impliers.discard(implier)
    self._impliers.add(implier)

  def _counter_implied_by(self, counter_implier):
    self._impliers.discard(counter_implier)
    self._counter_impliers.add(counter_implier)

  def set(self, value):
    """Set this Proposition's value.

    If the value changed, update any dependent values/files.

    Args:
      value:  The new value.
    """
    if value == self._value:
      return

    self._value = value
    self.export()
    for parent in self.parents:
      parent.export()
    logging.debug('%s is now %s', self._name, self._value)

    if value:
      for implication in self._implications:
        implication.set(True)
      for counter_implication in self._counter_implications:
        counter_implication.set(False)
      # Contrapositive:  (A -> ~B) -> (B -> ~A)
      for counter_implier in self._counter_impliers:
        counter_implier.set(False)
    # Contrapositive:  (A -> B) -> (~B -> ~A)
    else:
      for implier in self._impliers:
        implier.set(False)

  def value(self):
    return self._value


class Disjunction(ExportedValue):

  def __init__(self, name, export_path, disjuncts):
    super(Disjunction, self).__init__(name, export_path)
    self.disjuncts = disjuncts
    for disjunct in self.disjuncts:
      disjunct.parents.add(self)

  def value(self):
    return any(d.value() for d in self.disjuncts)


class Status(object):
  """Provides a convenient API for conman to describe system status."""

  def __init__(self, export_path):
    if not os.path.isdir(export_path):
      os.makedirs(export_path)

    self._export_path = export_path

    self._set_up_propositions()

  def _set_up_propositions(self):
    self._propositions = {
        p: Proposition(p, self._export_path)
        for p in dict(inspect.getmembers(P)) if not p.startswith('_')
    }

    for p, (implications, counter_implications) in IMPLICATIONS.iteritems():
      for implication in implications:
        self._propositions[p].implies(self._propositions[implication])
      for counter_implication in counter_implications:
        self._propositions[p].implies_not(
            self._propositions[counter_implication])

  def proposition(self, p):
    return self._propositions[p]

  def __setattr__(self, attr, value):
    """Allow setting of propositions with attributes.

    If _propositions contains an attribute 'FOO', then `Status().foo = True`
    will set that Proposition to True.  This means that this class doesn't have
    to be changed when IMPLICATIONS is updated.

    Args:
      attr:  The attribute name.
      value:  The attribute value.
    """
    if hasattr(self, '_propositions') and not hasattr(self, attr):
      if attr.islower():
        if attr.upper() in self._propositions:
          self._propositions[attr.upper()].set(value)
          return

    super(Status, self).__setattr__(attr, value)


class CompositeStatus(Status):

  def __init__(self, export_path, children):
    self._children = children
    super(CompositeStatus, self).__init__(export_path)

  def _set_up_propositions(self):
    self._propositions = {
        p: Disjunction(p, self._export_path,
                       [c.proposition(p) for c in self._children])
        for p in dict(inspect.getmembers(P)) if not p.startswith('_')
    }