Skip to content

Instantly share code, notes, and snippets.

@yihui
Forked from wereHamster/README.md
Created May 1, 2012 16:05
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save yihui/2569165 to your computer and use it in GitHub Desktop.
Save yihui/2569165 to your computer and use it in GitHub Desktop.
Script to upload files to github: ./github-upload <file> [<repo>]
#!/usr/bin/env ruby
# under Ubuntu, you should install ruby1.9.1 or higher (default 1.8 does not work)
require 'json'
require 'net/https'
require 'pathname'
# Extensions
# ----------
# We extend Pathname a bit to get the content type.
class Pathname
def type
flags = RUBY_PLATFORM =~ /darwin/ ? 'Ib' : 'ib'
`file -#{flags} #{realpath}`.chomp.gsub(/;.*/,'')
end
end
# Helpers
# -------
# Die if something goes wrong.
def die(msg); puts(msg); exit!(1); end
# Do a post to the given url, with the payload and optional basic auth.
def post(url, token, params, headers)
uri = URI.parse(url)
http = Net::HTTP.new(uri.host, uri.port)
http.use_ssl = true
req = Net::HTTP::Post.new(uri.path, headers)
req['Authorization'] = "token #{token}" if token
return http.request(req, params)
end
def urlencode(str)
str.gsub(/[^a-zA-Z0-9_\.\-]/n) {|s| sprintf('%%%02x', s[0].to_i) }
end
# Yep, ruby net/http doesn't support multipart. Write our own multipart generator.
# The order of the params is important, the file needs to go as last!
def build_multipart_content(params)
parts, boundary = [], "#{rand(1000000)}-we-are-all-doomed-#{rand(1000000)}"
params.each do |name, value|
data = []
if value.is_a?(Pathname) then
data << "Content-Disposition: form-data; name=\"#{urlencode(name.to_s)}\"; filename=\"#{value.basename}\""
data << "Content-Type: #{value.type}"
data << "Content-Length: #{value.size}"
data << "Content-Transfer-Encoding: binary"
data << ""
data << value.read
else
data << "Content-Disposition: form-data; name=\"#{urlencode(name.to_s)}\""
data << ""
data << value
end
parts << data.join("\r\n") + "\r\n"
end
[ "--#{boundary}\r\n" + parts.join("--#{boundary}\r\n") + "--#{boundary}--", {
"Content-Type" => "multipart/form-data; boundary=#{boundary}"
}]
end
# Configuration and setup
# -----------------------
# Get Oauth token for this script.
token = `git config --get github.upload-script-token`.chomp
# The file we want to upload, and repo where to upload it to.
file = Pathname.new(ARGV[0])
repo = ARGV[1] || `git config --get remote.origin.url`.match(/git@github.com:(.+?)\.git/)[1]
# The actual, hard work
# ---------------------
# Register the download at github.
res = post("https://api.github.com/repos/#{repo}/downloads", token, {
'name' => file.basename.to_s, 'size' => file.size.to_s,
'content_type' => file.type.gsub(/;.*/, '')
}.to_json, {})
die("File already exists.") if res.class == Net::HTTPClientError
die("GitHub doens't want us to upload the file.") unless res.class == Net::HTTPCreated
# Parse the body and use the info to upload the file to S3.
info = JSON.parse(res.body)
res = post(info['s3_url'], nil, *build_multipart_content({
'key' => info['path'], 'acl' => info['acl'], 'success_action_status' => 201,
'Filename' => info['name'], 'AWSAccessKeyId' => info['accesskeyid'],
'policy' => info['policy'], 'Signature' => info['signature'],
'Content-Type' => info['mime_type'], 'file' => file
}))
die("S3 is mean to us.") unless res.class == Net::HTTPCreated
# Print the URL to the file to stdout.
puts "#{info['s3_url']}#{info['path']}"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment