// Copyright (c) 2012-present The upper.io/db authors. All rights reserved.
//
// Permission is hereby granted, free of charge, to any person obtaining
// a copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to
// permit persons to whom the Software is furnished to do so, subject to
// the following conditions:
//
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

package ql

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

const defaultConnectionScheme = `file`

// ConnectionURL implements a QL connection struct.
type ConnectionURL struct {
	Scheme   string
	Database string
	Options  map[string]string
}

func (c ConnectionURL) String() (s string) {
	vv := url.Values{}

	if c.Database == "" {
		return ""
	}

	// Does the database have an existing name?
	if !strings.HasPrefix(c.Database, "/") {
		c.Database, _ = filepath.Abs(c.Database)
	}

	// Do we have any options?
	if c.Options == nil {
		c.Options = map[string]string{}
	}

	// Converting options into URL values.
	for k, v := range c.Options {
		vv.Set(k, v)
	}

	// Building URL.
	u := url.URL{
		Scheme:   c.Scheme,
		Path:     c.Database,
		RawQuery: vv.Encode(),
	}
	if u.Scheme == "" {
		u.Scheme = defaultConnectionScheme
	}

	return u.String()
}

// ParseURL parses s into a ConnectionURL struct.
func ParseURL(s string) (conn ConnectionURL, err error) {
	var u *url.URL

	if u, err = url.Parse(s); err != nil {
		return conn, err
	}

	if u.Scheme != "file" && u.Scheme != "memory" && u.Scheme != "" {
		return conn, fmt.Errorf(`Expecting file:// or memory:// connection scheme.`)
	}

	var vv url.Values
	if vv, err = url.ParseQuery(u.RawQuery); err != nil {
		return conn, err
	}

	conn.Scheme = u.Scheme
	conn.Database = u.Host + u.Path
	conn.Options = map[string]string{}
	for k := range vv {
		conn.Options[k] = vv.Get(k)
	}

	return conn, err
}