Support Github and Gitea OAuth providers

internal/models/admin_setting.go

@@ -1,8 +1,6 @@
 package models
 
 import (
-	"errors"
-	"github.com/mattn/go-sqlite3"
 	"gorm.io/gorm/clause"
 )
 
@@ -38,7 +36,7 @@ func setSetting(key string, value string) error {
 func initAdminSettings(settings map[string]string) error {
 	for key, value := range settings {
 		if err := setSetting(key, value); err != nil {
-			if !isUniqueConstraintViolation(err) {
+			if !IsUniqueConstraintViolation(err) {
 				return err
 			}
 		}
@@ -46,11 +44,3 @@ func initAdminSettings(settings map[string]string) error {
 
 	return nil
 }
-
-func isUniqueConstraintViolation(err error) bool {
-	var sqliteErr sqlite3.Error
-	if errors.As(err, &sqliteErr) && sqliteErr.ExtendedCode == sqlite3.ErrConstraintUnique {
-		return true
-	}
-	return false
-}

internal/models/db.go

@@ -1,6 +1,8 @@
 package models
 
 import (
+	"errors"
+	"github.com/mattn/go-sqlite3"
 	"gorm.io/driver/sqlite"
 	"gorm.io/gorm"
 	"gorm.io/gorm/logger"
@@ -32,3 +34,11 @@ func CountAll(table interface{}) (int64, error) {
 	err := db.Model(table).Count(&count).Error
 	return count, err
 }
+
+func IsUniqueConstraintViolation(err error) bool {
+	var sqliteErr sqlite3.Error
+	if errors.As(err, &sqliteErr) && sqliteErr.ExtendedCode == sqlite3.ErrConstraintUnique {
+		return true
+	}
+	return false
+}

internal/models/sshkey.go

@@ -1,6 +1,12 @@
 package models
 
-import "time"
+import (
+	"crypto/sha256"
+	"encoding/base64"
+	"golang.org/x/crypto/ssh"
+	"gorm.io/gorm"
+	"time"
+)
 
 type SSHKey struct {
 	ID         uint `gorm:"primaryKey"`
@@ -13,6 +19,16 @@ type SSHKey struct {
 	User       User `validate:"-" `
 }
 
+func (sshKey *SSHKey) BeforeCreate(tx *gorm.DB) error {
+	pubKey, _, _, _, err := ssh.ParseAuthorizedKey([]byte(sshKey.Content))
+	if err != nil {
+		return err
+	}
+	sha := sha256.Sum256(pubKey.Marshal())
+	sshKey.SHA = base64.StdEncoding.EncodeToString(sha[:])
+	return nil
+}
+
 func GetSSHKeysByUserID(userId uint) ([]*SSHKey, error) {
 	var sshKeys []*SSHKey
 	err := db.

internal/models/user.go

@@ -12,6 +12,8 @@ type User struct {
 	CreatedAt int64
 	Email     string
 	MD5Hash   string // for gravatar, if no Email is specified, the value is random
+	GithubID  string
+	GiteaID   string
 
 	Gists   []Gist   `gorm:"constraint:OnUpdate:CASCADE,OnDelete:CASCADE;foreignKey:UserID"`
 	SSHKeys []SSHKey `gorm:"foreignKey:UserID"`
@@ -90,6 +92,19 @@ func GetUserBySSHKeyID(sshKeyId uint) (*User, error) {
 	return user, err
 }
 
+func GetUserByProvider(id string, provider string) (*User, error) {
+	user := new(User)
+	var err error
+	switch provider {
+	case "github":
+		err = db.Where("github_id = ?", id).First(&user).Error
+	case "gitea":
+		err = db.Where("gitea_id = ?", id).First(&user).Error
+	}
+
+	return user, err
+}
+
 func (user *User) Create() error {
 	return db.Create(&user).Error
 }
@@ -118,6 +133,17 @@ func (user *User) HasLiked(gist *Gist) (bool, error) {
 	return true, nil
 }
 
+func (user *User) DeleteProvider(provider string) error {
+	switch provider {
+	case "github":
+		return db.Model(&user).Update("github_id", nil).Error
+	case "gitea":
+		return db.Model(&user).Update("gitea_id", nil).Error
+	}
+
+	return nil
+}
+
 // -- DTO -- //
 
 type UserDTO struct {