class DownloadTV::MyEpisodes

def build_show_strings(shows)
  shows.map do |i|
    sname = i.css('td.showname').text
    ep = i.css('td.longnumber').text
    ep.insert(0, 'S')
    ep.sub!('x', 'E')
    "#{sname} #{ep}"
  end
end

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

def get_shows(last)
  return [] if last.nil?
  page = @agent.get 'https://www.myepisodes.com/ajax/service.php?mode=view_privatelist'
  shows = page.parser.css('tr.past')
  shows = filter_newer_shows(shows, last)
  build_show_strings(shows)
end

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 load_cookie
  if File.exist? @cookie_path
    @agent.cookie_jar.load @cookie_path
    return @agent if logged_in?
    puts 'The cookie is invalid/has expired.'
  else
    puts 'Cookie file not found'
  end
  login
end

def logged_in?
  page = @agent.get 'https://www.myepisodes.com/login.php'
  page.links[1].text != 'Register'
end

def login
  pass = prompt_user_data
  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 prompt_user_data
  if !@user || @user == ''
    print 'Enter your MyEpisodes username: '
    @user = STDIN.gets.chomp
  end
  print 'Enter your MyEpisodes password: '
  pass = STDIN.noecho(&:gets).chomp
  puts
  pass
end

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