require 'ws2801'
require 'ws_light/color'
require 'pp'
require 'date'
require 'json'
require 'open-uri'

require 'ws_light/animation/slide_left_animation'
require 'ws_light/animation/slide_right_animation'
require 'ws_light/animation/fade_animation'

require 'ws_light/set/color_set'
require 'ws_light/set/gradient_set'
require 'ws_light/set/random_set'
require 'ws_light/set/rainbow_set'
require 'ws_light/set/strawberry_set'
require 'ws_light/set/watermelon_set'
require 'ws_light/set/semolina_set'
require 'ws_light/set/star_set'



# Ideas
# - Fire?
# Config file

module WSLight
  # Controls the led strip
  class Strip
    attr_accessor :direction, :last_event, :state, :current_set, :debug

    LENGTH = 160
    TYPE = :double

    TIMEOUT = 12
    
    WEATHER_URL = 'http://api.openweathermap.org/data/2.5/weather?q=Hannover,de'

    FRAMES_PER_SECOND = 25

    def initialize
      WS2801.length(Strip::TYPE == :double ? Strip::LENGTH * 2 : Strip::LENGTH)
      WS2801.autowrite(true)
      self_test
      @listen_thread = Thread.new { while true do check_timer; sleep 0.5; end }
      @last_event = Time.now - 3600 # set last event to a longer time ago
      @state = :state_off
      @debug = false
      @current_set = Set::ColorSet.new
      @current_set.color = Color.new(0,0,0)
    end

    def on(direction)
      @last_event = Time.now
      puts "triggered event 'on': #{last_event.to_f} from state #{@state}" if @debug
      @state = :state_starting_up if @state == :state_shutting_down
      return if @state != :state_off

      puts 'Loading a new set...' if @debug

      @direction = direction
      @state = :state_starting_up

      case rand(100)
      when 0..3
        set = Set::RainbowSet.new
      when 4..6
        set = Set::RandomSet.new
      when 7..9
        set = Set::StrawberrySet.new
      when 10..12
        set = Set::WatermelonSet.new
      when 13..15
        set = Set::SemolinaSet.new
      else
        set = Set::GradientSet.new
        set.color_from = Color.random_from_set
        set.color_to = Color.random_from_set
      end

      set = Set::StarSet.new if night?

      puts "Set #{set.class}" if @debug

      animation = animation_for(direction).new(@current_set, set)

      animate(animation)
      @current_set = set

      @state = :state_on

      # Move show() into background, so we can accept new events on the main thread
      Thread.new { show(@current_set, animation.frames) }
    end

    def off(direction = nil)
      puts "triggered event 'off': #{Time.now.to_f} during state #{@state}" if @debug
      return if @state != :state_on

      @state = :state_shutting_down
      sleep 0.2
      @direction = direction if direction

      set = Set::ColorSet.new
      set.color = Color.by_name :black

      animation = animation_for(@direction).new(@current_set, set)

      if animate(animation)
        @state = :state_off
        @current_set = set
      else
        @state = :state_on
        Thread.new { show(@current_set, animation.frames) }
      end

      puts "finished shutting off: #{Time.now.to_f}" if @debug
    end

    def animation_for(direction)
      return Animation::FadeAnimation if night?

      if direction == :direction_left
        Animation::SlideLeftAnimation
      else
        Animation::SlideRightAnimation
      end
    end

    def animate(animation)
      current_frame = 0
      beginning_state = @state

      animation.frames.times do |i|
        WS2801.strip(animation.frame_data(current_frame = i))
        WS2801.write
        sleep (1.0/animation.frames_per_second) if animation.frames_per_second
        break if @state != beginning_state # Reverse shutting off when a new event is triggered
      end

      # This is run when the animation is reversed
      if (current_frame + 1) < animation.frames
        current_frame.times do |i|
          WS2801.strip(animation.frame_data(current_frame - i - 1))
          WS2801.write
          sleep (1.0/animation.frames_per_second) if animation.frames_per_second
        end
        false
      else
        true
      end
    end

    def show(set, start_frame = 0)
      current_state = @state
      i = start_frame
      while @state == current_state
        WS2801.strip(set.frame_data)
        WS2801.write
        sleep 1.0/FRAMES_PER_SECOND.to_f
        i += 1
      end
    end

    def night?
      time = Time.now
      time.hour > 22 || time.hour < 6
    end

    def shutdown
      WS2801.set(r: 0, g: 0, b: 0)
    end

    def self_test
      WS2801.set(r: 0, g: 0, b: 255)
      sleep 1
      WS2801.set(r: 0, g: 255, b: 0)
      sleep 1
      WS2801.set(r: 255, g: 0, b: 0)
      sleep 1
      WS2801.set(r: 0, g: 0, b: 0)
    end

    def check_timer
      WS2801.set(r: 0, g: 0, b: 0) if @state == :state_off
      off if timeout?
    end

    def timeout?
      @last_event < (Time.now - TIMEOUT)
    end
  end
end