Merge branch 'import-timeout' into 'master'

Import Timeout

CHANGELOG

+v1.8.3
+  - Add timeout option for repository import
+
 v1.8.2
   - Fix broken 1.8.1
 

README.md

@@ -36,8 +36,12 @@ Remove repo
 
 Import repo 
 
+    # Default timeout is 2 minutes 
     ./bin/gitlab-projects import-project randx/six.git https://github.com/randx/six.git 
     
+    # Override timeout in seconds
+    ./bin/gitlab-projects import-project randx/six.git https://github.com/randx/six.git 90
+
 Fork repo
 
     ./bin/gitlab-projects fork-project gitlab/gitlab-ci.git randx

VERSION

-1.8.2
+1.8.3

lib/gitlab_projects.rb

 require 'fileutils'
+require 'timeout'
 
 require_relative 'gitlab_config'
 require_relative 'gitlab_logger'
@@ -92,9 +93,26 @@ class GitlabProjects
   # URL must be publicly cloneable
   def import_project
     @source = ARGV.shift
+
+    # timeout for clone
+    timeout = (ARGV.shift || 120).to_i
     $logger.info "Importing project #{@project_name} from <#{@source}> to <#{full_path}>."
     cmd = %W(git clone --bare -- #{@source} #{full_path})
-    system(*cmd) && self.class.create_hooks(full_path)
+
+    pid = Process.spawn(*cmd)
+
+    begin
+      Timeout.timeout(timeout) do
+        Process.wait(pid)
+      end
+    rescue Timeout::Error
+      Process.kill('KILL', pid)
+      $logger.error "Importing project #{@project_name} from <#{@source}> failed due to timeout."
+      FileUtils.rm_rf(full_path)
+      false
+    else
+      self.class.create_hooks(full_path)
+    end
   end
 
   # Move repository from one directory to another

spec/gitlab_projects_spec.rb

@@ -196,8 +196,11 @@ describe GitlabProjects do
   end
 
   describe :import_project do
+    context 'success import' do
       let(:gl_projects) { build_gitlab_projects('import-project', repo_name, 'https://github.com/randx/six.git') }
 
+      it { gl_projects.exec.should be_true }
+
       it "should import a repo" do
         gl_projects.exec
         File.exists?(File.join(tmp_repo_path, 'HEAD')).should be_true
@@ -210,6 +213,24 @@ describe GitlabProjects do
       end
     end
 
+    context 'timeout' do
+      let(:gl_projects) { build_gitlab_projects('import-project', repo_name, 'https://github.com/gitlabhq/gitlabhq.git', '1') }
+
+      it { gl_projects.exec.should be_false }
+
+      it "should not import a repo" do
+        gl_projects.exec
+        File.exists?(File.join(tmp_repo_path, 'HEAD')).should be_false
+      end
+
+      it "should log an import-project event" do
+        message = "Importing project #{repo_name} from <https://github.com/gitlabhq/gitlabhq.git> failed due to timeout."
+        $logger.should_receive(:error).with(message)
+        gl_projects.exec
+      end
+    end
+  end
+
   describe :fork_project do
     let(:source_repo_name) { File.join('source-namespace', repo_name) }
     let(:dest_repo) { File.join(tmp_repos_path, 'forked-to-namespace', repo_name) }