confgit 0.0.3

data/lib/confgit/repo.rb

@@ -0,0 +1,614 @@
+# coding: UTF-8
+
+require 'fileutils'
+require 'pathname'
+require 'etc'
+require 'shellwords'
+require 'open3'
+
+require 'rubygems'
+require 'json'
+
+require 'confgit/with_color'
+
+
+module Confgit
+
+class Repo
+	ROOT_KEY = 'confgit.root'
+
+	include WithColor
+
+	def initialize(path = '~/.etc/confgit')
+		@base_path = File.expand_path(path)
+		@repos_path = File.join(@base_path, 'repos')
+
+		FileUtils.mkpath(@repos_path)
+
+		@config = read_config(File.join(@base_path, 'confgit.conf'))
+		@repo_path = File.expand_path('current', @repos_path)
+
+		valid_repo unless File.symlink?(@repo_path)
+	end
+
+	# ホスト名
+	def hostname
+		`hostname`.chomp
+	end
+
+	# カレントリポジトリがない場合の処理
+	def valid_repo
+		repo = nil
+
+		repo_each { |file, is_current|
+			repo = file
+			break
+		}
+
+		chrepo(repo || hostname)
+	end
+
+	# リポジトリの変更
+	def chrepo(repo)
+		Dir.chdir(@repos_path) { |path|
+			begin
+				if File.symlink?('current')
+					return if File.readlink('current') == repo
+					File.unlink('current')
+				end
+
+				unless File.exist?(repo)
+					FileUtils.mkpath(repo)
+
+					Dir.chdir(repo) { |path|
+						begin
+							out, err, status = Open3.capture3('git', 'init')
+							$stderr.puts err unless err.empty?
+						rescue => e
+							FileUtils.remove_entry_secure(repo)
+							abort e.to_s
+						end
+					}
+				end
+
+				File.symlink(repo, 'current') 
+			rescue => e
+				abort e.to_s
+			end
+		}
+	end
+
+	# リポジトリの削除
+	def rmrepo(repo, force = false)
+		Dir.chdir(@repos_path) { |path|
+			begin
+				if File.symlink?('current') && File.readlink('current') == repo
+					abort "'#{repo}' is current repository!" unless force
+					File.unlink('current')
+				end
+
+				FileUtils.remove_entry_secure(repo)
+
+				valid_repo unless File.symlink?('current')
+			rescue => e
+				abort e.to_s
+			end
+		}
+	end
+
+	# 設定の初期値
+	def default_config
+		{}
+	end
+
+	# 設定の読込み
+	def read_config(file)
+		if File.exist?(file)
+			config = JSON.parse(File.read(file))
+			config = default_config.merge(config)
+		else
+			config = default_config
+			File.write(file, JSON.pretty_generate(config)+"\n")
+		end
+
+		return config
+	end
+
+	# 外部コマンドを定義する
+	def self.define_command(command, *opts)
+		define_method "confgit_#{command}" do |options, *args|
+			args = getargs(args)
+
+			Dir.chdir(@repo_path) { |path|
+				begin
+					args = opts + args
+					args.push(options)
+					system_(command, *args)
+				rescue => e
+					abort e.to_s
+				end
+			}
+		end
+	end
+
+	# メソッドがない場合
+	def method_missing(name, *args, &block)
+		if name.to_s =~ /^confgit_(.+)$/
+			options = args.shift
+			args = git_args(args).push(options)
+
+			command = $1.gsub(/_/, '-')
+			git(command, *args)
+
+#			abort "#{CMD} '#{$'}' is not a git command. See '#{CMD} --help'.\n"
+		else
+			super
+		end
+	end
+
+	# 引数を利用可能にする
+	def getargs(args, force = false)
+		args.collect { |x|
+			run = false
+
+			case x
+			when /^-/
+			when /\//
+				run = true
+			else
+				run = force
+			end
+
+			if run
+				repo = File.realpath(@repo_path)
+				path = File.join(repo, x)
+				x = Pathname(path).relative_path_from(Pathname(repo)).to_s
+			end
+
+			x
+		}
+	end
+
+	# 引数の最後が Hash ならオプションとして取出す
+	def arg_last_options(args)
+		if args.last && args.last.kind_of?(Hash)
+			args.pop
+		else
+			{}
+		end
+	end
+
+	# オプションに応じて外部呼出しを行う
+	def system_(command, *args)
+		options = arg_last_options(args)
+
+		if options[:interactive] == false
+			out, err, status = Open3.capture3(command, *args)
+
+			$stdout.print out unless out.empty?
+			$stderr.print err unless err.empty?
+
+			status
+		elsif options[:capture]
+			Open3.capture3(command, *args)
+		else
+			system(command, *args)
+		end
+	end
+
+	# git を呼出す
+	def git(*args)
+		Dir.chdir(@repo_path) { |path|
+			begin
+				system_('git', *args)
+			rescue => e
+				abort e.to_s
+			end
+		}
+	end
+
+	# git コマンドの引数を生成する
+	def git_args(args)
+		args.collect { |item|
+			item = $' if item.kind_of?(String) && %r|^/| =~ item
+			item
+		}
+	end
+
+	# ルートのパスを取得する
+	def root
+		return @root if @root
+
+		# 表示
+		out, err, status = git('config', '--path', '--local', ROOT_KEY, :capture => true)
+		out.chomp!
+		out = '/' if out.empty?
+
+		@root = out
+	end
+
+	# ルートのパスを設定する
+	def root=(value)
+		if value && ! value.empty?
+			git('config', '--path', '--local', ROOT_KEY, value)
+		else
+			git('config', '--unset', '--local', ROOT_KEY)
+		end
+
+		@root = nil
+	end
+
+	# ルートからの相対パス
+	def relative_path(path)
+		root_path = Pathname.new(root)
+		Pathname.new(path).relative_path_from(root_path).to_s
+	end
+
+	# ファイルの hash値を求める
+	def hash_object(file)
+		path = File.expand_path(file)
+		open("| git hash-object \"#{path}\"") {|f|
+			return f.gets.chomp
+		}
+	end
+
+	# 確認プロンプトを表示する
+	def yes?(prompt, y = true)
+		yn = y ? 'Yn' : 'yN'
+		print "#{prompt} [#{yn}]: "
+
+		result = $stdin.gets.chomp
+
+		return y if result.empty?
+
+		if /^(y|yes)$/i =~ result
+			y = true
+		else
+			y = false
+		end
+
+		y
+	end
+
+	# ファイルのコピー（属性は維持する）
+	def filecopy(from, to, exiting = false)
+		begin
+			to_dir = File.dirname(to)
+			FileUtils.mkpath(to_dir)
+
+			if File.exist?(to) && ! File.writable_real?(to)
+				# 書込みできない場合は削除を試みる
+				File.unlink(to)
+			end
+
+			FileUtils.copy(from, to)
+			stat = File.stat(from)
+			File.utime(stat.atime, stat.mtime, to)
+			File.chmod(stat.mode, to)
+
+			return true
+		rescue => e
+			abort e.to_s if exiting
+			$stderr.puts e.to_s
+		end
+	end
+
+	# ディレクトリ内のファイルを繰返す
+	def dir_each(subdir = '.')
+		Dir.chdir(File.expand_path(subdir, @repo_path)) { |path|
+			Dir.foreach('.') { |file|
+				next if /^(\.git|\.$|\.\.$)/ =~ file
+	
+				yield(file)
+	
+				if File.directory?(file)
+					Dir.glob("#{file}/**/*", File::FNM_DOTMATCH) { |file|
+						if /(^|\/)(\.git|\.|\.\.)$/ !~ file
+							yield(file)
+						end
+					}
+				end
+			}
+		}
+	end
+
+	# git に管理されているファイルを繰返す
+	def git_each(*args)
+		args = getargs(args, true)
+		files = args.collect { |f| f.shellescape }
+
+		Dir.chdir(@repo_path) { |path|
+			open("| git ls-files --stage --full-name " + files.join(' ')) {|f|
+				while line = f.gets
+					mode, hash, stage, file = line.split
+
+#					file = line.chomp
+					next if /^\.git/ =~ file
+					next if File.directory?(file)
+
+					yield(file, hash)
+				end
+			}
+		}
+	end
+
+	# リポジトリを繰返す
+	def repo_each
+		Dir.chdir(@repos_path) { |path|
+			begin
+				current = File.expand_path(File.readlink('current'))
+			rescue
+			end
+
+			Dir.glob('*') { |file|
+				next if /^current$/ =~ file
+
+				if current && File.realpath(file) == current
+					is_current = true
+					current = nil
+				else
+					is_current = false
+				end
+
+				yield(file, is_current)
+			}
+
+			yield(File.readlink('current'), true) if current
+		}
+	end
+
+	# パスを展開する
+	def expand_path(path, dir = nil)
+		File.expand_path(path, dir).gsub(%r|^/private(/[^/]+)|) { |m|
+			begin
+				subdir = $1
+				m = subdir if File.realpath(subdir) == m
+			rescue
+			end
+
+			m
+		}
+	end
+
+	# オプションを取出す
+	def getopts(args)
+		options = []
+		args.each { |opt|
+			break unless /^-/ =~ opt
+			options << args.shift
+		}
+
+		options
+	end
+
+	# ファイルの更新チェック
+	def modfile?(from, to)
+		! File.exist?(to) || hash_object(from) != hash_object(to)
+	end
+
+	# ファイル属性を文字列にする
+	def mode2str(bits)
+		case bits & 0170000	# S_IFMT
+		when 0010000	# S_IFIFO	パイプ
+			mode = 'p'
+		when 0020000	# S_IFCHR	キャラクタ・デバイス
+			mode = 'c'
+		when 0040000	# S_IFDIR	ディレクトリ
+			mode = 'd'
+		when 0060000	# S_IFBLK	ブロック・デバイス
+			mode = 'b'
+		when 0100000	# S_IFREG	通常ファイル
+			mode = '-'
+		when 0120000	# S_IFLNK	シンボリックリンク
+			mode = 'l'
+		when 0140000	# S_IFSOCK	ソケット
+			mode = 's'
+		when 0160000	# S_IFWHT	BSD空白ファイル
+			mode = 'w'
+		end
+
+		mode += 'rwx'*3
+
+		(0..8).each { |i| 
+			mask = 1<<i
+			mode[-(i+1)] = '-' if (bits & mask) == 0
+		}
+
+		if (bits & 0001000) != 0	# S_ISVTX	スティッキービット
+			if mode[-1] == '-'
+				mode[-1] = 'T'
+			else
+				mode[-1] = 't'
+			end
+		end
+
+		mode
+	end
+
+	# コマンド
+
+	# カレントリポジトリの表示・変更
+	def confgit_repo(options, repo = nil)
+		if repo
+			# 変更
+			if options[:remove]
+				rmrepo(repo, options[:force])
+			else
+				chrepo(repo)
+			end
+		else
+			# 表示
+			repo_each { |file, is_current|
+				mark = is_current ? '*' : ' '
+				print "#{mark} #{file}\n"
+			}
+		end
+	end
+
+	# ルートの表示・変更
+	def confgit_root(options, value = nil)
+		if options[:remove]
+			# 削除
+			self.root = nil
+		elsif value
+			# 変更
+			self.root = value
+		else
+			# 表示
+			puts root
+		end
+	end
+
+	# リポジトリの初期化
+	def confgit_init(options)
+		FileUtils.mkpath(@repo_path)
+		git('init')
+	end
+
+	# ファイルを管理対象に追加
+	def confgit_add(options, *files)
+		confgit_init unless File.exist?(@repo_path)
+		repo = File.realpath(@repo_path)
+
+		files.each { |path|
+			path = expand_path(path)
+
+			if relative_path(path) =~ /^[.]{2}/
+				$stderr.puts "'#{path}' is outside directory"
+				next
+			end
+
+			if File.directory?(path)
+				dir_each(path) { |file|
+					next if File.directory?(file)
+	
+					from = File.join(path, file)
+					to = File.join(repo, relative_path(from))
+
+					if filecopy(from, to)
+						git('add', to)
+					end
+				}
+			else
+				from = path
+				to = File.join(repo, relative_path(from))
+
+				if filecopy(from, to)
+					git('add', to)
+				end
+			end
+		}
+	end
+
+	# ファイルを管理対象から削除
+	def confgit_rm(options, *args)
+		return unless File.exist?(@repo_path)
+
+		options = getopts(args)
+		repo = File.realpath(@repo_path)
+
+		files = args.collect { |from|
+			File.join(repo, relative_path(expand_path(from)))
+		}
+
+		git('rm', *(options + files), :interactive => false)
+	end
+
+	# バックアップする
+	def confgit_backup(options, *args)
+		git_each(*args) { |file, hash|
+			next if File.directory?(file)
+
+			from = File.join(root, file)
+			to = File.join(@repo_path, file)
+
+			unless File.exist?(from)
+				with_color(:fg_red) { print "[?] #{file}" }
+				puts
+				next
+			end
+
+			if options[:force] || modfile?(from, to)
+				with_color(:fg_blue) { print "--> #{file}" }
+				write = options[:yes]
+
+				if write == nil
+					# 書込みが決定していない場合
+					write = yes?(nil, false)
+				else
+					puts
+				end
+
+				filecopy(from, to) if write
+			end
+		}
+
+		git('status', :interactive => false)
+	end
+
+	# リストアする
+	def confgit_restore(options, *args)
+		git_each(*args) { |file, hash|
+			next if File.directory?(file)
+
+			from = File.join(@repo_path, file)
+			to = File.join(root, file)
+
+			unless File.exist?(from)
+				with_color(:fg_red) { print "[?] #{file}" }
+				puts
+				next
+			end
+
+			if options[:force] || modfile?(from, to)
+				color = File.writable_real?(to) ? :fg_blue : :fg_magenta
+				with_color(color) { print "<-- #{file}" }
+				write = options[:yes]
+
+				if write == nil
+					# 書込みが決定していない場合
+					write = yes?(nil, false)
+				else
+					puts
+				end
+
+				filecopy(from, to) if write
+			end
+		}
+	end
+
+	# 一覧表示する
+	def confgit_list(options, *args)
+		git_each(*args) { |file, hash|
+			next if File.directory?(file)
+
+			from = File.join(root, file)
+			to = File.join(@repo_path, file)
+
+			if File.exist?(from)
+				stat = File.stat(from)
+				mode = options[:octal] ? stat.mode.to_s(8) : mode2str(stat.mode)
+				user = Etc.getpwuid(stat.uid).name
+				group = Etc.getgrgid(stat.gid).name
+			else
+				mode = ' ' * (options[:octal] ? 6 : 10)
+				user = '-'
+				group = '-'
+			end
+
+			print "#{mode}\t#{user}\t#{group}\t#{from}\n"
+		}
+	end
+
+	# リポジトリのパスを表示
+	def confgit_path(options, subdir = '.')
+		path = File.realpath(File.expand_path(subdir, @repo_path))
+		print path, "\n"
+	end
+
+	# 外部コマンド
+	define_command('tree', '-I', '.git')	# tree表示する
+	define_command('tig')					# tigで表示する
+end
+
+end

