connection.go

// connection
package rets

import (
	"errors"
	"github.com/edmore/goca/auth"
	"github.com/mlapping/rets/results"
	"net/http"
	"net/http/cookiejar"
	"net/url"
)

type Session struct {
	DialInfo      *DialInfo
	HttpClient    *http.Client
	BaseAddress   string
	LoggedIn      bool
	Capabilities  Capabilities
	Authorization string
}

type DialInfo struct {
	LoginUrl string
	UserName string
	Password string
}

func DialWithInfo(info *DialInfo) (*Session, error) {
	err := info.validateDialInfo()
	if err != nil {
		return nil, err
	}

	// create a new session
	session := &Session{}
	session.DialInfo = info

	// start setting up the capabilities early
	loginUrl, err := url.Parse(session.DialInfo.LoginUrl)
	if err != nil {
		return nil, err
	}

	session.Capabilities = Capabilities{
		Host: loginUrl.Scheme + "://" + loginUrl.Host,
	}

	// create an http connection with a cookiejar
	cookieJar, _ := cookiejar.New(nil)
	session.HttpClient = &http.Client{
		Jar: cookieJar,
	}

	// login, set server capabilities
	err = session.tryToLogin()
	if err != nil {
		return nil, err
	}

	// Login was successful
	return session, nil
}

func (sess *Session) tryToLogin() error {

	// The first GET will fail, but set our cookie jar
	req, err := sess.newRequest("GET", sess.DialInfo.LoginUrl, nil)
	if err != nil {
		return err
	}

	// The first GET will fail, but set our cookie jar
	res, _ := sess.HttpClient.Do(req)

	// The second GET will SHOULD not fail
	req, err = sess.newRequest("GET", sess.DialInfo.LoginUrl, nil)
	// set our digest auth
	req = auth.SetDigestAuth(req, sess.DialInfo.UserName, sess.DialInfo.Password, res, 1)
	res, err = sess.HttpClient.Do(req)

	if res.StatusCode != http.StatusOK {
		return errors.New("login failed: " + err.Error())
	}

	// grab the authorization string
	sess.Authorization = req.Header.Get("Authorization")

	// convert the response to a RetsReply Object
	retsReply := &results.LoginReply{}
	err = results.ConvertServerResponse(res.Body, retsReply)
	if err != nil {
		return err
	}
	// setup our capabilities
	return sess.Capabilities.setFromLogin(retsReply)
}

func (dial *DialInfo) validateDialInfo() error {
	if dial == nil {
		return errors.New("passed a nil DialInfo object")
	}

	if len(dial.Password) == 0 {
		return errors.New("passed a blank password")
	}

	if len(dial.UserName) == 0 {
		return errors.New("passed a blank username")
	}

	return nil
}