etvnet_seek 0.3.0 → 0.4.0

data/lib/etvnet_seek/core/login_page.rb

@@ -0,0 +1,46 @@
+require 'stringio'
+
+class LoginPage < ServiceCall
+  LOGIN_URL = "#{Page::BASE_URL}/cgi-bin/video/login.fcgi"
+
+  def initialize
+    super(LOGIN_URL)
+  end
+
+  def login username, password
+    headers = { "Content-Type" => "application/x-www-form-urlencoded" }
+
+    response = post({ 'action' => 'login', 'username' => username, 'pwd'=> password }, headers)
+
+    cookie = response.response['set-cookie']
+
+    unless cookie.nil?
+      cookie = cleanup_cookie(cookie)
+    end
+
+    cookie
+  end
+
+  private
+
+  def cleanup_cookie cookie
+    auth, expires = CookieHelper.get_auth_and_expires(cookie)
+    username = CookieHelper.get_username(cookie)
+    path = "/"
+    domain = ".etvnet.ca"
+
+    cookies_text = <<-TEXT
+      auth=#{auth}; expires=#{expires}; path=#{path}; domain=#{domain}
+      username=#{username}; expires=#{expires}; path=#{path}; domain=#{domain}
+    TEXT
+
+    new_cookie = ""
+
+    StringIO.new(cookies_text).each_line do |line|
+      new_cookie = new_cookie + line.strip + "; "
+    end
+
+    new_cookie
+  end
+
+end

data/lib/etvnet_seek/core/main_page.rb

@@ -0,0 +1,20 @@
+class MainPage < BasePage
+
+  def items
+    list = []
+
+    document.css("#tblCategories a").each do |item|
+      text = item.css("img").at(0).attributes['alt'].value
+      href = item['href']
+
+       unless href =~ /forum.etvnet.ca/ or href =~ /action=browse_persons/ or href =~ /valentines2010/
+        list << MediaItem.new(text, href)
+       end
+    end
+
+    list.delete_at(0)
+
+    list
+  end
+
+end

data/lib/etvnet_seek/core/media_info.rb

@@ -0,0 +1,31 @@
+require 'json'
+
+class MediaInfo
+  attr_reader :name, :link, :description, :description_production, :rtsp_link, :speech_lang,
+              :file_length, :channel, :datetime, :popularity, :duration, :rating
+
+  def initialize params
+    @link = params["REDIRECT_URL"]
+    @session_expired = (params["error_session_expire"] == 1)
+    @description_production = params["description_production"]
+    @rtsp_link = params["REDIRECT_URL_RTSP"]
+    @speech_lang = params["speech_lang"]
+    @name = params["name"]
+    @description = params["description"]
+    @file_length = params["file_length"]
+    @channel = params["channel"]
+    @datetime = params["datetime"]
+    @popularity = params["popularity_24h"]
+    @duration = params["duration"]
+    @rating = params["rating"]
+  end
+
+  def resolved?
+    not @link.nil? and not @link.strip.size == 0
+  end
+
+  def session_expired?
+    @session_expired
+  end
+end
+

data/lib/etvnet_seek/core/media_item.rb

@@ -0,0 +1,45 @@
+class MediaItem
+  attr_reader :text, :link, :additional_info
+  attr_reader :underscore_name, :media_file
+
+  def initialize(text, link, additional_info = nil)
+    @text = text
+    @link = link
+    @additional_info = additional_info
+
+    @underscore_name = extract_underscore_name
+    @media_file = extract_media_file
+  end
+
+  def folder?
+    false
+  end
+
+  def ==(object)
+    object.text == text and object.link == link
+  end
+
+  def to_s
+    text
+  end
+
+  private
+
+  def extract_underscore_name
+    return nil if link.nil?
+
+    result = link.match(/(\w*)\/(\w*)\/(\w*)\/([\w|-]*)/)
+
+    return nil if result.nil?
+
+    (result.size > 3) ? result[4] : ""
+  end
+
+  def extract_media_file
+    return nil if link.nil?
+
+    result = link.match(/(\w*)\/(\w*)\/(\w*)\/([\w|-]*)/)
+
+    (not result.nil? and result.size > 2) ? result[3] : ""
+  end
+end

data/lib/etvnet_seek/core/media_page.rb

@@ -0,0 +1,43 @@
+class MediaPage < BasePage
+  def items
+    list = []
+
+    document.css("b a.media_file").each do |item|
+      link = item.attributes['href'].value
+      new_link = list.select {|l| l.link == link}.empty?
+
+      if new_link
+        text = item.content.strip
+        additional_info = additional_info(item, 1)
+
+        text += additional_info unless additional_info.nil?
+
+        showtime = item.parent.parent.parent.css('td[1]').text.strip
+        year = item.parent.parent.next.next.content.strip
+        duration = ""
+
+        if link =~ /action=browse_container/
+          folder = true
+          link = link[Page::BASE_URL.size..link.size]
+        else
+          folder = false
+          duration = item.parent.parent.next.next.next.next.content.strip unless
+            item.parent.parent.next.next.next.next.nil?
+        end
+
+        record = BrowseMediaItem.new(text, link)
+        record.folder = folder
+        record.showtime = showtime
+        record.year = year
+        record.duration = duration
+
+        list << record
+      end
+    end
+
+    list
+  end  
+
+end
+
+

