photoprism/internal/config/dsn.go

63 lines
1.2 KiB
Go
Raw Normal View History

package config
import "regexp"
// dsnPattern is a regular expression matching a database DSN string.
var dsnPattern = regexp.MustCompile(
`^((?P<driver>.*):\/\/)?(?:(?P<user>.*?)(?::(?P<password>.*))?@)?` +
`(?:(?P<net>[^\(]*)(?:\((?P<server>[^\)]*)\))?)?` +
`\/(?P<name>.*?)` +
`(?:\?(?P<params>[^\?]*))?$`)
// DSN represents parts of a data source name.
type DSN struct {
Driver string
User string
Password string
Net string
Server string
Name string
Params string
}
// NewDSN creates a new DSN struct from a string.
func NewDSN(dsn string) DSN {
d := DSN{}
d.Parse(dsn)
return d
}
// Parse parses a data source name string.
func (d *DSN) Parse(dsn string) {
if dsn == "" {
return
}
matches := dsnPattern.FindStringSubmatch(dsn)
names := dsnPattern.SubexpNames()
for i, match := range matches {
switch names[i] {
case "driver":
d.Driver = match
case "user":
d.User = match
case "password":
d.Password = match
case "net":
d.Net = match
case "server":
d.Server = match
case "name":
d.Name = match
case "params":
d.Params = match
}
}
if d.Net != "" && d.Server == "" {
d.Server = d.Net
d.Net = ""
}
}