80 lines
1.9 KiB
Go
80 lines
1.9 KiB
Go
|
// Package pgservicefile is a parser for PostgreSQL service files (e.g. .pg_service.conf).
|
||
|
|
package pgservicefile
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bufio"
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Service struct {
|
||
|
|
Name string
|
||
|
|
Settings map[string]string
|
||
|
|
}
|
||
|
|
|
||
|
|
type Servicefile struct {
|
||
|
|
Services []*Service
|
||
|
|
servicesByName map[string]*Service
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetService returns the named service.
|
||
|
|
func (sf *Servicefile) GetService(name string) (*Service, error) {
|
||
|
|
service, present := sf.servicesByName[name]
|
||
|
|
if !present {
|
||
|
|
return nil, errors.New("not found")
|
||
|
|
}
|
||
|
|
return service, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// ReadServicefile reads the file at path and parses it into a Servicefile.
|
||
|
|
func ReadServicefile(path string) (*Servicefile, error) {
|
||
|
|
f, err := os.Open(path)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
defer f.Close()
|
||
|
|
|
||
|
|
return ParseServicefile(f)
|
||
|
|
}
|
||
|
|
|
||
|
|
// ParseServicefile reads r and parses it into a Servicefile.
|
||
|
|
func ParseServicefile(r io.Reader) (*Servicefile, error) {
|
||
|
|
servicefile := &Servicefile{}
|
||
|
|
|
||
|
|
var service *Service
|
||
|
|
scanner := bufio.NewScanner(r)
|
||
|
|
lineNum := 0
|
||
|
|
for scanner.Scan() {
|
||
|
|
lineNum += 1
|
||
|
|
line := scanner.Text()
|
||
|
|
line = strings.TrimSpace(line)
|
||
|
|
|
||
|
|
if line == "" || strings.HasPrefix(line, "#") {
|
||
|
|
// ignore comments and empty lines
|
||
|
|
} else if strings.HasPrefix(line, "[") && strings.HasSuffix(line, "]") {
|
||
|
|
service = &Service{Name: line[1 : len(line)-1], Settings: make(map[string]string)}
|
||
|
|
servicefile.Services = append(servicefile.Services, service)
|
||
|
|
} else {
|
||
|
|
parts := strings.SplitN(line, "=", 2)
|
||
|
|
if len(parts) != 2 {
|
||
|
|
return nil, fmt.Errorf("unable to parse line %d", lineNum)
|
||
|
|
}
|
||
|
|
|
||
|
|
key := strings.TrimSpace(parts[0])
|
||
|
|
value := strings.TrimSpace(parts[1])
|
||
|
|
|
||
|
|
service.Settings[key] = value
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
servicefile.servicesByName = make(map[string]*Service, len(servicefile.Services))
|
||
|
|
for _, service := range servicefile.Services {
|
||
|
|
servicefile.servicesByName[service.Name] = service
|
||
|
|
}
|
||
|
|
|
||
|
|
return servicefile, scanner.Err()
|
||
|
|
}
|