data/lib/etvnet_seek/core/page.rb

@@ -0,0 +1,19 @@
+require 'nokogiri'
+
+class Page < ServiceCall
+  BASE_URL = "http://www.etvnet.ca"
+
+  attr_reader :document
+
+  def initialize(url = BASE_URL)
+    super(url.index(BASE_URL).nil? ? "#{BASE_URL}/#{url}" : url)
+
+    @document = get_document
+  end
+
+  protected
+
+  def get_document
+    Nokogiri::HTML(get)
+  end
+end

data/lib/etvnet_seek/core/page_factory.rb

@@ -0,0 +1,52 @@
+
+require 'etvnet_seek/core/media_item'
+require 'etvnet_seek/core/media_info'
+require 'etvnet_seek/core/browse_media_item'
+require 'etvnet_seek/core/channel_media_item'
+require 'etvnet_seek/core/group_media_item'
+
+require 'etvnet_seek/core/service_call'
+require 'etvnet_seek/core/page'
+require 'etvnet_seek/core/base_page'
+require 'etvnet_seek/core/media_page'
+require 'etvnet_seek/core/main_page'
+require 'etvnet_seek/core/search_page'
+require 'etvnet_seek/core/freetv_page'
+require 'etvnet_seek/core/channels_page'
+require 'etvnet_seek/core/group_page'
+require 'etvnet_seek/core/announces_page'
+require 'etvnet_seek/core/access_page'
+require 'etvnet_seek/core/login_page'
+
+class PageFactory
+  def self.create mode, params = []
+    url = (mode == 'search') ? nil : (params.class == String ? params : params[0])
+
+    case mode
+      when 'search' then
+        SearchPage.new *params
+      when 'main' then
+        MainPage.new
+      when 'channels' then
+        ChannelsPage.new
+      when 'best_ten' then
+        BestTenPage.new
+      when 'popular' then
+        PopularPage.new
+      when 'we_recommend' then
+        WeRecommendPage.new
+      when 'announces' then
+        AnnouncesPage.new
+      when 'freetv' then
+        FreetvPage.new
+      when 'media' then
+        MediaPage.new url
+      when 'access' then
+        AccessPage.new
+      when 'login' then
+        LoginPage.new
+      else
+        nil
+    end
+  end
+end

data/lib/etvnet_seek/core/search_page.rb

@@ -0,0 +1,7 @@
+class SearchPage < MediaPage
+  SEARCH_URL = BASE_URL + "/cgi-bin/video/eitv_browse.fcgi?action=search"
+
+  def initialize(params)
+    super("#{SEARCH_URL}&keywords=#{CGI.escape(*params)}&order_direction=-")
+  end
+end

data/lib/etvnet_seek/core/service_call.rb

@@ -0,0 +1,32 @@
+require 'open-uri'
+
+class ServiceCall
+  attr_reader :url
+
+  def initialize(url)
+    @url = url
+  end
+
+  def get
+    open(url)
+  end
+
+  def post params, headers
+    request = Net::HTTP::Post.new(url, headers)
+
+    request.set_form_data(params)
+
+    request(request)
+  end
+
+  protected
+
+  def request request
+    uri = URI.parse(url)
+
+    connection = Net::HTTP.new(uri.host, uri.port)
+
+    connection.request(request)
+  end
+  
+end

data/lib/etvnet_seek/link_info.rb

@@ -0,0 +1,33 @@
+class LinkInfo
+  attr_reader :media_item, :media_info
+
+  def initialize(media_item, media_info)
+    @media_item = media_item
+    @media_info = media_info
+  end
+
+  def resolved?
+    not @media_info.link.nil? and not @media_info.link.strip.size == 0
+  end
+
+  def session_expired?
+    @media_info.session_expired
+  end
+
+  def text
+    media_item.text
+  end
+
+  def name
+    media_item.underscore_name
+  end
+
+  def media_file
+    media_item.media_file
+  end
+
+  def link
+    media_info.link
+  end
+
+end

data/lib/etvnet_seek/main.rb

