biliupload.rb

#!/usr/bin/env ruby
#encoding: utf-8
require 'json'
require 'net/http'
require 'uri'

class NoSessionError < RuntimeError
end

module BiliUpload
  module_function
  @@tasklist = []
  def upload(local_filename)
    begin
      listitem = {:local => local_filename}
      begin
        open(local_filename, "r") {|f| f.read(1) } # if the local file fails, ruby will raise a exception
      rescue
        listitem[:result] = :failed
        return
      end

      begin
        cookies = open(File.dirname(__FILE__) + "/bilicookies","r") {|f|f.read}
      rescue Errno::ENOENT
        cookies = "NO_COOKIE_ERROR"
      end
      http = Net::HTTP.new("member.bilibili.com")
      response = http.get('/get_upload_url', {"Cookie" => cookies})
      json = JSON.parse(response.body)
      if json["error_code"]
        raise NoSessionError.new(json["error_msg"])
      end
      upload_url = json["url"]
      remote_filename = json["file_name"]

      #print "> curl -F \"file=@#{local_filename};filename=#{remote_filename}\" '#{upload_url}'\n"
      IO.popen("curl -F \"file=@#{local_filename};filename=#{remote_filename}\" '#{upload_url}'") do |pipe|
        json = JSON.parse(pipe.read)
        if json["code"] == 0
          listitem[:result] = :success
          listitem[:vu] = remote_filename.split('|').first
        else
          listitem[:result] = :failed
        end
      end
    ensure
      @@tasklist.push listitem
    end
  end

  def tasklist
    @@tasklist
  end
end

begin
  ARGV.each {|file| BiliUpload.upload(file) }
ensure
  print "\n\nUpload results:\n"
  BiliUpload.tasklist.each {|h| p h }
end