kp_folder 0.0.4

data/Gemfile

@@ -0,0 +1,2 @@
+source :rubygems
+gem "kuaipan"

data/Gemfile.lock

@@ -0,0 +1,16 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    json (1.6.7)
+    kuaipan (0.0.3)
+      json (~> 1.6.6)
+      rest-client
+    mime-types (1.18)
+    rest-client (1.6.7)
+      mime-types (>= 1.16)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  kuaipan

data/README.rdoc

@@ -0,0 +1,16 @@
+= kp_folder
+
+kuaipan sync folders tool
+To install then command tool:
+
+  gem install kp_folder
+
+commands:
+  kp --help    # help doc
+  kp -a yourfolder_absolute_path # add a folder to sync
+  kp -rm yourfolder_absolute_ath # remove a folder
+  kp -s        # start sync all folders to kuaipan
+  kp -s ^\\..* # filter some files whose name start by '.'. You can give other regex str
+  kp -l        # list all folders added
+
+

data/Rakefile

@@ -0,0 +1,9 @@
+require 'rake/testtask'
+
+Rake::TestTask.new do |t|
+  t.libs << 'test'
+  t.pattern = 'test/**/test_*.rb'
+end
+
+desc "Run tests"
+task :default => :test

data/bin/kf

@@ -0,0 +1,44 @@
+#!/usr/bin/env ruby
+
+require 'kp_folder'
+
+case ARGV.shift
+when '-a'
+  arg1 = ARGV.shift if ARGV[0]
+  ARGV.clear
+  KpFolder.init
+  if arg1 && File.exist?(arg1) && File.directory?(arg1)
+    KpFolder.add(File.absolute_path(arg1)) # trip trailing '/'
+  end  
+when '-rm'
+  arg1 = ARGV.shift if ARGV[0]
+  ARGV.clear
+  KpFolder.init
+  if arg1 && File.exist?(arg1) && File.directory?(arg1)
+    KpFolder.remove(File.absolute_path(arg1)) # trip trailing '/'
+  end
+when '-s'
+  arg1 = ARGV.shift if ARGV[0]
+  ARGV.clear
+  KpFolder.init
+  if arg1
+    KpFolder.start(:filter => arg1)
+  else
+    KpFolder.start
+  end
+when '-l'
+  ARGV.clear
+  KpFolder.init
+  KpFolder.list 
+when '--help'
+  puts "kf -i         : init firstly"
+  puts "kf -a  folder : add a folder"
+  puts "kf -rm folder : remove a folder"
+  puts "kf -s         : start to sync all folders"
+  puts "kf -l         : list all folders which were added"
+  puts "a simple example, that will sync all files in folder '/home/tom/Documents':
+        1 kf -a /home/tom/Documents
+        2 kf -s"
+else
+  puts 'wrong params!please refer to help:kf --help'
+end

data/kp_folder.gemspec

@@ -0,0 +1,29 @@
+Gem::Specification.new do |s|
+  s.name = 'kp_folder'
+  s.version = '0.0.4'
+  s.executables << 'kf'
+  s.date = '2012-05-01'
+  s.summary = "kuaipan sync folder command tool"
+  s.description = "You can define folders which should be synced by 'kp' command."
+  s.authors = ["tiemei"]
+  s.email = 'jiadongkai@gmail.com'
+  s.files = [
+    "lib/kp_folder/config.rb",
+    "lib/kp_folder.rb",
+    "bin/kf",
+    "Gemfile",
+    "Gemfile.lock",
+    "README.rdoc",
+    "Rakefile",
+    "kp_folder.gemspec",
+    "test/kp_folder/test_config.rb",
+    "test/test_client.rb"
+    ]
+  s.require_paths = ["lib"]
+  s.homepage = 'http://rubygems.org/gems/kp_folder'
+  s.required_ruby_version = '>= 1.9.3'
+  s.required_rubygems_version = ">= 1.3.6"
+
+  s.add_runtime_dependency 'kuaipan', '~> 0.0.3'
+
+end

data/lib/kp_folder.rb

