class TestLab

  # Network Error Class
  class NetworkError < TestLabError; end

  # Network Class
  #
  # @author Zachary Patten <zachary@jovelabs.net>
  class Network < ZTK::DSL::Base
    STATUS_KEYS   = %w(node_id id state interface).map(&:to_sym)

    belongs_to  :node,        :class_name => 'TestLab::Node'

    attribute   :bridge

    attribute   :cidr
    attribute   :config

    def initialize(*args)
      super(*args)

      @ui = TestLab.ui
    end

    # Network status
    def status
      interface = "#{bridge}:#{cidr}"
      {
        :id => self.id,
        :node_id => self.node.id,
        :state => self.state,
        :interface => interface
      }
    end

################################################################################

    # Create the network
    def create
      @ui.logger.debug { "Network Create: #{self.id} " }

      self.node.ssh.exec(%(sudo brctl addbr #{self.bridge}), :silence => true, :ignore_exit_status => true)
      self.node.ssh.exec(%(sudo ifconfig #{self.bridge} #{self.cidr} down), :silence => true, :ignore_exit_status => true)
    end

    # Destroy the network
    def destroy
      @ui.logger.debug { "Network Destroy: #{self.id} " }

      self.node.ssh.exec(%(sudo brctl delbr #{self.bridge}), :silence => true, :ignore_exit_status => true)
    end

    # Start the network
    def up
      @ui.logger.debug { "Network Up: #{self.id} " }

      self.node.ssh.exec(%(sudo ifconfig #{self.bridge} up), :silence => true, :ignore_exit_status => true)
    end

    # Stop the network
    def down
      @ui.logger.debug { "Network Down: #{self.id} " }

      self.node.ssh.exec(%(sudo ifconfig #{self.bridge} down), :silence => true, :ignore_exit_status => true)
    end

################################################################################

    # Reload the network
    def reload
      @ui.logger.debug { "Network Reload: #{self.id} " }

      self.down
      self.up
    end

################################################################################

    # State of the network
    def state
      output = self.node.ssh.exec(%(sudo ifconfig #{self.bridge} | grep 'MTU'), :silence => true, :ignore_exit_status => true).output.strip
      if ((output =~ /UP/) && (output =~ /RUNNING/))
        :running
      else
        :stopped
      end
    end

################################################################################

    # Network Callback: after_create
    def after_create
      @ui.logger.debug { "Network Callback: After Create: #{self.id} " }
    end

    # Network Callback: after_up
    def after_up
      @ui.logger.debug { "Network Callback: After Up: #{self.id} " }

      self.create
      self.up
    end

    # Network Callback: before_down
    def before_down
      @ui.logger.debug { "Network Callback: Before Down: #{self.id} " }

      self.down
      self.destroy
    end

    # Network Callback: before_destroy
    def before_destroy
      @ui.logger.debug { "Network Callback: Before Destroy: #{self.id} " }
    end

################################################################################

    # Method missing handler
    def method_missing(method_name, *method_args)
      @ui.logger.debug { "NETWORK METHOD MISSING: #{method_name.inspect}(#{method_args.inspect})" }
      super(method_name, *method_args)
    end

  end

end