yggdrasil 0.0.0 → 0.0.1

data/lib/yggdrasil/update.rb

@@ -0,0 +1,62 @@
+class Yggdrasil
+
+  # @param [Array] args
+  def update(args)
+    target_paths, options = parse_options(args,
+                                  {'--username'=>:username, '--password'=>:password,
+                                   '-r'=>:revision, '--revision'=>:revision,
+                                   '--non-interactive'=>:non_interactive?})
+    options = input_user_pass(options)
+    sync_mirror options
+
+    updates = Array.new
+    FileUtils.cd @mirror_dir do
+      out = system3("#@svn status -qu --depth infinity --no-auth-cache --non-interactive" +
+                      " --username '#{options[:username]}' --password '#{options[:password]}'")
+      out.split(/\n/).each do |line|
+        updates << $1 if /^.*\*.*\s(\S+)\s*$/ =~ line
+      end
+    end
+
+    matched_updates = select_updates(updates, target_paths)
+
+    confirmed_updates = confirm_updates(matched_updates,options) do |relative_path|
+      FileUtils.cd @mirror_dir do
+        cmd = "#@svn diff"
+        cmd += " --no-auth-cache --non-interactive"
+        cmd += " --username #{options[:username]} --password #{options[:password]}"
+        if options.has_key?(:revision)
+          cmd += " --old=#{relative_path} --new=#{relative_path}@#{options[:revision]}"
+        else
+          cmd += " --old=#{relative_path} --new=#{relative_path}@HEAD"
+        end
+        puts system3(cmd)
+      end
+    end
+    # res == 'Y' or --non-interactive
+
+    return unless confirmed_updates
+    return if confirmed_updates == 0 # no files to update
+
+    cmd_arg = "#@svn update --no-auth-cache --non-interactive"
+    cmd_arg += " --username #{options[:username]} --password #{options[:password]}"
+    if options.has_key?(:revision)
+      cmd_arg += " -r #{options[:revision]}"
+    else
+      cmd_arg += " -r HEAD"
+    end
+    cmd_arg += ' ' + confirmed_updates.join(' ')
+    FileUtils.cd @mirror_dir do
+      puts system3(cmd_arg)
+
+      # reflect mirror to real file
+      confirmed_updates.each do |update_file|
+        if File.exist?(update_file)
+          FileUtils.copy_file @mirror_dir+'/'+update_file, '/'+update_file
+        else
+          system3 "rm -rf /#{update_file}"
+        end
+      end
+    end
+  end
+end

data/lib/yggdrasil/version.rb

@@ -1,3 +1,14 @@
-module Yggdrasil
-  VERSION = "0.0.0"
+class Yggdrasil
+  VERSION = "0.0.1"
+  CMD = File::basename($0)
+
+  def Yggdrasil.version
+    puts <<"EOS"
+#{CMD}, version #{VERSION}
+
+Copyright (C) 2012-2013 Tomohisa Kusukawa.
+Yggdrasil is open source software, see https://github.com/tkusukawa/yggdrasil/
+
+EOS
+  end
 end

data/lib/yggdrasil.rb

@@ -1,5 +1,245 @@
+require 'fileutils'
+require "open3"
 require "yggdrasil/version"
+require "yggdrasil/help"
+require "yggdrasil/init"
+require "yggdrasil/add"
+require "yggdrasil/commit"
+require "yggdrasil/cleanup"
+require "yggdrasil/diff"
+require "yggdrasil/list"
+require "yggdrasil/log"
+require "yggdrasil/status"
+require "yggdrasil/update"
+require "yggdrasil/revert"
 
