geoserver.go

package geoserver

import (
	"io"
	"io/ioutil"
	"net/http"

	"github.com/sirupsen/logrus"
	yaml "gopkg.in/yaml.v2"
)

//GeoServer is the configuration Object
type GeoServer struct {
	WorkspaceName string `yaml:"workspace"`
	ServerURL     string `yaml:"geoserver_url"`
	Username      string `yaml:"username"`
	Password      string `yaml:"password"`
	httpClient    *http.Client
	logger        *logrus.Logger
}

//LoadConfig load geoserver config from yaml file
func (g *GeoServer) LoadConfig(configFile string) (geoserver *GeoServer, err error) {

	yamlFile, err := ioutil.ReadFile(configFile)
	if err != nil {
		g.logger.Errorf("yamlFile.Get err   %v ", err)
		return
	}
	err = yaml.Unmarshal(yamlFile, g)
	if err != nil {
		g.logger.Errorf("Unmarshal: %v", err)
		return
	}
	g.logger = GetLogger()
	geoserver = g
	return
}

// GetGeoserverRequest creates a HTTP request with geoserver credintails and header
func (g *GeoServer) GetGeoserverRequest(
	targetURL string,
	method string,
	accept string,
	data io.Reader,
	contentType string) (request *http.Request) {
	request, _ = http.NewRequest(method, targetURL, data)
	if data != nil {
		request.Header.Set(contentTypeHeader, contentType)
	}
	if accept != "" {
		request.Header.Set(acceptHeader, accept)
	}

	request.SetBasicAuth(g.Username, g.Password)
	return
}