Support Golang subpackage fetching

Closes #13805

Support Golang subpackage fetching
Closes #13805
cc065fbe · Stan Hu · f7da99ae · cc065fbe · cc065fbe · cc065fbe
Commit cc065fbe authored Mar 12, 2016 by Stan Hu
Show whitespace changes
Inline Side-by-side

Showing with 53 additions and 0 deletions

CHANGELOG CHANGELOG +1 -0

config/initializers/go_get.rb config/initializers/go_get.rb +1 -0

lib/gitlab/middleware/go.rb lib/gitlab/middleware/go.rb +51 -0

No files found.
--- a/CHANGELOG
+++ b/CHANGELOG
 Please view this file on the master branch, on stable branches it's out of date.
 v 8.6.0 (unreleased)
+  - Support Golang subpackage fetching (Stan Hu)
  - Contributions to forked projects are included in calendar
  - Improve the formatting for the user page bio (Connor Shea)
  - Removed the default password from the initial admin account created during

--- a/config/initializers/go_get.rb
+++ b/config/initializers/go_get.rb
+Rails.application.config.middleware.use(Gitlab::Middleware::Go)
--- a/lib/gitlab/middleware/go.rb
+++ b/lib/gitlab/middleware/go.rb
+# A dumb middleware that returns a Go HTML document if the go-get=1 query string
+# is used irrespective if the namespace/project exists
+module Gitlab
+  module Middleware
+    class Go
+      def initialize(app)
+        @app = app
+      end
+      def call(env)
+        request = Rack::Request.new(env)
+        if go_request?(request)
+          render_go_doc(request)
+        else
+          @app.call(env)
+        end
+      end
+      private
+      def render_go_doc(request)
+        body = go_body(request)
+        response = Rack::Response.new(body, 200, { 'Content-Type' => 'text/html' })
+        response.finish
+      end
+      def go_request?(request)
+        return request["go-get"].to_i == 1
+      end
+      def go_body(request)
+        base_url = Settings.gitlab['url']
+        # Go subpackages may be in the form of namespace/project/path1/path2/../pathN
+        # We can just ignore the paths and leave the namespace/project
+        path_info = request.env["PATH_INFO"]
+        path_info.sub!(/^\//, '')
+        project_path = path_info.split('/').first(2).join('/')
+        request_url = URI.join(base_url, project_path)
+        domain_path = strip_url(request_url.to_s)
+        "<!DOCTYPE html><html><head><meta content='#{domain_path} git #{request_url}.git' name='go-import'></head></html>\n";
+      end
+      def strip_url(url)
+        url.gsub('http://', '').
+            gsub('https://', '')
+      end
+    end
+  end
+end