etvnet-seek 0.7.2

data/lib/etvnet_seek/core/new_item.rb

@@ -0,0 +1,9 @@
+class NewItem < MediaItem
+
+  attr_accessor :image
+  
+  def to_s
+    "#{text} ---  #{link}"
+  end
+
+end

data/lib/etvnet_seek/core/new_items_page.rb

@@ -0,0 +1,20 @@
+class NewItemsPage < BasePage
+
+  def items
+    list = []
+
+    document.css(".gallery ul li a").each do |item|
+      text = item.css("img").at(0).attributes['alt'].value.strip
+      src = item.css("img").at(0).attributes['src'].value.strip
+      href = item['href']
+
+      item = BrowseMediaItem.new(text, href)
+      item.image = src
+      
+      list << item
+    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.com"
+
+  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,51 @@
+require 'etvnet_seek/core/media_item'
+require 'etvnet_seek/core/new_item'
+require 'etvnet_seek/core/catalog_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/home_page'
+require 'etvnet_seek/core/base_page'
+require 'etvnet_seek/core/media_page'
+require 'etvnet_seek/core/search_page'
+require 'etvnet_seek/core/channels_page'
+require 'etvnet_seek/core/catalog_page'
+require 'etvnet_seek/core/new_items_page'
+require 'etvnet_seek/core/group_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
+        HomePage.new
+      when 'channels' then
+        ChannelsPage.new
+       when 'catalog' then
+        CatalogPage.new
+      when 'best_hundred' then
+        BestHundredPage.new
+      when 'new_items' then
+        NewItemsPage.new
+      when 'premiere' then
+        PremierePage.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,42 @@
+require 'cgi'
+
+class SearchPage < Page
+  SEARCH_URL = BASE_URL + "/search/"
+
+  def initialize(params)
+    super("#{SEARCH_URL}?q=#{CGI.escape(*params)}")
+  end
+
+  def items
+    list = []
+
+    document.css(".conteiner #results #table-onecolumn tr").each_with_index do |item, index|
+      next if index == 0
+      
+      showtime = item.css("td[1]").text.strip
+      rating_image = item.css("td[2] img").at(0) ? item.css("td[2] img").at(0).attributes['src'].value.strip : ""
+      rating = item.css("td[3]") ? item.css("td[3]").text.strip : ""
+      name = item.css("td[4]").text.strip
+      duration = item.css("td[5]") ? item.css("td[5]").text.strip : ""
+      link = item.css("td[4] a").at(0)
+
+      href = link.attributes['href'].value
+
+      amount_expr = name.scan(%r{(.*)+\((\d*)+ (.*)+\)})[0]
+
+      folder = (not amount_expr.nil? and amount_expr.size > 1) ? true : false
+
+      record = BrowseMediaItem.new(name, href)
+      record.folder = folder
+      record.showtime = showtime
+      record.duration = duration
+      #record.channel = channel
+      record.rating_image = rating_image
+      record.rating = rating
+
+      list << record
+    end
+
+    list
+  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/easy_auth.rb

@@ -0,0 +1,52 @@
+require 'digest/sha1'
+require 'digest/sha2'
+
+module EasyAuth
+  # http://techspeak.plainlystated.com/2010/03/drop-dead-simple-authentication-for.html
+
+  # To generate a hashed password (in irb):
+  # require 'easy_auth'
+  # EasyAuth.hash('my_password') # Put this in AUTHORIZED_USERS
+
+  AUTHORIZED_USERS = {
+    'patrick' => "4ded8fa58a5c16298e665b35353555c89b786d8"
+  }
+
+  def self.hash_password(password)
+    #Digest::SHA256.digest(password)
+
+    Digest::SHA1.hexdigest(password)
+  end
+
+  def hash_password(password)
+    EasyAuth.hash_password(password)
+  end
+
+  def auth_digest(username, password)
+    key = "#{username}::#{hash_password(password)}::#{request.env['REMOTE_ADDR']}::#{request.env['HTTP_USER_AGENT']}"
+    hash_password key
+  end
+
+  def if_auth(username = request.cookies["username"], password = nil)
+    if password.nil?
+      key = request.cookies["key"]
+    else
+      key = auth_digest(username, hash_password(password))
+    end
+
+    if authorized?(username, key)
+      response.set_cookie("username", username)
+      response.set_cookie("key", key)
+      yield
+    else
+      @error = "Login failed"
+      erb :'/admin/login'
+    end
+  end
+
+  def authorized?(username, key)
+    return false unless AUTHORIZED_USERS.has_key?(username)
+
+    key == auth_digest(username, AUTHORIZED_USERS[username])
+  end
+end

data/lib/etvnet_seek/etvnet_seek.rb

@@ -0,0 +1,10 @@
+require 'rubygems' unless RUBY_VERSION =~ /1.9.*/
+
+$:.unshift(File.dirname(__FILE__)) unless
+  $:.include?(File.dirname(__FILE__)) || $:.include?(File.expand_path(File.dirname(__FILE__)))
+
+require 'etvnet_seek/core/page_factory'
+
+require 'etvnet_seek/cookie_helper'
+require 'etvnet_seek/link_info'
+

data/lib/etvnet_seek/link_info.rb

@@ -0,0 +1,37 @@
+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
+
+  def rtsp_link
+    media_info.rtsp_link
+  end
+
+end

data/lib/etvnet_seek/main.rb

@@ -0,0 +1,180 @@
+require "highline/import"
+require 'optparse'
+require 'date'
+
+require 'etvnet_seek/commander'
+require 'etvnet_seek/user_selection'
+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 '/' then
+                process("main")
+              when /aired_today/ then
+                process('media', current_item.link)
+              when /catalog/ then
+                process('media', current_item.link)
+              when /tv_channels/ then
+                process('channels', current_item.link)
+            end
+          elsif mode == 'channels'
+            p user_selection.catalog?
+            if user_selection.catalog?
+              process('media', current_item.catalog_link)
+            else
+              process('media', current_item.link)
+            end
+          elsif mode == 'new_items'
+            process('access', current_item)
+          elsif mode == 'premiere'
+            process('access', current_item)
+          elsif mode == 'catalog'
+             p 'catalog'
+            process('media', current_item.link)
+          else # media
+            if current_item.folder? or current_item.link =~ /(catalog|tv_channel)/
+              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
+      expires = CookieHelper.get_expires(cookie)
+      cookie_expire_date =  DateTime.strptime(expires, "%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> c => catalog" 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