location.go

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

package gowsdl

import (
	"net/url"
	"path/filepath"
	"strings"
)

// A Location encapsulate information about the loc of WSDL/XSD.
//
// It could be either URL or an absolute file path.
type Location struct {
	u *url.URL
	f string
}

// ParseLocation parses a rawloc into a Location structure.
//
// If rawloc is URL then it should be absolute.
// Relative file path will be converted into absolute path.
func ParseLocation(rawloc string) (*Location, error) {
	u, _ := url.Parse(rawloc)
	if u.Scheme != "" {
		return &Location{u: u}, nil
	}

	absURI, err := filepath.Abs(rawloc)
	if err != nil {
		return nil, err
	}

	return &Location{f: absURI}, nil
}

// Parse parses path in the context of the receiver. The provided path may be relative or absolute.
// Parse returns nil, err on parse failure.
func (r *Location) Parse(ref string) (*Location, error) {
	if r.u != nil {
		u, err := r.u.Parse(fixReference(ref))
		if err != nil {
			return nil, err
		}
		return &Location{u: u}, nil
	}

	if filepath.IsAbs(ref) {
		return &Location{f: ref}, nil
	}

	if u, err := url.Parse(ref); err == nil {
		if u.Scheme != "" {
			return &Location{u: u}, nil
		}
	}

	return &Location{f: filepath.Join(filepath.Dir(r.f), ref)}, nil
}

// IsFile determines whether the Location contains a file path.
func (r *Location) isFile() bool {
	return r.f != ""
}

// IsFile determines whether the Location contains URL.
func (r *Location) isURL() bool {
	return r.u != nil
}

// String reassembles the Location either into a valid URL string or a file path.
func (r *Location) String() string {
	if r.isFile() {
		return r.f
	}
	if r.isURL() {
		return r.u.String()
	}
	return ""
}

// Fix various mistakes in the specified URL
func fixReference(ref string) string {
	fixedRef := ref
	if strings.HasPrefix(fixedRef, "//") {
		fixedRef = strings.Replace(fixedRef, "//", "/", 1)
	}
	return fixedRef
}