class Kuport

# File lib/kuport.rb, line 171
def self.filter(key, pattern)
  JSON.load(STDIN.read).select{|m| m[key].to_s.match?(pattern)}
end

# File lib/kuport.rb, line 54
def self.module_url(*parts)
  URI.join(@@base_url, @@base_module, *parts)
end

# File lib/kuport.rb, line 36
def initialize(**opts)
  agent.html_parser = self.class
  cookies_load

  # If proxy is blank, don't use proxy
  proxy = opts.key?(:proxy) ? opts[:proxy] : Kuport.get_proxy_env_var

  if proxy.blank?
    Kuport.clear_proxy_env_var
  else
    agent.set_proxy(*Kuport.parse_proxy_str(proxy))
  end
end

# File lib/kuport.rb, line 176
def self.parse(text, url = nil, encoding = nil, options = Nokogiri::XML::ParseOptions::DEFAULT_HTML, &block)
  Nokogiri::HTML::Document.parse(text.toutf8, url, 'UTF-8', options, &block)
end

# File lib/kuport.rb, line 50
def agent
  @agent ||= Mechanize.new
end

# File lib/kuport.rb, line 74
def cookies_clear
  agent.cookie_jar.clear!
  FileUtils.rm(@@cookies_file) if File.exist?(@@cookies_file)
end

# File lib/kuport.rb, line 70
def cookies_load
  File.open(@@cookies_file, 'r'){|f| agent.cookie_jar.load(f) } rescue nil
end

# File lib/kuport.rb, line 66
def cookies_save
  File.open(@@cookies_file, 'w'){|f| agent.cookie_jar.save(f, {session: true})}
end

# File lib/kuport.rb, line 134
def download_file(file_path, url)
  File.write(file_path, agent.get(url).content)
rescue
  DownloadError.new(file_path)
end

# File lib/kuport.rb, line 141
def download_with_json(json_str)
  h = JSON.parse(json_str, {symbolize_names: true})
  find_links(h).each do |h|
    download_file(h[:name], h[:path])
    puts h[:name]
  end

rescue JSON::ParserError
  raise DownloadError.new("JSON parse error '#{json}'")
end

# File lib/kuport.rb, line 153
def find_links(j)
  links = []

  if j.is_a?(Array)
    j.each{|v| links.push(*find_links(v))}
  elsif j.is_a?(Hash)
    link = j.take_with_keys(:name, :path)
    links.push(link) if link

    j.each do |_,v|
      links.push(*find_links(v)) if v.is_a?(Array) || v.is_a?(Hash)
    end
  end

  return links
end

# File lib/kuport.rb, line 58
def get_module(symb)
  agent.get(self.class.module_url(@@modules[symb]))
end

# File lib/kuport.rb, line 79
def loggedin?
  nil != get_module(:login).form.has_value?('Logout')
end

# File lib/kuport.rb, line 101
def login(id)
  return true if login_cookie
  Kuport.quit('Please student id', 4) if id.blank?

  3.times do
    return true if login_passwd(id, Kuport.input_passwd)
    warn 'Login failed'
  end

  warn 'Login error'
  return false
end

# File lib/kuport.rb, line 83
def login_cookie
  return false unless cookies_load
  return true if loggedin?
  agent.cookie_jar.clear!
  false
end

# File lib/kuport.rb, line 90
def login_passwd(id, passwd)
  get_module(:login).form_with(name: 'login_form') do |f|
    f.login = id
    f.passwd = passwd
  end.submit

  return false unless loggedin?
  cookies_save
  true
end

# File lib/kuport.rb, line 130
def materials
  @materials ||= Materials.new(get_module(:materials))
end

# File lib/kuport.rb, line 62
def menu_links
  @menu_links ||= get_module(:menu).links.reject{|l| l.text.empty? }
end

# File lib/kuport.rb, line 114
def messages
  @messages ||= Message.parse_page(agent, get_module(:messages))
end

# File lib/kuport.rb, line 122
def messages_backno
  raise NotImplementedError.new('Kuport#messages_backno') # TODO
end

# File lib/kuport.rb, line 118
def messages_read
  @messages_read ||= Message.parse_page(agent, get_module(:messages_read))
end

# File lib/kuport.rb, line 126
def timetable
  @timetable ||= Timetable.new(get_module(:timetable))
end