Merge pull request #212 from jirutka/patch-1

Allow to configure location of the secret file

2 files changed, 5 insertions, 1 deletions

diff --git a/lib/gitlab_config.rb b/lib/gitlab_config.rb
index 300d6ba..caca176 100644
--- a/lib/gitlab_config.rb
+++ b/lib/gitlab_config.rb
@@ -19,6 +19,10 @@ class GitlabConfig
     @config['auth_file'] ||= File.join(home, ".ssh/authorized_keys")
   end
 
+  def secret_file
+    @config['secret_file'] ||= File.join(ROOT_PATH, '.gitlab_shell_secret')
+  end
+
   def gitlab_url
     @config['gitlab_url'] ||= "http://localhost/"
   end
diff --git a/lib/gitlab_net.rb b/lib/gitlab_net.rb
index d5bda88..8eb63ae 100644
--- a/lib/gitlab_net.rb
+++ b/lib/gitlab_net.rb
@@ -137,6 +137,6 @@ class GitlabNet
   end
 
   def secret_token
-    @secret_token ||= File.read File.join(ROOT_PATH, '.gitlab_shell_secret')
+    @secret_token ||= File.read config.secret_file
   end
 end