v1.0

lib/file_size_validator.rb

@@ -0,0 +1,65 @@
+class FileSizeValidator < ActiveModel::EachValidator
+  MESSAGES  = { :is => :wrong_size, :minimum => :size_too_small, :maximum => :size_too_big }.freeze
+  CHECKS    = { :is => :==, :minimum => :>=, :maximum => :<= }.freeze
+
+  DEFAULT_TOKENIZER = lambda { |value| value.split(//) }
+  RESERVED_OPTIONS  = [:minimum, :maximum, :within, :is, :tokenizer, :too_short, :too_long]
+
+  def initialize(options)
+    if range = (options.delete(:in) || options.delete(:within))
+      raise ArgumentError, ":in and :within must be a Range" unless range.is_a?(Range)
+      options[:minimum], options[:maximum] = range.begin, range.end
+      options[:maximum] -= 1 if range.exclude_end?
+    end
+
+    super
+  end
+
+  def check_validity!
+    keys = CHECKS.keys & options.keys
+
+    if keys.empty?
+      raise ArgumentError, 'Range unspecified. Specify the :within, :maximum, :minimum, or :is option.'
+    end
+
+    keys.each do |key|
+      value = options[key]
+
+      unless value.is_a?(Integer) && value >= 0
+        raise ArgumentError, ":#{key} must be a nonnegative Integer"
+      end
+    end
+  end
+
+  def validate_each(record, attribute, value)
+    raise(ArgumentError, "A CarrierWave::Uploader::Base object was expected") unless value.kind_of? CarrierWave::Uploader::Base
+    
+    value = (options[:tokenizer] || DEFAULT_TOKENIZER).call(value) if value.kind_of?(String)
+
+    CHECKS.each do |key, validity_check|
+      next unless check_value = options[key]
+
+      value ||= [] if key == :maximum
+
+      value_size = value.size
+      next if value_size.send(validity_check, check_value)
+
+      errors_options = options.except(*RESERVED_OPTIONS)
+      errors_options[:file_size] = help.number_to_human_size check_value
+
+      default_message = options[MESSAGES[key]]
+      errors_options[:message] ||= default_message if default_message
+
+      record.errors.add(attribute, MESSAGES[key], errors_options)
+    end
+  end
+  
+  def help
+    Helper.instance
+  end
+
+  class Helper
+    include Singleton
+    include ActionView::Helpers::NumberHelper
+  end
+end

lib/gitosis.rb

@@ -0,0 +1,73 @@
+require 'inifile'
+require 'timeout'
+require 'fileutils'
+
+class Gitosis
+  class AccessDenied < StandardError; end
+
+  def pull
+    # create tmp dir
+    @local_dir = File.join(Dir.tmpdir,"gitlabhq-gitosis-#{Time.now.to_i}")
+
+    Dir.mkdir @local_dir
+
+    `git clone #{GITOSIS['admin_uri']} #{@local_dir}/gitosis`
+  end
+
+  def push
+    Dir.chdir(File.join(@local_dir, "gitosis"))
+    `git add -A`
+    `git commit -am "Gitlab"`
+    `git push`
+    Dir.chdir(Rails.root)
+
+    FileUtils.rm_rf(@local_dir)
+  end
+
+  def configure
+    status = Timeout::timeout(20) do
+      File.open(File.join(Dir.tmpdir,"gitlabhq-gitosis.lock"), "w+") do |f|
+        f.flock(File::LOCK_EX)
+
+        pull
+        yield(self)
+        push
+
+        f.flock(File::LOCK_UN)
+      end
+    end
+  rescue Exception => ex
+    raise Gitosis::AccessDenied.new("gitosis timeout")
+  end
+
+  def destroy_project(project)
+    FileUtils.rm_rf(project.path_to_repo)
+    
+    conf = IniFile.new(File.join(@local_dir,'gitosis','gitosis.conf'))
+
+    conf.delete_section("group #{project.path}")
+
+    conf.write
+  end
+
+   #update or create
+  def update_keys(user, key)
+    File.open(File.join(@local_dir, 'gitosis/keydir',"#{user}.pub"), 'w') {|f| f.write(key.gsub(/\n/,'')) }
+  end
+
+  def delete_key(user)
+    File.unlink(File.join(@local_dir, 'gitosis/keydir',"#{user}.pub"))
+    `cd #{File.join(@local_dir,'gitosis')} ; git rm keydir/#{user}.pub`
+  end
+
+  #update or create
+  def update_project(repo_name, name_writers)
+    # write config file
+    conf = IniFile.new(File.join(@local_dir,'gitosis','gitosis.conf'))
+
+    conf["group #{repo_name}"]['writable'] = repo_name
+    conf["group #{repo_name}"]['members'] = name_writers.join(' ')
+
+    conf.write
+  end
+end

lib/utils.rb

@@ -0,0 +1,8 @@
+module Utils
+  def self.binary?(string) 
+    string.each_byte do |x|
+      x.nonzero? or return true 
+    end
+    false
+  end
+end