Skip to content

M
mycard
Commit 3a302366 authored by 神楽坂玲奈's avatar 神楽坂玲奈
Browse files
Options

server

parent e783bb9d
Hide whitespace changes
Inline Side-by-side
Showing with 78 additions and 16 deletions
main.rb
View file @ 3a302366
......@@ -14,16 +14,48 @@ if defined?(Ocra) or defined?(Exerb)
require 'rb-notifu'
require 'sqlite3'
require 'socket'
require 'digest/md5'
#open-uri protocol
require 'net/http'
#websocket autoload
WebSocket::Error
WebSocket::ExceptionHandler
WebSocket::Frame::Base
WebSocket::Frame::Data
WebSocket::Frame::Handler::Base
WebSocket::Frame::Handler::Handler03
WebSocket::Frame::Handler::Handler04
WebSocket::Frame::Handler::Handler05
WebSocket::Frame::Handler::Handler07
WebSocket::Frame::Handler::Handler75
WebSocket::Frame::Incoming::Client
WebSocket::Frame::Incoming::Server
WebSocket::Frame::Outgoing::Client
WebSocket::Frame::Outgoing::Server
WebSocket::Handshake::Base
WebSocket::Handshake::Client
WebSocket::Handshake::Handler::Base
WebSocket::Handshake::Handler::Client
WebSocket::Handshake::Handler::Client01
WebSocket::Handshake::Handler::Client04
WebSocket::Handshake::Handler::Client75
WebSocket::Handshake::Handler::Client76
WebSocket::Handshake::Handler::Server
WebSocket::Handshake::Handler::Server04
WebSocket::Handshake::Handler::Server75
WebSocket::Handshake::Handler::Server76
WebSocket::Handshake::Server
exit
end
# = Runtime
begin
# == initialize
Version = "0.1.0"
Version = "2.0.0"
Platform = (RUBY_PLATFORM['mswin'] || RUBY_PLATFORM['mingw']) ? :win32 : :linux
System_Encoding = Encoding.find("locale") rescue Encoding.find(Encoding.locale_charmap)
Dir.chdir File.dirname(defined?(ExerbRuntime) ? ExerbRuntime.filepath.dup.force_encoding(System_Encoding).encode!(Encoding::UTF_8) : ENV["OCRA_EXECUTABLE"] || __FILE__)
......@@ -165,9 +197,14 @@ begin
TCPServer.new('0.0.0.0', Config['port']).close rescue return #check port in use, seems eventmachine enabled IP_REUSEADDR.
require 'websocket-eventmachine-server'
EventMachine.run do
ygopro_version = nil
connections = []
WebSocket::EventMachine::Server.start(:host => "0.0.0.0", :port => Config['port']) do |ws|
ws.onopen do
ws.send({'version' => Version}.to_json)
connections.push ws
msg = {'version' => Version}
msg['ygopro_version'] = ygopro_version if ygopro_version
ws.send(msg.to_json)
end
ws.onmessage do |msg, type|
......@@ -175,9 +212,18 @@ begin
end
ws.onclose do
puts "Client disconnected"
connections.delete ws
exit if connections.empty?
end
end
require 'em-http'
http = EM::HttpRequest.new('https://my-card.in/ygopro_version.json').get
http.callback { |http|
md5 = Digest::MD5.file(Config['ygopro']['path']).hexdigest
ygopro_version = JSON.parse(http.response)[md5] || md5
connections.each { |ws| ws.send ({'ygopro_version' => ygopro_version}).to_json }
}
if File.file? File.join(File.dirname(Config['ygopro']['path']), 'cards.cdb')
require 'sqlite3'
db = SQLite3::Database.new File.join(File.dirname(Config['ygopro']['path']), 'cards.cdb')
......@@ -186,12 +232,10 @@ begin
thumbnails_to_download = cards - Dir.glob(File.join(File.dirname(Config['ygopro']['path']), 'pics', 'thumbnail', '*.jpg')).collect { |file| File.basename(file, '.jpg').to_i }
end
unless images_to_download.empty? and thumbnails_to_download.empty?
require 'em-http'
require 'uri'
http = EM::HttpRequest.new('https://my-card.in/cards/image.json').get
http.callback {
response = JSON.parse http.response
p response
image_url = URI(response['url'])
thumbnail_url = URI(response['thumbnail_url'])
Dir.mkdir File.join(File.dirname(Config['ygopro']['path']), 'pics') unless File.directory? File.join(File.dirname(Config['ygopro']['path']), 'pics')
......@@ -199,7 +243,20 @@ begin
files = {}
thumbnails_to_download.each { |card_id| files[thumbnail_url.path.gsub(':id', card_id.to_s)] = File.join(File.dirname(Config['ygopro']['path']), 'pics', 'thumbnail', card_id.to_s + '.jpg') }
images_to_download.each { |card_id| files[image_url.path.gsub(':id', card_id.to_s)] = File.join(File.dirname(Config['ygopro']['path']), 'pics', card_id.to_s + '.jpg') }
batch_download(image_url.to_s, files, 'image/jpeg')
thumbnails_count = thumbnails_to_download.size
images_count = images_to_download.size
errors_count = 0
batch_download(image_url.to_s, files, 'image/jpeg') { |http|
if http.req.path["thumbnail"]
thumbnails_count -= 1
else
images_count -= 1
end
unless http.response_header.status == 200 and http.response_header['CONTENT_TYPE'] == 'image/jpeg'
errors_count += 1
end
connections.each { |ws| ws.send ({'images_download_images' => images_count, 'images_download_thumbnails' => thumbnails_count, 'images_download_errors' => errors_count}).to_json }
}
}
require 'rb-notifu'
Notifu::show :message => "缩略: #{thumbnails_to_download.size}, 完整: #{images_to_download.size}".encode(System_Encoding), :baloon => false, :type => :info, :title => "正在下载卡图".encode(System_Encoding) do |status|
......@@ -211,13 +268,13 @@ begin
end
end
def batch_download(main_url, files, content_type=nil)
def batch_download(main_url, files, content_type=nil, &block)
connections = {}
count = {total: files.size, error: 0}
[10*100, files.size].min.times { do_download(main_url, files, content_type, count, connections) }
[10*100, files.size].min.times { do_download(main_url, files, content_type, count, connections, &block) }
end
def do_download(main_url, files, content_type, count, connections)
def do_download(main_url, files, content_type, count, connections, &block)
if connections.size < 10
connection = EventMachine::HttpRequest.new(main_url)
connections[connection] = 0
......@@ -232,13 +289,13 @@ begin
remote_path, local_path = files.shift
connections[connection] += 1
connection.get(path: remote_path, keepalive: connections[connection] != 100).callback { |http|
puts File.basename local_path
#puts File.basename local_path
count[:error] = 0
count[:total] -= 1
if http.response_header['CONNECTION'] != 'keep-alive'
connection.close
connections.delete(connection)
do_download(main_url, files, content_type, count, connections) while !files.empty? and (connections.size < 10 or connections.values.min < 100)
do_download(main_url, files, content_type, count, connections, &block) while !files.empty? and (connections.size < 10 or connections.values.min < 100)
end
if http.response_header.status == 200 and (!content_type or http.response_header['CONTENT_TYPE'] == content_type)
......@@ -247,12 +304,13 @@ begin
puts http.response_header.http_status
end
yield http
if count[:total].zero?
connections.each_key { |connection| connection.close }
connections.clear
puts 'all done'
EM.stop
end
}.errback { |http|
puts http.error
connection.close
......@@ -260,12 +318,11 @@ begin
files[remote_path] = local_path
count[:error] += 1
if count[:error] <= 10*100
do_download(main_url, files, content_type, count, connections) while !files.empty? and (connections.size < 10 or connections.values.min < 100)
do_download(main_url, files, content_type, count, connections, &block) while !files.empty? and (connections.size < 10 or connections.values.min < 100)
else
connections.each_key { |connection| connection.close }
connections.clear
puts 'network error'
EM.stop
end
}
end
......@@ -455,10 +512,13 @@ begin
class <<self
def join(*args)
end
def expand_path(*args)
end
def dirname(*args)
end
def open(*args)
result = ExerbRuntime.open('unicode.data')
if block_given?
......@@ -481,7 +541,9 @@ begin
parse ARGV.first.dup.force_encoding(System_Encoding).encode!(Encoding::UTF_8)
else
register if !registed?
if File.file? 'ruby\bin\rubyw.exe'
if File.file? 'nw.exe'
spawn 'nw.exe'
elsif File.file? 'ruby\bin\rubyw.exe'
spawn 'ruby\bin\rubyw.exe', '-KU', 'lib/main.rb'
else
System.web(Config['url'])
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment