DB and model changes for Sentry project selection dropdown

1 files changed, 78 insertions, 5 deletions

diff --git a/app/models/error_tracking/project_error_tracking_setting.rb b/app/models/error_tracking/project_error_tracking_setting.rb
index 7f4947ba27a..31084c54bdc 100644
--- a/app/models/error_tracking/project_error_tracking_setting.rb
+++ b/app/models/error_tracking/project_error_tracking_setting.rb
@@ -8,9 +8,13 @@ module ErrorTracking
 
     belongs_to :project
 
-    validates :api_url, length: { maximum: 255 }, public_url: true, url: { enforce_sanitization: true }
+    validates :api_url, length: { maximum: 255 }, public_url: true, url: { enforce_sanitization: true, ascii_only: true }, allow_nil: true
 
-    validate :validate_api_url_path
+    validates :api_url, presence: true, if: :enabled
+
+    validate :validate_api_url_path, if: :enabled
+
+    validates :token, presence: true, if: :enabled
 
     attr_encrypted :token,
       mode: :per_attribute_iv,
@@ -19,6 +23,31 @@ module ErrorTracking
 
     after_save :clear_reactive_cache!
 
+    def project_name
+      super || project_name_from_slug
+    end
+
+    def organization_name
+      super || organization_name_from_slug
+    end
+
+    def project_slug
+      project_slug_from_api_url
+    end
+
+    def organization_slug
+      organization_slug_from_api_url
+    end
+
+    def self.build_api_url_from(api_host:, project_slug:, organization_slug:)
+      uri = Addressable::URI.parse("#{api_host}/api/0/projects/#{organization_slug}/#{project_slug}/")
+      uri.path = uri.path.squeeze('/')
+
+      uri.to_s
+    rescue Addressable::URI::InvalidURIError
+      api_host
+    end
+
     def sentry_client
       Sentry::Client.new(api_url, token)
     end
@@ -33,6 +62,10 @@ module ErrorTracking
       end
     end
 
+    def list_sentry_projects
+      { projects: sentry_client.list_projects }
+    end
+
     def calculate_reactive_cache(request, opts)
       case request
       when 'list_issues'
@@ -47,13 +80,53 @@ module ErrorTracking
       url.sub('api/0/projects/', '')
     end
 
+    def api_host
+      return if api_url.blank?
+
+      # This returns http://example.com/
+      Addressable::URI.join(api_url, '/').to_s
+    end
+
     private
 
+    def project_name_from_slug
+      @project_name_from_slug ||= project_slug_from_api_url&.titleize
+    end
+
+    def organization_name_from_slug
+      @organization_name_from_slug ||= organization_slug_from_api_url&.titleize
+    end
+
+    def project_slug_from_api_url
+      extract_slug(:project)
+    end
+
+    def organization_slug_from_api_url
+      extract_slug(:organization)
+    end
+
+    def extract_slug(capture)
+      return if api_url.blank?
+
+      begin
+        url = Addressable::URI.parse(api_url)
+      rescue Addressable::URI::InvalidURIError
+        return nil
+      end
+
+      @slug_match ||= url.path.match(%r{^/api/0/projects/+(?<organization>[^/]+)/+(?<project>[^/|$]+)}) || {}
+      @slug_match[capture]
+    end
+
     def validate_api_url_path
-      unless URI(api_url).path.starts_with?('/api/0/projects')
-        errors.add(:api_url, 'path needs to start with /api/0/projects')
+      return if api_url.blank?
+
+      begin
+        unless Addressable::URI.parse(api_url).path.starts_with?('/api/0/projects')
+          errors.add(:api_url, 'path needs to start with /api/0/projects')
+        end
+      rescue Addressable::URI::InvalidURIError
       end
-    rescue URI::InvalidURIError
     end
   end
 end