# encoding:utf-8

module Javlibrary
    def Javlibrary.downloader(identifer)
        baseurl = "http://www.jav11b.com/cn/?v=#{identifer}"
        response = Mechanize.new
        response.user_agent = Mechanize::AGENT_ALIASES.values[rand(21)]
        begin
            response.get baseurl
        rescue
            retry
        end

        doc = Nokogiri::HTML(response.page.body)

        video_title, details, video_genres, video_jacket_img = String.new, Array.new, String.new, String.new

        video_title = doc.search('div[@id="video_title"]/h3/a').children.text
        doc.search('//div[@id="video_info"]/div[@class="item"]/table/tr/td[@class="text"]').map do |row|
            details << row.children.text
        end

        doc.search('//div[@id="video_genres"]/table/tr/td[@class="text"]/span[@class="genre"]/a').each do |row|
            video_genres << row.children.text << " "
        end

        doc.search('//img[@id="video_jacket_img"]').each do |row|
            video_jacket_img = row['src']
        end

        # return data format: title$id$date$director$maker$label$cast$genres$img_url
        "#{video_title}$#{details[0]}$#{details[1]}$#{details[2]}$#{details[3]}$#{details[4]}$#{details[-1]}$#{video_genres}$#{video_jacket_img}"
        #result = Hash.new
        #result["title"] = video_title; result["id"] = details[0]; result["date"] = details[1]
        #result["director"] = details[2]; result["maker"] = details[3]; result["label"] = details[4]
        #result["cast"] = details[-1]; result["genres"] = video_genres; result["img_url"] = video_jacket_img
    end

    def Javlibrary.video_info_insert(client, index, identifer, actor_hash, genres_hash)
        result = downloader(identifer)
        title, id, date, director, maker, label, cast_tmp, genres_tmp, img_url = downloader(identifer).split('$')
        cast = cast_tmp.split.reject(&:empty?)
        genres = genres_tmp.split.reject(&:empty?)
        begin
            client.query("INSERT INTO video (video_id,video_name,license,url,director,label,date,maker)
            VALUES (#{index},'#{title}','#{id}','#{img_url}','#{director}','#{label}','#{date}','#{maker}')")
        rescue
            return
        end
        cast.each do |a|
            a_tmp = actor_hash[a]
            next if a_tmp == nil
            client.query("INSERT INTO v2a (v2a_fk_video,v2a_fk_actor) VALUES(#{index}, #{a_tmp.to_i})")
        end

        genres.each do |g|
            g_tmp = genres_hash[g]
            next if g_tmp == nil
            client.query("INSERT INTO v2c (v2c_fk_video,v2c_fk_category) VALUES(#{index}, #{g_tmp.to_i})")
        end

        client.query("UPDATE label SET video_download=1 WHERE video_num=#{index}")
        return nil
    end

    def download_all_video
        client = Javlibrary.client
        result = client.query("SELECT video_num, video_label FROM label WHERE video_download=0")
        client.close

        video_array = Array.new
        result.each do |item|
            video_array << item
        end

        video_array = video_array.each_slice(5000).to_a
        actor_hash = Javlibrary::actor_hash
        genre_hash = Javlibrary::genre_hash
        thread_pool = Array::new

        video_array.each do |group|
            # Create a download thread
            thread_temp = Thread.new {
                client = Javlibrary.client
                group.each do |item|
                    begin
                        video_info_insert(client, item['video_num'], item['video_label'],
                            actor_hash, genre_hash)
                    rescue
                        next
                    end
                end
                client.close
            }
            thread_pool << thread_temp
        end
        thread_pool.map(&:join)
    end

    def test
        pp downloader('javlia322m')
    end

    module_function :test
    module_function :download_all_video
end

Javlibrary::download_all_video