@@ -0,0 +1,177 @@
+require "highline/import"
+require 'optparse'
+require 'date'
+
+require 'etvnet_seek/core/page_factory'
+
+require 'etvnet_seek/cookie_helper'
+require 'etvnet_seek/user_selection'
+require 'etvnet_seek/link_info'
+require 'etvnet_seek/commander'
+require 'runglish'
+
+class Main
+  COOKIE_FILE_NAME = ENV['HOME'] + "/.etvnet-seek"
+
+  def initialize
+    @cookie_helper = CookieHelper.new COOKIE_FILE_NAME
+    @commander = Commander.new
+  end
+
+  def seek *params
+    if @commander.search_mode?
+      params = read_keywords(*params)
+
+      puts "Keywords: #{params}" if @commander.runglish_mode?
+    end
+
+    process @commander.get_initial_mode, params
+  end
+
+  def process mode, *params
+    page = PageFactory.create(mode, params)
+
+    if mode == 'access'
+      process_access page, params[0]
+    elsif mode == 'login'
+      item = params[0]
+      cookie = page.login(*get_credentials)
+
+      @cookie_helper.save_cookie cookie
+
+      process("access", item)
+    else
+      items = page.items
+
+      if items.size > 0
+        display_items items
+        display_bottom_menu_part(mode)
+
+        user_selection = read_user_selection items
+
+        if not user_selection.quit?
+          current_item = items[user_selection.index]
+
+          if mode == 'main'
+            case current_item.link
+              when /announces.html/ then
+                process('announces')
+              when /freeTV.html/ then
+                process('freetv')
+              when /category=/ then
+                process('media', current_item.link)
+              when /action=channels/ then
+                process('channels', current_item.link)
+            end
+          elsif mode == 'channels'
+            if user_selection.archive?
+              process('media', current_item.archive_link)
+            else
+              process('media', current_item.link)
+            end
+          else # media : announces, freetv, category
+            if current_item.folder? or current_item.link =~ /action=view_recommended/
+              process('media', current_item.link)
+            else
+              process("access", current_item)
+            end
+          end
+        end
+      end
+    end
+  end
+
+  def process_access page, item
+    cookie = @cookie_helper.load_cookie
+
+    if cookie.nil?
+      process("login", item)
+    else
+      result = CookieHelper.get_auth_and_expires(cookie)
+      cookie_expire_date =  DateTime.strptime(result[1], "%A, %d-%b-%Y %H:%M:%S %Z")
+
+      if cookie_expire_date < DateTime.now # cookie expired?
+        @cookie_helper.delete_cookie
+
+        process("login", item)
+      else
+        media_info = page.request_media_info(item.media_file, cookie)
+
+        if media_info.session_expired?
+          @cookie_helper.delete_cookie
+
+          process("login", item)
+        else
+          LinkInfo.new(item, media_info)
+        end
+      end
+    end
+  end
+
+  def get_credentials
+    username = ask("Enter username :  " )
+    password = ask("Enter password : " ) { |q| q.echo = '*' }
+
+    [username, password]
+  end
+
+  def display_items items
+    if items.size == 0
+      puts "Empty search result."
+    else
+      items.each_with_index do |item, index|
+        puts "#{index+1}. #{item}"
+      end
+    end
+  end
+
+  def display_bottom_menu_part mode
+    puts "<number> => today; <number>.a => archive" if mode == 'channels'
+    puts "q. to exit"
+  end
+
+  def launch_link link
+    if RUBY_PLATFORM =~ /(win|w)32$/
+      `start wmplayer #{link}`
+    elsif RUBY_PLATFORM =~ /darwin/
+      `open #{link}`
+    end
+  end
+
+  private
+
+  def read_keywords input
+    keywords = input
+
+    if(keywords.strip.size == 0)
+      while keywords.strip.size == 0
+        keywords = ask("Keywords: ")
+      end
+    end
+
+    if RUBY_PLATFORM =~ /mswin32/ or @commander.runglish_mode?
+      keywords = Runglish::LatToRusConverter.new.transliterate(keywords)
+    end
+
+    keywords
+  end
+
+  def read_user_selection items
+    user_selection = UserSelection.new
+    
+    while true
+      user_selection.parse(ask("Select the number: "))
+
+      if not user_selection.blank?
+        if user_selection.quit? or user_selection.index < items.size
+          break
+        else
+          puts "Selection is out of range: [1..#{items.size}]"
+        end
+      end
+    end
+
+    user_selection
+  end
+
+end

data/lib/etvnet_seek/user_selection.rb

@@ -0,0 +1,45 @@
+class UserSelection
+  attr_reader :index
+
+  def parse text
+    @blank = text.strip.size == 0
+
+    result = text.split
+
+    @index = result[0].to_i-1
+    @quit = (result & ['q', 'Q']).empty? ? false : true
+    @archive = (result & ['a', 'A']).empty? ? false : true
+
+#    @quit = (['q', 'Q'].include? text) ? true : false
+#    @blank = text.strip.size == 0
+#    dot_index = text.index('.')
+#
+#    if not dot_index.nil?
+#      part1 = text[0..dot_index-1]
+#      part2 = text[dot_index+1..-1]
+#
+#      @index = (part1.to_i)-1
+#
+#      @archive = (['a', 'A'].include? part2) ? true : false
+#    else
+#      @index = text.to_i-1
+#    end
+  end
+
+  def quit?
+    @quit
+  end
+
+  def blank?
+    @blank
+  end
+
+  def archive?
+    @archive
+  end
+
+  def item items
+    items[index]
+  end
+
+end