Add bandwidth limit option.

[osdn-codes/osdn-cli.git] / lib / osdn / cli / command / relfile.rb

diff --git a/lib/osdn/cli/command/relfile.rb b/lib/osdn/cli/command/relfile.rb
index b001c38..b7ad732 100644 (file)
--- a/lib/osdn/cli/command/relfile.rb
+++ b/lib/osdn/cli/command/relfile.rb
@@ -5,19 +5,23 @@ module OSDN; module CLI; module Command
   class Relfile < FrsBase
     def help
       puts "#{$0} relfile [opts] [list]"
-      puts "#{$0} relfile [opts] create <target-file>"
+      puts "#{$0} relfile [opts] create <target-file> [target-files...]"
       puts "#{$0} relfile [opts] update <numeric-file-id>"
       puts "#{$0} relfile [opts] delete <numeric-file-id>"
       puts "Options:"
       puts "  -f --format=<pretty|json>  Set output format"
       puts "  -p --project=<project>     Target project (numeric id or name)"
-      puts "     --package=<project>     Target package (numeric id)"
-      puts "     --release=<project>     Target release (numeric id)"
+      puts "     --package=<package-id>  Target package (numeric id)"
+      puts "     --release=<release-id>  Target release (numeric id)"
       puts "  -v --visibility=<public|private|hidden>"
+      puts "      --force-digest         Calc local file digest forcely"
+      puts "      --progress             Force to show upload progress"
+      puts "      --no-progress          Force to hide upload progress"
+      puts "      --bwlimit=RATE         Limit bandwidth (in KB)"
     end
 
     def self.description
-      "Manipulate release (includes by package) of project"
+      "Manipulate frs files of project"
     end
 
     def process_options
@@ -27,6 +31,10 @@ module OSDN; module CLI; module Command
         [ '--package', GetoptLong::REQUIRED_ARGUMENT ],
         [ '--release', GetoptLong::REQUIRED_ARGUMENT ],
         [ '--visibility', '-v', GetoptLong::REQUIRED_ARGUMENT ],
+        [ '--force-digest', GetoptLong::NO_ARGUMENT],
+        [ '--progress', GetoptLong::NO_ARGUMENT],
+        [ '--no-progress', GetoptLong::NO_ARGUMENT],
+        [ '--bwlimit', GetoptLong::REQUIRED_ARGUMENT ],
       )
       opts.each do |opt, arg|
         case opt
@@ -42,12 +50,21 @@ module OSDN; module CLI; module Command
         when '--release'
           arg.empty? or
             @target_release = arg
+        when '--force-digest'
+          @force_digest = true
         when '--visibility'
           unless %w(public private hidden).member?(arg)
             logger.fatal "Invalid visibility status: #{arg}"
             exit
           end
           @visibility = arg
+        when '--progress'
+          @show_progress = true
+        when '--no-progress'
+          @show_progress = false
+        when '--bwlimit'
+          arg.to_i != 0 and
+            OSDN::CLI._rate_limit = arg.to_i * 1024
         end
       end
     end
@@ -65,21 +82,43 @@ module OSDN; module CLI; module Command
     end
 
     def create
-      filename = ARGV.shift
-      if !filename
+      if ARGV.empty? || ARGV.first == ""
         logger.fatal "Target filename is missing."
         help
         return
       end
+
+      ARGV.each do |f|
+        create_one(f)
+      end
+    end
+
+    def create_one(filename)
       file = Pathname('.') + filename
       logger.debug "Calculating digest for #{file}..."
-      digests = {
-        sha256: hexdigest(Digest::SHA256, file),
-        sha1:   hexdigest(Digest::SHA1, file),
-        md5:    hexdigest(Digest::MD5, file),
-      }
+
+      vars = load_variables(file.dirname)
+      digests = nil
+      if !@force_digest && vars.local_file_info &&
+         vars.local_file_info[file.basename.to_s]
+        finfo = vars.local_file_info[file.basename.to_s]
+        if finfo[:size] == file.size && finfo.mtime == file.mtime
+          digests = vars.local_file_info[file.basename.to_s].digests
+        end
+      end
+
+      unless digests
+        logger.info "Calculating digest for #{file}..."
+        digests = {
+          sha256: hexdigest(Digest::SHA256, file),
+          sha1:   hexdigest(Digest::SHA1, file),
+          md5:    hexdigest(Digest::MD5, file),
+        }
+        update_variables file.dirname, {local_file_info: {file.basename.to_s => {digests: digests, mtime: file.mtime, size: file.size}}}
+      end
+
       fio = file.open
-      logger.level <= Logger::INFO and
+      logger.level <= Logger::INFO && @show_progress != false || @show_progress and
         OSDN::CLI._show_progress = true
       logger.info "Starting upload #{file}..."
       f = api.create_release_file target_proj, target_package, target_release, fio, visibility: @visibility