util.go

// Copyright © 2020 zc2638 <zc2638@qq.com>.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package swag

import (
	"fmt"
	"reflect"
	"regexp"
	"strconv"
	"strings"

	"github.com/modern-go/reflect2"
)

var (
	rePath         = regexp.MustCompile(`\{([^}]+)}`)
	reAlphaNumeric = regexp.MustCompile(`[^0-9a-zA-Z]`)
)

// ColonPath accepts a swagger path.
//
// e.g. /api/org/{org} and returns a colon identified path
// e.g. /api/org/:org
func ColonPath(path string) string {
	matches := rePath.FindAllStringSubmatch(path, -1)
	for _, match := range matches {
		path = strings.Replace(path, match[0], ":"+match[1], -1)
	}
	return path
}

func camel(v string) string {
	segments := strings.Split(v, "/")
	results := make([]string, 0, len(segments))

	for _, segment := range segments {
		v := reAlphaNumeric.ReplaceAllString(segment, "")
		if v == "" {
			continue
		}

		results = append(results, strings.ToUpper(v[0:1])+v[1:])
	}

	return strings.Join(results, "")
}

func makeRef(name string) string {
	return fmt.Sprintf("#/definitions/%v", name)
}

func makeName(t reflect.Type) string {
	name := t.Name()
	if name == "" {
		ptr := reflect2.PtrOf(t)
		name = "ptr" + strconv.FormatUint(uint64(uintptr(ptr)), 10)
	}
	pkgPath := t.PkgPath()
	if pkgPath != "." {
		pkgPath += "."
	}
	fullName := pkgPath + name
	fullName = strings.ReplaceAll(fullName, "/", "_")
	return strings.ReplaceAll(fullName, "-", "_")
}