data/lib/confgit/version.rb

@@ -0,0 +1,4 @@
+module Confgit
+	LONG_VERSION = File.read(File.expand_path('../../../VERSION', __FILE__)).chomp
+	VERSION = LONG_VERSION.gsub(/-.*$/, '')
+end

data/lib/confgit/with_color.rb

@@ -0,0 +1,54 @@
+# coding: UTF-8
+
+module Confgit
+
+module WithColor
+	ESC_CODES = {
+		# Text attributes
+		:clear		=> 0,
+		:bold		=> 1,
+		:underscore => 4,
+		:blink		=> 5,
+		:reverse	=> 7,
+		:concealed	=> 8,
+
+		# Foreground colors
+		:fg_black	=> 30,
+		:fg_red 	=> 31,
+		:fg_green	=> 32,
+		:fg_yellow	=> 33,
+		:fg_blue	=> 34,
+		:fg_magenta	=> 35,
+		:fg_Cyan	=> 36,
+		:fg_White	=> 37,
+
+		# Background colors
+		:bg_black	=> 40,
+		:bg_red 	=> 41,
+		:bg_green	=> 42,
+		:bg_yellow	=> 43,
+		:bg_blue	=> 44,
+		:bg_magenta	=> 45,
+		:bg_Cyan	=> 46,
+		:bg_White	=> 47,
+	}
+
+	# エスケープシーケンスをセットする
+	def set_color(*colors)
+		colors.each { |color|
+			print "\e[", ESC_CODES[color], "m"
+		}
+	end
+
+	# カラー表示する
+	def with_color(*colors)
+		begin
+			set_color(*colors)
+			yield
+		ensure
+			set_color(0)
+		end
+	end
+end
+
+end

data/lib/confgit.rb

@@ -0,0 +1,10 @@
+require "confgit/version"
+require "confgit/repo"
+require "confgit/cli"
+
+
+module Confgit
+	def self.run(argv = ARGV, options = {})
+		CLI.run(argv, options)
+	end
+end