dlc 1.1.0

data/README.rdoc

@@ -0,0 +1,5 @@
+= Ruby DLC
+
+Take a look at the documentation for more information: http://rdoc.info/jphastings/ruby-dlc/
+
+There's a gist about using the library for the first time here: http://gist.github.com/106978

data/Rakefile

@@ -0,0 +1,42 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "dlc"
+    gem.description = "Allows the generation of DLC container files (of JDownloader fame) from ruby"
+    gem.summary = "Allows the generation of DLC container files (of JDownloader fame) from ruby"
+    gem.email = "jphastings@gmail.com"
+    gem.homepage = "http://github.com/jphastings/ruby-DLC"
+    gem.authors = ["JP Hastings-Spital"]
+    gem.add_dependency "builder"
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/test_*.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+task :test => :check_dependencies
+
+task :default => :test

data/VERSION

@@ -0,0 +1 @@
+1.1.0

data/dlc_settings.yml

@@ -0,0 +1,8 @@
+--- 
+:keycache: 
+  :encoded_key: Sm91aTRQUXpQWkxUK3c1NjFKeDNvcGR4V09LdWdtbVQ4ZGFST1Nja2Q4RFQ0dFU2c2c3M1J2N2FCUzBPU2pKdw==
+  :key: 7aac8b326e4ca53e
+  :expires: 1276026954
+:url: http://google.com
+:email: jphastings@gmail.com
+:name: JP Hastings-Spital

data/lib/dlc.rb