@@ -0,0 +1,149 @@
+# conding: utf-8
+require 'kuaipan'
+require 'json'
+
+require 'set'
+
+require 'kp_folder/config'
+
+module KpFolder
+  extend Kuaipan::OpenAPI
+  class << self
+    attr_reader :k_session
+  end
+  def self.init
+    @oauth_result_file = "#{ Dir.home }/.kf_oauth_result.json"
+    input_config(Config.consumer_key, Config.consumer_secret) 
+    @k_session = nil
+    if File.exist?(@oauth_result_file)
+      File.open(@oauth_result_file, 'r')do |f|
+        oauth_result = JSON.parse(f.read)
+        @k_session = g_session_skip_oauth(oauth_result['oauth_token'],
+                                          oauth_result['oauth_token_secret'],
+                                          oauth_result['user_id'])
+      end
+    else
+      @k_session = g_session  
+      authorize_url = @k_session[:authorize_url]
+      puts 'please copy the url to your brower,then get number here:'
+      puts authorize_url
+      oauth_verifier = gets
+      oauth_result = @k_session.set_atoken oauth_verifier
+      File.open(@oauth_result_file, 'w+')do |f|
+        f.write(JSON.generate(oauth_result))
+      end
+    end
+    @folders_file = "#{ Dir.home }/.kf_folders"
+    @db_file = "#{ Dir.home }/.kf_db"
+    File.open(@folders_file, 'w'){|f| f.write('[]')} unless File.exist?(@folders_file)
+    File.open(@db_file, 'w'){|f| f.write('{}')} unless File.exist?(@db_file)
+  end
+
+  def self.list
+    File.open(@folders_file, 'r')do |f|
+      arr = JSON.parse(f.read)
+      arr.each {|item| puts item}
+    end
+  end
+
+  def self.remove(folder)
+    arr = []
+    File.open(@folders_file, 'r')do |f|
+      arr = JSON.parse(f.read)
+    end
+    if arr.delete(folder)
+      File.open(@folders_file, 'w')do |f|
+        f.write(JSON.generate(arr))
+      end
+      puts "success!remove #{ folder }"
+    else
+      puts "failed to remove #{ folder }"
+    end
+  end
+
+  def self.add(folder)
+    arr = []
+    File.open(@folders_file, 'r')do |f|
+      arr = JSON.parse(f.read)
+    end
+    set = arr.to_set
+    set.add(folder)
+    File.open(@folders_file, 'w')do |f|
+      f.write(JSON.generate(set.to_a))
+    end
+    puts "success to add #{folder}"
+  end
+
+  # 1.upload local files that aren't in file '.db',and update .db
+  # 2.clear remote files
+  def self.start(opts={})
+    filter = Regexp.compile(filter) if (filter = opts.delete(:filter))
+    @local_files = []
+    File.open(@folders_file, 'r'){|f| @folders = JSON.parse(f.read)}
+    @folders.each do |folder|
+      files_in_folder(folder)
+    end
+    
+    File.open(@db_file, 'r')do |f|
+      @db = JSON.parse(f.read)
+    end
+    @db_tmp = {}
+    @local_files.each do |file_name|
+      next if filter && (filter =~ File.basename(file_name))
+      m = %r{/.*/}.match(file_name)
+      file = File.open(file_name, 'rb')
+      if @db[file_name] == nil 
+        puts "upload file: \t\t#{ file_name }"
+        k_session.upload_file(file, :path => m[0])
+        @db_tmp[file_name] = k_session.metadata(file_name)
+      elsif @db[file_name]['size'] != File.size(file_name)
+        k_session.delete(file_name)
+        puts "update file: \t\t#{ file_name }"
+        k_session.upload_file(file, :path => m[0])
+        @db_tmp[file_name] = k_session.metadata(file_name)
+      else
+        @db_tmp[file_name] = @db[file_name]
+      end
+    end
+    @db = @db_tmp
+    File.open(@db_file, 'w'){|f| f.write(JSON.generate(@db))}
+    
+    @files_remote = {}
+    @folders.each do |folder|
+      check_folder(folder)
+    end
+    
+    diff = @files_remote.select {|k,_| @db[k] == nil }
+    diff.each do |k,_| 
+      k_session.delete(k) 
+      puts "clear remote file: \t#{ k }"
+    end
+  end
+
+  private
+    def self.files_in_folder(folder)
+      Dir.foreach(folder)do |name|
+        absolute_path = "#{ folder }/#{ name }"
+        @local_files <<  absolute_path  if File.file?(absolute_path)
+        next if (name == '.' or name == '..')
+        files_in_folder(absolute_path) if File.directory?(absolute_path)
+      end
+    end
+  
+    def self.check_folder(folder)
+      puts "check remote folder: \t#{ folder }"
+      begin
+        @k_session.metadata(folder)
+      rescue Kuaipan::NotFound => nf
+        @k_session.create_folder(folder)
+      ensure
+        hash = @k_session.metadata(folder)
+        files = hash['files']
+        files.each do |f|
+          @files_remote["#{ hash['path'] }/#{ f['name'] }"] = f if f['type'] == 'file'        
+          check_folder("#{ hash['path'] }/#{ f['name'] }") if f['type'] == 'folder'
+        end 
+      end
+    end
+
+end