-module Yggdrasil
-  # Your code goes here...
+class Yggdrasil
+
+  def Yggdrasil.command(args, input = nil)
+    $stdin = StringIO.new(input) if input != nil
+    ENV['LANG'] = 'en_US.UTF-8'
+
+    if args.size == 0
+      Yggdrasil::help([])
+      return
+    end
+    case args[0]
+      when 'add'
+        new.add(args[1..-1])
+      when 'cleanup'
+        new.cleanup(args[1..-1])
+      when 'commit', 'ci'
+        new.commit(args[1..-1])
+      when 'diff', 'di'
+        new.diff(args[1..-1])
+      when 'help', 'h', '?'
+        help(args[1..-1])
+      when 'init'
+        init(args[1..-1])
+      when 'list', 'ls'
+        new.list(args[1..-1])
+      when 'log'
+        new.log(args[1..-1])
+      when 'status', 'stat', 'st'
+        new.status(args[1..-1])
+      when 'revert'
+        new.revert(args[1..-1])
+      when 'update'
+        new.update(args[1..-1])
+      when 'version', '--version'
+        version
+      else
+        error "Unknown subcommand: '#{args[0]}'"
+    end
+  end
+
+  # @param [String] cmd
+  def Yggdrasil.system3(cmd, err_exit=true, stdin=nil)
+    if stdin.nil?
+      out,stat = Open3.capture2e cmd
+    else
+      out,stat = Open3.capture2e cmd, :stdin_data=>stdin
+    end
+    unless stat.success?
+      return nil unless err_exit
+      $stderr.puts "#{CMD} error: command failure: #{cmd}"
+      $stderr.puts "command output:"
+      $stderr.puts out
+      exit stat.exitstatus
+    end
+    return out
+  end
+
+  protected
+  # @param [String] msg
+  def Yggdrasil.error(msg)
+    puts "#{CMD} error: #{msg}"
+    puts
+    exit 1
+  end
+
+  def Yggdrasil.parse_options(args, valid_params)
+    options = Hash.new
+    pos = 0
+    while args.size > pos
+      if valid_params.has_key?(args[pos])
+        option_note = args[pos]
+        option_key = valid_params[option_note]
+        args = args[0...pos]+args[pos+1..-1]
+        if option_key.to_s[-1] == '?'
+          options[option_key] = true
+        else
+          error "Not enough arguments provided: #{option_note}" unless args.size > pos
+          option_value = args[pos].dup
+          args = args[0...pos]+args[pos+1..-1]
+          options[option_key] = option_value
+        end
+        next
+      end
+      pos += 1
+    end
+    return args, options
+  end
+
+  def Yggdrasil.input_user_pass(options)
+    until options.has_key?(:username) do
+      print "Input svn username: "
+      input = $stdin.gets
+      options[:username] = input.chomp
+    end
+    until options.has_key?(:password) do
+      print "Input svn password: "
+      #input = `sh -c 'read -s hoge;echo $hoge'`
+      system3 'stty -echo', false
+      input = $stdin.gets
+      system3 'stty echo', false
+      options[:password] = input.chomp
+    end
+    return options
+  end
+
+  def initialize
+
+    @config = read_config
+    ENV["PATH"] = @config[:path]
+    @svn = @config[:svn]
+    @repo = @config[:repo]
+    @current_dir = `readlink -f .`.chomp
+    @mirror_dir = ENV["HOME"]+"/.yggdrasil/mirror"
+  end
+
+  # load config value from config file
+  def read_config
+    @config=Hash.new
+    begin
+      config_file = open("#{ENV['HOME']}/.yggdrasil/config")
+    rescue
+      puts "#{CMD} error: can not open config file: #{ENV['HOME']}/.yggdrasil/config"
+      exit 1
+    end
+    l = 0
+    while (line = config_file.gets)
+      l += 1
+      next if /^\s*#.*$/ =~ line  # comment line
+      if /^\s*(\S+)\s*=\s*(\S+).*$/ =~ line
+        @config[$1.to_sym] = $2
+      else
+        puts "#{CMD} error: syntax error. :#{ENV['HOME']}/.yggdrasil/config(#{l})"
+        exit 1
+      end
+    end
+    config_file.close
+    @config
+  end
+
+  def sync_mirror(options)
+    updates = Array.new
+    FileUtils.cd @mirror_dir do
+      out = system3("#@svn ls #@repo --depth infinity --no-auth-cache --non-interactive" +
+                           " --username '#{options[:username]}' --password '#{options[:password]}'")
+      files = out.split(/\n/)
+      out = system3("#@svn status -q --depth infinity --no-auth-cache --non-interactive" +
+                           " --username '#{options[:username]}' --password '#{options[:password]}'")
+      out.split(/\n/).each do |line|
+        files << $1 if /^.*\s(\S+)\s*$/ =~ line
+      end
+      files.sort!
+      files.uniq!
+      files.each do |file|
+        if !File.exist?('/'+file)
+          system3 "#@svn delete #{file} --force" +
+                      " --no-auth-cache --non-interactive"
+        elsif File.file?('/'+file)
+          if !File.exist?(@mirror_dir+'/'+file)
+            system3 "#@svn revert --no-auth-cache --non-interactive #{file}"
+          end
+          FileUtils.copy_file '/'+file, @mirror_dir+'/'+file
+        end
+      end
+      out = system3("#@svn status -q --depth infinity --no-auth-cache --non-interactive" +
+                        " --username '#{options[:username]}' --password '#{options[:password]}'")
+      out.split(/\n/).each do |line|
+        updates << $1 if /^.*\s(\S+)\s*$/ =~ line
+      end
+    end
+    updates
+  end
+
+  def select_updates(updates, target_paths)
+
+    target_relatives = Array.new
+    if  target_paths.size == 0
+      target_relatives << @current_dir.sub(%r{^/},'')
+    else
+      target_paths.each do |path|
+        if %r{^/} =~ path
+          target_relatives << path.sub(%r{^/},'') # cut first '/'
+        else
+          target_relatives << @current_dir.sub(%r{^/},'') + '/' + path
+        end
+      end
+    end
+
+    # search updated files in the specified dir
+    cond = '^'+target_relatives.join('|^') # make reg exp
+    matched_updates = Array.new
+    updates.each do |update|
+      matched_updates << update if update.match(cond)
+    end
+
+    # search parent updates of matched updates
+    parents = Array.new
+    updates.each do |update|
+      matched_updates.each do |matched_update|
+        parents << update if matched_update.match("^#{update}/")
+      end
+    end
+    matched_updates += parents
+    matched_updates.sort.uniq
+  end
+
+  def confirm_updates(updates, options)
+    until options.has_key?(:non_interactive?)
+      puts
+      (0...updates.size).each do |i|
+        puts "#{i}:#{updates[i]}"
+      end
+      print "OK? [Y|n|<num to diff>]:"
+      res = $stdin.gets
+      return nil unless res
+      res.chomp!
+      return nil if res == 'n'
+      break if res == 'Y'
+      next unless updates[res.to_i]
+      if /^\d+$/ =~ res
+        yield updates[res.to_i]
+      end
+    end
+    # res == 'Y'
+    updates
+  end
+
+  def method_missing(action, *args)
+    Yggdrasil.__send__ action, *args
+  end
 end

