Update API and fetching task

2 files changed, 46 insertions, 38 deletions

diff --git a/lib/gitlab/gitignore.rb b/lib/gitlab/gitignore.rb
index a2de2831e38..f46b43b61a4 100644
--- a/lib/gitlab/gitignore.rb
+++ b/lib/gitlab/gitignore.rb
@@ -2,15 +2,16 @@ module Gitlab
   class Gitignore
     FILTER_REGEX = /\.gitignore\z/.freeze
 
-    attr_accessor :name, :directory
+    def initialize(path)
+      @path = path
+    end
 
-    def initialize(name, directory)
-      @name       = name
-      @directory  = directory
+    def name
+      File.basename(@path, '.gitignore')
     end
 
     def content
-      File.read(path)
+      File.read(@path)
     end
 
     class << self
@@ -22,46 +23,33 @@ module Gitlab
         file_name = "#{key}.gitignore"
 
         directory = select_directory(file_name)
-        directory ? new(key, directory) : nil
+        directory ? new(File.join(directory, file_name)) : nil
       end
 
       def global
-        files_for_folder(global_dir).map { |f| new(f, global_dir) }
+        files_for_folder(global_dir).map { |file| new(File.join(global_dir, file)) }
       end
 
       def languages_frameworks
-        files_for_folder(gitignore_dir).map { |f| new(f, gitignore_dir) }
+        files_for_folder(gitignore_dir).map { |file| new(File.join(gitignore_dir, file)) }
       end
-    end
 
-    private
-
-    def path
-      File.expand_path("#{name}.gitignore", directory)
-    end
+      private
 
-    class << self
       def select_directory(file_name)
-        [self.gitignore_dir, self.global_dir].find { |dir| File.exist?(File.expand_path(file_name, dir)) }
+        [gitignore_dir, global_dir].find { |dir| File.exist?(File.join(dir, file_name)) }
       end
 
       def global_dir
-        File.expand_path('Global', gitignore_dir)
+        File.join(gitignore_dir, 'Global')
       end
 
       def gitignore_dir
-        File.expand_path('vendor/gitignore', Rails.root)
+        Rails.root.join('vendor/gitignore')
       end
 
       def files_for_folder(dir)
-        gitignores = []
-        Dir.entries(dir).each do |e|
-          next unless e.end_with?('.gitignore')
-
-          gitignores << e.gsub(FILTER_REGEX, '')
-        end
-
-        gitignores
+        Dir.glob("#{dir.to_s}/*.gitignore").map { |file| file.gsub(FILTER_REGEX, '') }
       end
     end
   end
diff --git a/lib/tasks/gitlab/update_gitignore.rake b/lib/tasks/gitlab/update_gitignore.rake
index 61cbfd6737d..84aa312002b 100644
--- a/lib/tasks/gitlab/update_gitignore.rake
+++ b/lib/tasks/gitlab/update_gitignore.rake
@@ -1,26 +1,46 @@
 namespace :gitlab do
   desc "GitLab | Update gitignore"
   task :update_gitignore do
-    dir = File.expand_path('vendor', Rails.root)
-    FileUtils.cd(dir)
+    unless clone_gitignores
+      puts "Cloning the gitignores failed".red
+      return
+    end
 
-    dir = File.expand_path('gitignore', dir)
-    clone_gitignores(dir)
-    remove_unneeded_files(dir)
+    remove_unneeded_files(gitignore_directory)
+    remove_unneeded_files(global_directory)
 
     puts "Done".green
   end
 
-  def clone_gitignores(dir)
-    FileUtils.rm_rf(dir) if Dir.exist?(dir)
+  def clone_gitignores
+    FileUtils.rm_rf(gitignore_directory) if Dir.exist?(gitignore_directory)
+    FileUtils.cd vendor_directory
+
     system('git clone --depth=1 --branch=master https://github.com/github/gitignore.git')
   end
 
-  def remove_unneeded_files(dir)
-    [File.expand_path('Global', dir), dir].each do |path|
-      Dir.entries(path).reject { |e| e =~ /(\.{1,2}|Global|\.gitignore)\z/ }.each do |file|
-        FileUtils.rm_rf File.expand_path(file, path)
-      end
+  # Retain only certain files:
+  # - The LICENSE, because we have to
+  # - The sub dir global
+  # - The gitignores themself
+  # - Dir.entires returns also the entries '.' and '..'
+  def remove_unneeded_files(path)
+    Dir.foreach(path) do |file|
+      FileUtils.rm_rf(File.join(path, file)) unless file =~ /(\.{1,2}|LICENSE|Global|\.gitignore)\z/
     end
   end
+
+  private
+
+  def vendor_directory
+    Rails.root.join('vendor')
+  end
+
+  def gitignore_directory
+    File.join(vendor_directory, 'gitignore')
+  end
+
+  def global_directory
+    File.join(gitignore_directory, 'Global')
+  end
 end