Add bookmark options to load username/password from env vars (#638)

This commit is contained in:
Jens Broekens
2023-01-20 23:59:39 +01:00
committed by GitHub
parent 79bd9c2f4e
commit 06212b4c34
2 changed files with 82 additions and 11 deletions

View File

@@ -1,21 +1,25 @@
package bookmarks
import (
"os"
"github.com/sosedoff/pgweb/pkg/command"
"github.com/sosedoff/pgweb/pkg/shared"
)
// Bookmark contains information about bookmarked database connection
type Bookmark struct {
ID string // ID generated from the filename
URL string // Postgres connection URL
Host string // Server hostname
Port int // Server port
User string // Database user
Password string // User password
Database string // Database name
SSLMode string // Connection SSL mode
SSH *shared.SSHInfo // SSH tunnel config
ID string // ID generated from the filename
URL string // Postgres connection URL
Host string // Server hostname
Port int // Server port
User string // Database user
UserVar string // Database user environment variable
Password string // User password
PasswordVar string // User password environment variable
Database string // Database name
SSLMode string // Connection SSL mode
SSH *shared.SSHInfo // SSH tunnel config
}
// SSHInfoIsEmpty returns true if ssh configuration is not provided
@@ -25,12 +29,22 @@ func (b Bookmark) SSHInfoIsEmpty() bool {
// ConvertToOptions returns an options struct from connection details
func (b Bookmark) ConvertToOptions() command.Options {
user := b.User
if b.User == "" {
user = os.Getenv(b.UserVar)
}
pass := b.Password
if b.Password == "" {
pass = os.Getenv(b.PasswordVar)
}
return command.Options{
URL: b.URL,
Host: b.Host,
Port: b.Port,
User: b.User,
Pass: b.Password,
User: user,
Pass: pass,
DbName: b.Database,
SSLMode: b.SSLMode,
}