Add EndpointType

Signed-off-by: Gabriel Adrian Samfira <gsamfira@cloudbasesolutions.com>

database/sql/github.go

@@ -97,6 +97,7 @@ func (s *sqlDatabase) sqlToCommonGithubEndpoint(ep GithubEndpoint) (params.Githu
 		UploadBaseURL: ep.UploadBaseURL,
 		CACertBundle:  ep.CACertBundle,
 		CreatedAt:     ep.CreatedAt,
+		EndpointType:  ep.EndpointType,
 		UpdatedAt:     ep.UpdatedAt,
 	}, nil
 }

database/sql/models.go

@@ -381,6 +381,8 @@ type GithubEndpoint struct {
 	UpdatedAt time.Time
 	DeletedAt gorm.DeletedAt `gorm:"index"`
 
+	EndpointType params.EndpointType
+
 	Description   string `gorm:"type:text"`
 	APIBaseURL    string `gorm:"type:text collate nocase"`
 	UploadBaseURL string `gorm:"type:text collate nocase"`

database/sql/sql.go

@@ -409,6 +409,17 @@ func (s *sqlDatabase) migrateDB() error {
 		}
 	}
 
+	if s.conn.Migrator().HasTable(&GithubEndpoint{}) {
+		if !s.conn.Migrator().HasColumn(&GithubEndpoint{}, "endpoint_type") {
+			if err := s.conn.Migrator().AutoMigrate(&GithubEndpoint{}); err != nil {
+				return errors.Wrap(err, "migrating github endpoints")
+			}
+			if err := s.conn.Exec("update github_endpoints set endpoint_type = 'github' where endpoint_type is null").Error; err != nil {
+				return errors.Wrap(err, "updating github endpoints")
+			}
+		}
+	}
+
 	var needsCredentialMigration bool
 	if !s.conn.Migrator().HasTable(&GithubCredentials{}) || !s.conn.Migrator().HasTable(&GithubEndpoint{}) {
 		needsCredentialMigration = true