data/lib/kp_folder/config.rb

@@ -0,0 +1,10 @@
+# conding: utf-8
+
+module KpFolder
+  module Config
+    @consumer_key, @consumer_secret = %w(xcWcZhCNKFJz1H8p 8RvkM0aGYiQF5kJF)
+    class << self
+      attr_reader :consumer_key, :consumer_secret
+    end
+  end 
+end

data/test/kp_folder/test_config.rb

@@ -0,0 +1,11 @@
+# conding: utf-8
+
+require 'kp_folder'
+require 'test/unit'
+
+class ConfigTest < Test::Unit::TestCase
+  def test_config
+    assert_not_nil KpFolder::Config.consumer_key
+    assert_not_nil KpFolder::Config.consumer_secret
+  end
+end

data/test/test_client.rb

@@ -0,0 +1,39 @@
+# conding: utf-8
+
+require 'kp_folder'
+require 'test/unit'
+require 'uri'
+
+class KpFolderTest < Test::Unit::TestCase
+  def test_init
+    KpFolder.init
+    p KpFolder.k_session.metadata('/') 
+    KpFolder.k_session.create_folder('/tiemei_test/')
+    p KpFolder.k_session.metadata('/tiemei_test')
+    p KpFolder.k_session.delete('/tiemei_test')
+#KpFolder.k_session.metadata('/tiemei_test')
+  end
+
+  def test_rm_add
+    KpFolder.add('/home/jdk')
+    KpFolder.add('/home/jdk/Desktop')
+    KpFolder.remove('/home/jdk/Desktop')
+    KpFolder.add('/home/jdk')
+    KpFolder.remove('/home/jdk')
+  end
+  
+  def test_files_in_folders
+    KpFolder.instance_eval do
+      @local_files = []
+      files_in_folder('/home/jdk/codespace/ruby_space/kp_folder/lib')
+      p @local_files
+    end
+  end
+
+  def test_start
+    KpFolder.add('/home/jdk/codespace/ruby_space/kp_folder/lib')
+    KpFolder.start
+  end
+  
+end
+

metadata

@@ -0,0 +1,71 @@
+--- !ruby/object:Gem::Specification
+name: kp_folder
+version: !ruby/object:Gem::Version
+  version: 0.0.4
+  prerelease: 
+platform: ruby
+authors:
+- tiemei
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-05-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: kuaipan
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.0.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.0.3
+description: You can define folders which should be synced by 'kp' command.
+email: jiadongkai@gmail.com
+executables:
+- kf
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/kp_folder/config.rb
+- lib/kp_folder.rb
+- bin/kf
+- Gemfile
+- Gemfile.lock
+- README.rdoc
+- Rakefile
+- kp_folder.gemspec
+- test/kp_folder/test_config.rb
+- test/test_client.rb
+homepage: http://rubygems.org/gems/kp_folder
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.9.3
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.3.6
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.22
+signing_key: 
+specification_version: 3
+summary: kuaipan sync folder command tool
+test_files: []