util.go

package main

import (
	"os"
	"path"
	"regexp"
	"strings"
)

var (
	EnvVarRegexp           = regexp.MustCompile(`\$(\w+)`)
	UrlSchemaRegexp        = regexp.MustCompile(`^\w+://`)
	HttpUrlSchemaRegexp    = regexp.MustCompile(`^(\w+\.)+(\w+)(/.*)?$`)
	CmdlineComponentRegexp = regexp.MustCompile(`(?:([^'"\s]+)|'([^']+)'|"([^"]+)")`)
)

func EscapeAmpersand(s string) string {
	return strings.Replace(s, "&", "&", -1)
}

func EscapeHTML(s string) string {
	s = strings.Replace(s, "&", "&", -1)
	s = strings.Replace(s, "<", "&lt;", -1)
	s = strings.Replace(s, ">", "&gt;", -1)
	return s
}

func ExpandEnvVars(query string) string {
	matches := EnvVarRegexp.FindAllStringSubmatch(query, -1)
	for _, match := range matches {
		envVar := match[0]
		cleanEnvVar := match[1]
		query = strings.Replace(query, envVar, os.Getenv(cleanEnvVar), 1)
	}
	return query
}

func ExpandPathString(query string) (bool, string) {
	if query == "" {
		return false, query
	}
	if query[0] != '/' && query[0] != '~' {
		return false, query
	}

	if query[0] == '~' {
		query = HOME + query[1:]
	}
	return true, path.Clean(query)
}

func IsExecutable(info os.FileInfo) bool {
	if info == nil || info.IsDir() || (info.Mode().Perm()&0111) == 0 {
		return false
	}
	return true
}

func IsUrl(query string) bool {
	return UrlSchemaRegexp.MatchString(query)
}

func IsHttpUrl(query string) bool {
	return HttpUrlSchemaRegexp.MatchString(query)
}

func SplitCommandline(cmdline string) []string {
	components := []string{}

	matches := CmdlineComponentRegexp.FindAllStringSubmatch(cmdline, -1)
	for _, match := range matches {
		fstNonEmptyGrp := match[0]
		for _, g := range match[1:] {
			if g != "" {
				fstNonEmptyGrp = g
				break
			}
		}

		components = append(components, fstNonEmptyGrp)
	}

	return components
}