module DownloadTV
  ##
  # API wrapper for MyEpisodes
  class MyEpisodes
    def initialize(user, save_cookie)
      @agent = Mechanize.new
      @agent.user_agent = DownloadTV::USER_AGENT
      @user = user
      @save_cookie = save_cookie
      @cookie_path = File.join(ENV['HOME'], '.config', 'download_tv', 'cookie')
    end

    def login
      if !@user || @user == ''
        print 'Enter your MyEpisodes username: '
        @user = STDIN.gets.chomp
      end

      print 'Enter your MyEpisodes password: '
      pass = STDIN.noecho(&:gets).chomp
      puts

      page = @agent.get 'https://www.myepisodes.com/login.php'

      login_form = page.forms[1]
      login_form.username = @user
      login_form.password = pass

      page = @agent.submit(login_form, login_form.buttons.first)

      raise InvalidLoginError if page.filename == 'login.php'

      save_cookie if @save_cookie

      @agent
    end

    def load_cookie
      if File.exist? @cookie_path
        @agent.cookie_jar.load @cookie_path
        page = @agent.get 'https://www.myepisodes.com/login.php'
        if page.links[1].text == 'Register'
          puts 'The cookie is invalid/has expired.'
          login
        end
        @agent
      else
        puts 'Cookie file not found'
        login
      end
    end

    def save_cookie
      @agent.cookie_jar.save(@cookie_path, session: true)
      @agent
    end

    def get_shows(last)
      page = @agent.get 'https://www.myepisodes.com/ajax/service.php?mode=view_privatelist'
      shows = page.parser.css('tr.past')

      s = shows.select do |i|
        airdate = i.css('td.date')[0].text
        Date.parse(airdate) >= last
      end

      s.map do |i|
        name = i.css('td.showname').text
        ep = i.css('td.longnumber').text

        ep.insert(0, 'S')
        ep.sub!('x', 'E')

        "#{name} #{ep}"
      end
    end
  end

  class InvalidLoginError < StandardError; end
end