mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
aa87442fdd
Source files keep their SPDX-License-Identifier and Hanzo copyright line; the Gitea, Gogs and LFS Test Server copyrights they used to repeat now live once in NOTICE, alongside LICENSE. The header lint accepts any copyright line above the SPDX line, so it no longer demands an upstream one. Co-authored-by: Hanzo Dev <[email protected]>
83 lines
2.0 KiB
Go
83 lines
2.0 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package v1_10
|
|
|
|
import (
|
|
"net/url"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/hanzoai/git/models/db"
|
|
)
|
|
|
|
func UpdateMigrationServiceTypes(x db.EngineMigration) error {
|
|
type Repository struct {
|
|
ID int64
|
|
OriginalServiceType int `xorm:"index default(0)"`
|
|
OriginalURL string `xorm:"VARCHAR(2048)"`
|
|
}
|
|
|
|
if err := x.Sync(new(Repository)); err != nil {
|
|
return err
|
|
}
|
|
|
|
var last int
|
|
const batchSize = 50
|
|
for {
|
|
results := make([]Repository, 0, batchSize)
|
|
err := x.Where("original_url <> '' AND original_url IS NOT NULL").
|
|
And("original_service_type = 0 OR original_service_type IS NULL").
|
|
OrderBy("id").
|
|
Limit(batchSize, last).
|
|
Find(&results)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(results) == 0 {
|
|
break
|
|
}
|
|
last += len(results)
|
|
|
|
const PlainGitService = 1 // 1 plain git service
|
|
const GithubService = 2 // 2 github.com
|
|
|
|
for _, res := range results {
|
|
u, err := url.Parse(res.OriginalURL)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
serviceType := PlainGitService
|
|
if strings.EqualFold(u.Host, "github.com") {
|
|
serviceType = GithubService
|
|
}
|
|
_, err = x.Exec("UPDATE repository SET original_service_type = ? WHERE id = ?", serviceType, res.ID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
type ExternalLoginUser struct {
|
|
ExternalID string `xorm:"pk NOT NULL"`
|
|
UserID int64 `xorm:"INDEX NOT NULL"`
|
|
LoginSourceID int64 `xorm:"pk NOT NULL"`
|
|
RawData map[string]any `xorm:"TEXT JSON"`
|
|
Provider string `xorm:"index VARCHAR(25)"`
|
|
Email string
|
|
Name string
|
|
FirstName string
|
|
LastName string
|
|
NickName string
|
|
Description string
|
|
AvatarURL string
|
|
Location string
|
|
AccessToken string
|
|
AccessTokenSecret string
|
|
RefreshToken string
|
|
ExpiresAt time.Time
|
|
}
|
|
|
|
return x.Sync(new(ExternalLoginUser))
|
|
}
|