moul/sshportal
1
1
Fork 0
mirror of https://github.com/moul/sshportal.git synced 2025-09-13 16:14:39 +08:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #253 from jwessel/feat_postgres

Browse source 
feat: Add postgres support
This commit is contained in:
Manfred Touron 2021-03-26 16:26:05 +01:00 • committed by GitHub
commit 84a0a31eda
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 39 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
pkg/dbmodels/dbmodels.go
View file

@ -167,6 +167,25 @@ const (
BastionSchemeTelnet BastionScheme = "telnet" BastionSchemeTelnet BastionScheme = "telnet"
) )
// Generic Helper
func GenericNameOrID(db *gorm.DB, identifiers []string) *gorm.DB {
var ids []string
var names []string
for _, s := range identifiers {
if _, err := strconv.Atoi(s); err == nil {
ids = append(ids, s)
} else {
names = append(names, s)
}
}
if len(ids) > 0 && len(names) > 0 {
return db.Where("id IN (?)", ids).Or("name IN (?)", names)
} else if len(ids) > 0 {
return db.Where("id IN (?)", ids)
}
return db.Where("name IN (?)", names)
}
// Host helpers // Host helpers
func (host *Host) DialAddr() string { func (host *Host) DialAddr() string {
@ -268,7 +287,7 @@ func HostsPreload(db *gorm.DB) *gorm.DB {
return db.Preload("Groups").Preload("SSHKey") return db.Preload("Groups").Preload("SSHKey")
} }
func HostsByIdentifiers(db *gorm.DB, identifiers []string) *gorm.DB { func HostsByIdentifiers(db *gorm.DB, identifiers []string) *gorm.DB {
return db.Where("id IN (?)", identifiers).Or("name IN (?)", identifiers) return GenericNameOrID(db, identifiers)
} }
func HostByName(db *gorm.DB, name string) (*Host, error) { func HostByName(db *gorm.DB, name string) (*Host, error) {
var host Host var host Host
@ -308,7 +327,7 @@ func SSHKeysPreload(db *gorm.DB) *gorm.DB {
return db.Preload("Hosts") return db.Preload("Hosts")
} }
func SSHKeysByIdentifiers(db *gorm.DB, identifiers []string) *gorm.DB { func SSHKeysByIdentifiers(db *gorm.DB, identifiers []string) *gorm.DB {
return db.Where("id IN (?)", identifiers).Or("name IN (?)", identifiers) return GenericNameOrID(db, identifiers)
} }
// HostGroup helpers // HostGroup helpers
@ -317,7 +336,7 @@ func HostGroupsPreload(db *gorm.DB) *gorm.DB {
return db.Preload("ACLs").Preload("Hosts") return db.Preload("ACLs").Preload("Hosts")
} }
func HostGroupsByIdentifiers(db *gorm.DB, identifiers []string) *gorm.DB { func HostGroupsByIdentifiers(db *gorm.DB, identifiers []string) *gorm.DB {
return db.Where("id IN (?)", identifiers).Or("name IN (?)", identifiers) return GenericNameOrID(db, identifiers)
} }
// UserGroup helpers // UserGroup helpers
@ -326,7 +345,7 @@ func UserGroupsPreload(db *gorm.DB) *gorm.DB {
return db.Preload("ACLs").Preload("Users") return db.Preload("ACLs").Preload("Users")
} }
func UserGroupsByIdentifiers(db *gorm.DB, identifiers []string) *gorm.DB { func UserGroupsByIdentifiers(db *gorm.DB, identifiers []string) *gorm.DB {
return db.Where("id IN (?)", identifiers).Or("name IN (?)", identifiers) return GenericNameOrID(db, identifiers)
} }
// User helpers // User helpers
@ -335,7 +354,21 @@ func UsersPreload(db *gorm.DB) *gorm.DB {
return db.Preload("Groups").Preload("Keys").Preload("Roles") return db.Preload("Groups").Preload("Keys").Preload("Roles")
} }
func UsersByIdentifiers(db *gorm.DB, identifiers []string) *gorm.DB { func UsersByIdentifiers(db *gorm.DB, identifiers []string) *gorm.DB {
return db.Where("id IN (?)", identifiers).Or("email IN (?)", identifiers).Or("name IN (?)", identifiers) var ids []string
var names []string
for _, s := range identifiers {
if _, err := strconv.Atoi(s); err == nil {
ids = append(ids, s)
} else {
names = append(names, s)
}
}
if len(ids) > 0 && len(names) > 0 {
db.Where("id IN (?)", identifiers).Or("email IN (?)", identifiers).Or("name IN (?)", identifiers)
} else if len(ids) > 0 {
return db.Where("id IN (?)", ids)
}
return db.Where("email IN (?)", identifiers).Or("name IN (?)", identifiers)
} }
func (u *User) HasRole(name string) bool { func (u *User) HasRole(name string) bool {
for _, role := range u.Roles { for _, role := range u.Roles {
@ -381,7 +414,7 @@ func UserKeysByUserID(db *gorm.DB, identifiers []string) *gorm.DB {
// return db.Preload("Users") // return db.Preload("Users")
//} //}
func UserRolesByIdentifiers(db *gorm.DB, identifiers []string) *gorm.DB { func UserRolesByIdentifiers(db *gorm.DB, identifiers []string) *gorm.DB {
return db.Where("id IN (?)", identifiers).Or("name IN (?)", identifiers) return GenericNameOrID(db, identifiers)
} }
// Session helpers // Session helpers