data/spec/add_spec.rb

@@ -0,0 +1,25 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe Yggdrasil, "add" do
+  it '-------- add' do
+    puts '-------- add'
+    prepare_environment
+    init_yggdrasil
+  end
+
+  it 'should warn: add non-exist files' do
+    puts '---- should warn: add non-exist files'
+    out = catch_stdout{Yggdrasil.command %w{add hoge}}
+    out.should == "no such file: #{`readlink -f hoge`}"
+
+    out = catch_stdout{Yggdrasil.command %w{add /etc/hoge}}
+    out.should == "no such file: /etc/hoge\n"
+  end
+
+  it 'should success: add exist files' do
+    puts '---- should success: add exist files'
+    Yggdrasil.command %w{add Gemfile /etc/fstab /etc/fstab}
+    File.exist?("/tmp/yggdrasil-test/.yggdrasil/mirror#{`readlink -f Gemfile`.chomp}").should be_true
+    File.exist?("/tmp/yggdrasil-test/.yggdrasil/mirror#{`readlink -f /etc/fstab`.chomp}").should be_true
+  end
+end

data/spec/cleanup_spec.rb

@@ -0,0 +1,23 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe Yggdrasil, "cleanup" do
+  it '-------- cleanup' do
+    puts '-------- cleanup'
+    prepare_environment
+    init_yggdrasil
+  end
+
+  it 'should success cleanup' do
+    puts "---- should success cleanup"
+    puts "-- rm .svn"
+    `rm -rf /tmp/yggdrasil-test/.yggdrasil/mirror/.svn`
+
+    puts "-- cleanup"
+    Yggdrasil.command %w{cleanup --username hoge --password foo}
+
+    puts "-- check .svn"
+    res = File.exist?("/tmp/yggdrasil-test/.yggdrasil/mirror/.svn")
+    p res
+    res.should == true
+  end
+end

data/spec/commit_spec.rb

