Мартина обнови решението на 22.11.2015 03:02 (преди над 8 години)
+require 'digest'
+require 'pp'
+
+class Transaction
+ attr_reader :message, :result
+
+ def initialize(message, return_code, result = nil)
+ @message = message
+ @status = return_code
+ @result = result
+ end
+
+ def success?
+ @status == true
+ end
+
+ def error?
+ @status == false
+ end
+end
+
+class CommitObject
+ attr_reader :name, :object
+ attr_accessor :for_deletion
+
+ def initialize(name, object = nil)
+ @name = name
+ @object = object
+ @for_deletion = false
+ end
+
+ def for_deletion?
+ @for_deletion
+ end
+
+ def ==(other)
+ @name == other.name
+ end
+end
+
+class CommitEntry
+ attr_reader :hash, :date, :message, :objects
+
+ def initialize(message, objects)
+ @message = message
+ @date = Time.now
+ @hash = Digest::SHA1.hexdigest("#{@date}#{@message}")
+ @objects = objects
+ end
+
+ def ==(other)
+ @hash == other.hash
+ end
+
+ def to_s
+ "Commit #{@hash}\nDate: #{@date.strftime('%a %b %-d %H:%M %Y %z')}\n\n\t" \
+ "#{@message}"
+ end
+end
+
+class Branch
+ attr_reader :name
+ attr_accessor :staged, :commits, :items
+
+ def initialize(branch_name, object_store, branch_items = [], commits = [])
+ @name = branch_name
+ @object_store = object_store
+ @items = branch_items
+ @staged = []
+ @commits = commits
+ end
+
+ def ==(other)
+ @name == other.name
+ end
+
+ def create(branch_name)
+ if @object_store.repository.any? { |branch| branch.name == branch_name }
+ Transaction.new("Branch #{branch_name} already exists.", false)
+ else
+ @object_store.repository.push(Branch.new(branch_name, @object_store,
+ items.clone, commits.clone))
+ Transaction.new("Created branch #{branch_name}.", true,
+ @object_store.repository.last)
+ end
+ end
+
+ def get_branch_index(branch_name)
+ @object_store.repository.index { |branch| branch.name == branch_name }
+ end
+
+ def checkout(branch_name)
+ branch_index = get_branch_index(branch_name)
+ if branch_index.nil?
+ Transaction.new("Branch #{branch_name} does not exist.", false)
+ else
+ @object_store.current_branch = branch_name
+ Transaction.new("Switched to branch #{branch_name}.", true,
+ @object_store.repository[branch_index])
+ end
+ end
+
+ def remove(branch_name)
+ if @object_store.current_branch == branch_name
+ return Transaction.new('Cannot remove current branch.', false)
+ end
+ branch_index = get_branch_index(branch_name)
+ if branch_index.nil?
+ Transaction.new("Branch #{branch_name} does not exist.", false)
+ else
+ @object_store.repository.delete_at(branch_index)
+ Transaction.new("Removed branch #{branch_name}.", true)
+ end
+ end
+
+ def list
+ branch_names = @object_store.repository.collect(&:name)
+ branch_names.sort!.map! do |name|
+ name == @object_store.current_branch ? "* #{name}" : " #{name}"
+ end
+ Transaction.new(branch_names.join('\n'), true)
+ end
+end
+
+class ObjectStore
+ attr_reader :repository
+ attr_accessor :current_branch
+
+ def initialize
+ @repository = [Branch.new('master', self)]
+ @current_branch = 'master'
+ end
+
+ def self.init(&block)
+ object_store = ObjectStore.new
+ object_store.instance_eval &block if block_given?
+ object_store
+ end
+
+ def head
+ if branch.commits.empty?
+ Transaction.new("Branch #{branch.name} does not have any commits yet.",
+ false)
+ else
+ Transaction.new("#{branch.commits.first.message}", true,
+ branch.commits.first)
+ end
+ end
+
+ def add(name, object)
+ branch.remove(name) if branch.staged.include?(CommitObject.new(name))
+ branch.staged.push(CommitObject.new(name, object))
+ Transaction.new("Added #{name} to stage.", true, object)
+ end
+
+ def log
+ if branch.commits.empty?
+ message = "Branch #{@current_branch} does not have any commits yet."
+ Transaction.new(message, false)
+ else
+ Transaction.new(branch.commits.join("\n\n"), true, branch.commits)
+ end
+ end
+
+ # cleans up staged array by either removing or adding items to the repository
+ def sweep(object)
+ if object.for_deletion?
+ branch.items.delete(object)
+ else
+ branch.items.push(object)
+ end
+ end
+
+ def commit(message)
+ if branch.staged.empty?
+ return Transaction.new('Nothing to commit, working directory clean.',
+ false)
+ end
+ # add the staged items to the repository
+ branch.staged.each do |object|
+ sweep(object)
+ end
+ branch.commits.insert(0, CommitEntry.new(message, branch.staged.clone))
+ length = branch.staged.length
+ branch.staged.clear
+ Transaction.new("#{message}\n\t#{length} objects changed", true,
+ branch.commits.first)
+ end
+
+ def remove(name)
+ for_deletion = object_exists?(name)
+ if for_deletion.equal? nil
+ Transaction.new("Object #{name} is not committed.", false)
+ else
+ for_deletion.for_deletion = true
+ branch.staged.push(for_deletion)
+ Transaction.new("Added #{name} for removal.", true, for_deletion.object)
+ end
+ end
+
+ def checkout(commit_hash)
+ commit_index = branch.commits.index { |entry| entry.hash == commit_hash }
+ if commit_index.nil?
+ return Transaction.new("Commit #{commit_hash} does not exist.", false)
+ end
+ 0.upto(commit_index - 1) do |index|
+ branch.commits[index].objects.each do |commit_object|
+ sweep(commit_object)
+ end
+ end
+ branch.commits = branch.commits.take(commit_index + 1).reverse
+ Transaction.new("HEAD is now at #{branch.commits.first.hash}.", true,
+ branch.commits.first)
+ end
+
+ def object_exists?(name)
+ branch.items.find { |commit_object| commit_object.name == name }
+ end
+
+ def branch
+ @repository.find { |branch| branch.name == @current_branch }
+ end
+
+ def get(object_name)
+ commit_object = branch.items.find { |item| item.name == object_name }
+ if commit_object.nil?
+ Transaction.new("Object #{object_name} is not committed.", false)
+ else
+ Transaction.new("Found object #{object_name}.", true,
+ commit_object.object)
+ end
+ end
+end