@@ -0,0 +1,317 @@
+# = The DLC API in ruby!
+# Visit http://jdownloader.org for more information on why this might be useful!
+#
+# My thanks go out to the JDownloader staff for making such an excellent application! I've spoken with them and this script is now available for download at the github page (http://github.com/jphastings/ruby-DLC) or you can install the gem with the usual:
+#   gem install dlc
+#
+# == How to Use
+# You can use this library from the irb, or in any script you like. Examples below are for irb, I'm sure people using this as a 
+# library will be able to figure out how to use it in that way.
+#
+# === Set Up
+# You'll need to set the settings file before you do anything else. Open up irb in the directory where this file is located:
+#   require 'dlc'
+#   # => true
+#   s = DLC::Settings.new
+#   # No settings file exists.
+#   s.name = "Your Name"
+#   s.url = "http://yourdomain.com"
+#   s.email = "your.name@yourdomain.com"
+#
+# Now your settings file has been created you can go about making DLCs!
+#
+# === Creating a Package and DLC
+# The following irb example shows the variety of options you have while creating a package
+#   pkg = DLC::Package.new
+#   # => Unnamed package (0 links, 0 passwords)
+#   pkg.name = "My Package"
+#   # => "My Package"
+#   pkg.comment = "An exciting package!"
+#   # => "An exciting package!"
+#   pkg.category = "Nothing useful"
+#   # => "Nothing useful"
+#   pkg.add_link("http://google.com/")
+#   # => true
+#   pkg.add_link(["http://bbc.co.uk","http://slashdot.org"])
+#   # => true
+#   pkg.add_password("I don't really need one of these")
+#   # => true
+#   pkg
+#   # "My Package" (3 links, 1 passwords)
+#   # # An exciting package!
+#
+# If you want to put the DLC data into a file you should do:
+#   open("my_dlc.dlc","w") do |f|
+#     f.write pkg.dlc
+#   end
+#
+# This will ensure the file gets closed after you've written your DLC to it.
+#
+# == Problems?
+# Found a bug? Leave me an issue report on the githib page: http://github.com/jphastings/ruby-DLC/issues - 
+# I'll get onto it as soon as I can and see if I can fix it.
+#
+# == More Information
+# I'm JP, you can find my things at http://byJP.me. Any questions can be sent to me via twitter: @jphastings.
+#
+# I did this entirely for fun, please take that into account if/when you ask for help or before you get in touch. If you have any code improvements
+# please do let me know! I hope you enjoy this!
+
+require 'yaml'
+require 'time'
+require 'net/http'
+require 'digest/md5'
+
+require 'rubygems'
+require 'builder'
+require 'ruby-aes'
+
+# A hack to make the AES module accept string keys when they look like hex!
+def Aes.check_iv(iv)
+  return iv
+end
+
+# The DLC module, this is the container for the Settings and Package classes. It also contains some private helper functions
+module DLC
+  Api = {
+    :version => 1.0,
+    :pair_expires_after => 3600,
+    :service_urls => ["http://service.jdownloader.org/dlcrypt/service.php"],
+  }
+  
+  # Settings is a class that deals with information about the group using the DLC api.
+  #
+  # The class contains code to write its own settings file. Before using this ruby DLC api
+  # you should require it in irb and set the details:
+  #   require 'dlc'
+  #   s = DLC::Settings.new
+  #   s.name = "Your name"
+  #   s.email = "you@yourdomain.com"
+  #   s.url = "http://yourdomain.com/why_i_use_dlc.html"
+  class Settings
+    attr_accessor :email,:name,:url
+    
+    # I may allow this to be changed in later versions
+    Settings = "dlc_settings.yml"
+    
+    def initialize
+      if File.exists? Settings
+        begin
+          s = YAML.load(open(Settings))
+          @email = s[:email]
+          @name = s[:name]
+          @url = s[:url]
+          @keycache = (s[:keycache][:expires].nil? or s[:keycache][:expires] < Time.now.to_i) ? {} : s[:keycache]
+        rescue
+          raise SettingsNotValidError, "Your settings file is not valid. Please remove it."
+        end
+      else
+        $stderr.puts "No settings file exists. Read the documentation to find out how to make one."
+        @keycache = {}
+      end
+    end
+    
+    # Validate email address entry
+    def email=(email)
+      if email =~ /\b[A-Z0-9._%+-]+@[A-Z0-9.-]+\.[A-Z]{2,4}\b/i
+        @email = email
+        write_settings
+      else
+        $stderr.puts "That is an invalid email address"
+      end
+    end
+    # Must have a Name to make DLCs
+    def name=(name)
+      if not name.nil? and name.length != 0
+        @name = name
+        write_settings
+      else
+        $stderr.puts "You must use a full name"
+      end
+    end
+    # Must have a URL (starting in http:// or https://) to make DLCs
+    def url=(url)
+      if url =~ /^http(s?)\:\/\//i
+        @url = url
+        write_settings
+      else
+        $stderr.puts "Please include a leading http://"
+      end
+    end
+    
+    # Allows the cache of the key/encoded key pairs
+    def set_keycache(key,encoded_key,expires = 3600)
+      @keycache = {
+        :expires     => Time.now.to_i + expires,
+        :key         => key,
+        :encoded_key => encoded_key
+      }
+      write_settings
+    end
+    
+    # Retrieve the key from the cache, if there is one there.
+    # This will raise a +NoKeyCachedError+ if there is no key.
+    def get_keycache
+      if @keycache[:expires].nil? or @keycache[:expires] < Time.now.to_i
+        @keycache = {}
+        raise NoKeyCachedError, "There is no key in the cache"
+      else
+        return @keycache[:key],@keycache[:encoded_key]
+      end
+    end
+    
+    # A helper for irb people, and hands-on developlers
+    def inspect
+      if @name.nil? or @email.nil? or @url.nil?
+        $stderr.puts "You need to specify a name, url and email for the DLC generator. See the documentation"
+      else
+        "DLC API owner: #{@name} <#{@email}> (#{@url})"
+      end
+    end
+    
+    private
+    def write_settings
+      open(Settings,"w") do |f|
+        f.write YAML.dump({:name => @name, :email => @email, :url => @url,:keycache => @keycache})
+      end
+    end
+  end
+  
+  # The DLC package handler class. Make a new one of these for each package you want to create.
+  class Package
+    attr_reader :links,:passwords
+    attr_accessor :comment, :name, :category
+    
+    # Makes sure all the defaults are set (this would make a valid, if useless, package)
+    def initialize
+      @links = []
+      @passwords = []
+      @category = "various"
+      @comment = ""
+    end
+      
+    # Adds a link to the package
+    # Will take an array of links too
+    # I will, at some point, include the ability to specify filename and size.
+    def add_link(url)
+      if url.is_a?(Array)
+        url.each do |u|
+          self.add_link(u)
+        end
+        return @links
+      end
+      if url.is_a?(String) and url =~ /^http(s)?\:\/\//
+        @links.push({:url=>url,:filename=>nil,:size=>0})
+        return @links
+      end
+      raise RuntimeError, "Invalid URL: #{url}"
+    end
+    
+    alias :add_links :add_link
+    
+    # Adds a password to the package
+    # Also accepts an array of passwords
+    def add_password(password)
+      if password.is_a?(Array)
+        password.each do |p|
+          self.add_password(p)
+        end
+        return @passwords
+      end
+      if password.is_a?(String)
+        @passwords.push(password)
+        return @passwords
+      end
+      raise RuntimeError, "Invalid password: #{password}"
+    end
+    
+    alias :add_passwords :add_password
+    
+    # Gives you the DLC of the package you've created. First run (every hour) will take longer than the others while
+    # the jdownloader service is queried for information.
+    def dlc
+      settings = DLC::Settings.new
+      if settings.inspect.nil?
+        raise NoGeneratorDetailsError, "You must enter a name, url and email for the generator. See the documentation."
+      end
+      
+      xml = Builder::XmlMarkup.new(:indent=>0)
+      xml.dlc do
+        xml.header do
+          xml.generator do
+            xml.app(DLC.encode("Ruby DLC API (kedakai)"))
+            xml.version(DLC.encode(DLC::Api[:version]))
+            xml.url(DLC.encode(settings.url))
+          end
+          xml.tribute do
+            xml.name(DLC.encode(settings.name))
+          end
+          xml.dlcxmlversion(DLC.encode('20_02_2008'))
+        end
+        xml.content do
+          package = {:name => DLC.encode(@name)}
+          package[:passwords] = DLC.encode(@passwords.collect{|pw| "\"#{pw}\""}.join(",")) if @passwords.length != 0
+          package[:comment] = DLC.encode(@comment) if @comment != ""
+          package[:category] = DLC.encode(@category) if @category != ""
+          xml.package(package) do
+            @links.each do |link|
+              xml.file do
+                xml.url(DLC.encode(link[:url]))
+                xml.filename(DLC.encode(link[:filename]))
+                xml.size(DLC.encode(link[:size]))
+              end
+            end
+          end
+        end
+      end
+      
+      # Lets get a key/encoded key pair
+      begin
+        key, encoded_key = settings.get_keycache
+      rescue NoKeyCachedError
+        # Generate a key
+        expires = 3600
+        key = Digest::MD5.hexdigest(Time.now.to_i.to_s+"salty salty"+rand(100000).to_s)[0..15]
+        begin
+          if Net::HTTP.post_form(URI.parse(DLC::Api[:service_urls][rand(DLC::Api[:service_urls].length)]),{
+            :data    => key, # A random key
+            :lid     => DLC.encode([settings.url,settings.email,expires].join("_")), # Details about the generator of the DLC
+            :version => DLC::Api[:version],
+            :client  => "rubydlc"
+          }).body =~ /^<rc>(.+)<\/rc><rcp>(.+)<\/rcp>$/
+            encoded_key = $1
+            # What is the second part?!
+            settings.set_keycache(key, encoded_key, expires)
+          else
+            raise ServerNotRespondingError
+          end
+        rescue
+          raise ServerNotRespondingError, "The DLC service is not responding in the expected way. Try again later."
+        end
+      end
+
+      b64 = DLC.encode(xml.target!)
+      DLC.encode(Aes.encrypt_buffer(128,"CBC",key,key,b64.ljust((b64.length/16).ceil*16,"\000")))+encoded_key
+    end
+    
+    # Gives some useful information when people use the library from irb
+    def inspect
+      ((@name.nil?) ? "Unnamed package" : "\"#{@name}\"" )+" (#{@links.length} links, #{@passwords.length} passwords)"+((@comment == "") ? "" : "\n# #{@comment}")
+    end
+  end
+  
+  # For when the settings file is invalid
+  class SettingsNotValidError < StandardError; end
+  # For when a DLC is requested without settings set
+  class NoGeneratorDetailsError < StandardError; end
+  # For when the keycache is accessed and no valid key is available
+  class NoKeyCachedError < StandardError; end
+  # For when the service is not responding in the expected manner
+  class ServerNotRespondingError < StandardError; end  
+  
+  private
+  def self.encode(string)
+    string = "n.A." if string.nil?
+    return [string.to_s].pack("m").gsub("\n","")
+  end
+end

metadata

@@ -0,0 +1,84 @@
+--- !ruby/object:Gem::Specification 
+name: dlc
+version: !ruby/object:Gem::Version 
+  hash: 19
+  prerelease: false
+  segments: 
+  - 1
+  - 1
+  - 0
+  version: 1.1.0
+platform: ruby
+authors: 
+- JP Hastings-Spital
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-06-08 00:00:00 +01:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: builder
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+description: Allows the generation of DLC container files (of JDownloader fame) from ruby
+email: jphastings@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.rdoc
+files: 
+- README.rdoc
+- Rakefile
+- VERSION
+- dlc_settings.yml
+- lib/dlc.rb
+has_rdoc: true
+homepage: http://github.com/jphastings/ruby-DLC
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Allows the generation of DLC container files (of JDownloader fame) from ruby
+test_files: []
+