@@ -0,0 +1,129 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe Yggdrasil, "commit" do
+  it '-------- commit' do
+    puts '-------- commit'
+    prepare_environment
+
+    puts '-- init'
+    Yggdrasil.command %w{init} +
+                          %w{--repo svn://localhost/tmp/yggdrasil-test/svn-repo/mng-repo/host-name/} +
+                          %w{--username hoge --password foo}
+  end
+
+  it 'should commit added files' do
+    puts '---- should commit added files'
+    `echo hoge > /tmp/yggdrasil-test/A`
+    `echo foo > /tmp/yggdrasil-test/B`
+    FileUtils.cd "/tmp/yggdrasil-test" do
+      puts '-- add'
+      Yggdrasil.command %w{add A /tmp/yggdrasil-test/B}
+    end
+
+    puts "-- commit"
+    FileUtils.cd '/tmp/yggdrasil-test' do
+      Yggdrasil.command %w{commit --username hoge --password foo},
+                        "0\nY\nadd A and B\n"
+    end
+
+    puts "\n-- check committed file 'tmp/yggdrasil-test/A'"
+    res = `svn cat file:///tmp/yggdrasil-test/svn-repo/mng-repo/host-name/tmp/yggdrasil-test/A`
+    puts res
+    res.should == "hoge\n"
+
+    puts "\n-- check committed file 'tmp/yggdrasil-test/B'"
+    res = `svn cat file:///tmp/yggdrasil-test/svn-repo/mng-repo/host-name/tmp/yggdrasil-test/B`
+    puts res
+    res.should == "foo\n"
+  end
+
+  it 'should commit modified file' do
+    puts "---- should commit modified file"
+    puts "-- modify"
+    `echo hoge >> /tmp/yggdrasil-test/A`
+
+    puts "-- commit"
+    Yggdrasil.command %w{commit / --username hoge --password foo},
+                      "0\nY\nmodify A\n"
+
+    puts "\n-- check committed file 'tmp/yggdrasil-test/A'"
+    res = `svn cat file:///tmp/yggdrasil-test/svn-repo/mng-repo/host-name/tmp/yggdrasil-test/A`
+    puts res
+    res.should == "hoge\nhoge\n"
+  end
+
+  it 'should accept password interactive' do
+    puts "---- should accept password interactive"
+    `echo A >> /tmp/yggdrasil-test/A`
+
+    Yggdrasil.command %w{commit /tmp --username hoge},
+                      "foo\nY\nmodify A\n" # interactive input: password,Y/n, commit message
+
+    puts "\n-- check committed file 'tmp/yggdrasil-test/A'"
+    res = `svn cat file:///tmp/yggdrasil-test/svn-repo/mng-repo/host-name/tmp/yggdrasil-test/A`
+    puts res
+    res.should == "hoge\nhoge\nA\n"
+  end
+
+  it 'should commit specified file only' do
+    puts "---- should commit specified file only"
+    `echo A >> /tmp/yggdrasil-test/A`
+    `echo B >> /tmp/yggdrasil-test/B`
+
+    Yggdrasil.command %w{commit
+--username hoge --password foo -m modify /tmp/yggdrasil-test/B},
+                      "0\nY\n"
+
+    puts "\n-- check committed file 'tmp/yggdrasil-test/B'"
+    res = `svn cat file:///tmp/yggdrasil-test/svn-repo/mng-repo/host-name/tmp/yggdrasil-test/B`
+    puts res
+    res.should == "foo\nB\n"
+  end
+
+  it 'should not commit deleted file' do
+    puts "---- should not commit deleted file"
+    `rm -f /tmp/yggdrasil-test/A`
+
+    Yggdrasil.command %w{commit --username hoge --password foo -m delete},
+                      "0\nn\n"
+    puts "\n-- check file exists on repo"
+    res = `svn ls file:///tmp/yggdrasil-test/svn-repo/mng-repo/host-name/tmp/yggdrasil-test`
+    puts res
+    res.should == "A\nB\n"
+  end
+
+  it 'should commit deleted file' do
+    puts "---- should commit deleted file"
+    `echo hoge > /tmp/yggdrasil-test/A`
+    `rm -f /tmp/yggdrasil-test/B`
+
+    Yggdrasil.command %w{commit  -m delete /tmp/yggdrasil-test} +
+                          %w{--username hoge --password foo},
+                      "0\n1\nY\n"
+
+    puts "\n-- check committed delete file"
+    res = `svn ls file:///tmp/yggdrasil-test/svn-repo/mng-repo/host-name/tmp/yggdrasil-test`
+    puts res
+    res.should == "A\n"
+  end
+
+  it 'should commit all files at once' do
+    puts "---- should revert all files at once"
+
+    `echo HOGE >> /tmp/yggdrasil-test/A`
+    `rm -f /tmp/yggdrasil-test/B`
+    `mkdir /tmp/yggdrasil-test/c`
+    `echo bar > /tmp/yggdrasil-test/c/C`
+    Yggdrasil.command %w{add /tmp/yggdrasil-test/c/C}
+
+    Yggdrasil.command %w{commit -m delete /tmp/yggdrasil-test/c/C} +
+                          %w{--username hoge --password foo},
+                      "0\n1\nY\n"
+
+    puts "\n-- check committed delete file"
+    res = `svn ls file:///tmp/yggdrasil-test/svn-repo/mng-repo/host-name/tmp/yggdrasil-test`
+    puts res
+    res.should == "A\nc/\n"
+  end
+
+end

data/spec/diff_spec.rb

@@ -0,0 +1,104 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe Yggdrasil, "diff" do
+  it '-------- diff' do
+    puts '-------- diff'
+    prepare_environment
+    init_yggdrasil
+
+    # modify and not commit yet
+    `echo HOGE >> /tmp/yggdrasil-test/A`
+    `echo FOO >> /tmp/yggdrasil-test/B`
+  end
+
+  it 'should success diff (absolute/relative path)' do
+    puts "---- should success diff (absolute/relative path)"
+    out = catch_stdout do
+      FileUtils.cd "/tmp/yggdrasil-test" do
+        Yggdrasil.command(%w{diff /tmp/yggdrasil-test/A B --username hoge --password foo})
+      end
+    end
+    out.should == <<"EOS"
+Index: tmp/yggdrasil-test/A
+===================================================================
+--- tmp/yggdrasil-test/A	(revision 3)
++++ tmp/yggdrasil-test/A	(working copy)
+@@ -1,2 +1,3 @@
+ hoge
+ hoge
++HOGE
+Index: tmp/yggdrasil-test/B
+===================================================================
+--- tmp/yggdrasil-test/B	(revision 3)
++++ tmp/yggdrasil-test/B	(working copy)
+@@ -1,2 +1,3 @@
+ foo
+ foo
++FOO
+EOS
+  end
+
+  it 'should success (no path)' do
+    puts "---- should success (no path)"
+    out = catch_stdout do
+      FileUtils.cd "/tmp/yggdrasil-test" do
+        Yggdrasil.command %w{diff --username hoge --password foo}
+      end
+    end
+    out.should == <<"EOS"
+Index: tmp/yggdrasil-test/A
+===================================================================
+--- tmp/yggdrasil-test/A\t(revision 3)
++++ tmp/yggdrasil-test/A\t(working copy)
+@@ -1,2 +1,3 @@
+ hoge
+ hoge
++HOGE
+Index: tmp/yggdrasil-test/B
+===================================================================
+--- tmp/yggdrasil-test/B\t(revision 3)
++++ tmp/yggdrasil-test/B\t(working copy)
+@@ -1,2 +1,3 @@
+ foo
+ foo
++FOO
+EOS
+  end
+
+  it 'should success (-r)' do
+    puts "---- should success (-r)"
+    out = catch_stdout do
+      FileUtils.cd "/tmp/yggdrasil-test" do
+        Yggdrasil.command %w{diff -r 2:3 A --username hoge --password foo}
+      end
+    end
+    out.should == <<"EOS"
+Index: tmp/yggdrasil-test/A
+===================================================================
+--- tmp/yggdrasil-test/A	(revision 2)
++++ tmp/yggdrasil-test/A	(revision 3)
+@@ -1 +1,2 @@
+ hoge
++hoge
+EOS
+  end
+
+  it 'should success (--revision)' do
+    puts "---- should success (--revision)"
+    out = catch_stdout do
+      FileUtils.cd "/tmp/yggdrasil-test" do
+        Yggdrasil.command %w{diff --revision 3 A --username hoge --password foo}
+      end
+    end
+    out.should == <<"EOS"
+Index: tmp/yggdrasil-test/A
+===================================================================
+--- tmp/yggdrasil-test/A	(revision 3)
++++ tmp/yggdrasil-test/A	(working copy)
+@@ -1,2 +1,3 @@
+ hoge
+ hoge
++HOGE
+EOS
+  end
+end