mirror of
https://github.com/steveiliop56/tinyauth.git
synced 2025-11-04 08:05:42 +00:00
03d06cb0a7
* feat: add sqlite database for storing sessions * refactor: use db instance instead of service in auth service * fix: coderabbit suggestions
103 lines
1.8 KiB
Go
103 lines
1.8 KiB
Go
package utils
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"errors"
|
|
"net"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func GetSecret(conf string, file string) string {
|
|
if conf == "" && file == "" {
|
|
return ""
|
|
}
|
|
|
|
if conf != "" {
|
|
return conf
|
|
}
|
|
|
|
contents, err := ReadFile(file)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
|
|
return ParseSecretFile(contents)
|
|
}
|
|
|
|
func ParseSecretFile(contents string) string {
|
|
lines := strings.Split(contents, "\n")
|
|
|
|
for _, line := range lines {
|
|
if strings.TrimSpace(line) == "" {
|
|
continue
|
|
}
|
|
return strings.TrimSpace(line)
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
func GetBasicAuth(username string, password string) string {
|
|
auth := username + ":" + password
|
|
return base64.StdEncoding.EncodeToString([]byte(auth))
|
|
}
|
|
|
|
func FilterIP(filter string, ip string) (bool, error) {
|
|
ipAddr := net.ParseIP(ip)
|
|
|
|
if strings.Contains(filter, "/") {
|
|
_, cidr, err := net.ParseCIDR(filter)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return cidr.Contains(ipAddr), nil
|
|
}
|
|
|
|
ipFilter := net.ParseIP(filter)
|
|
if ipFilter == nil {
|
|
return false, errors.New("invalid IP address in filter")
|
|
}
|
|
|
|
if ipFilter.Equal(ipAddr) {
|
|
return true, nil
|
|
}
|
|
|
|
return false, nil
|
|
}
|
|
|
|
func CheckFilter(filter string, str string) bool {
|
|
if len(strings.TrimSpace(filter)) == 0 {
|
|
return true
|
|
}
|
|
|
|
if strings.HasPrefix(filter, "/") && strings.HasSuffix(filter, "/") {
|
|
re, err := regexp.Compile(filter[1 : len(filter)-1])
|
|
if err != nil {
|
|
return false
|
|
}
|
|
|
|
if re.MatchString(strings.TrimSpace(str)) {
|
|
return true
|
|
}
|
|
}
|
|
|
|
filterSplit := strings.Split(filter, ",")
|
|
|
|
for _, item := range filterSplit {
|
|
if strings.TrimSpace(item) == strings.TrimSpace(str) {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func GenerateIdentifier(str string) string {
|
|
uuid := uuid.NewSHA1(uuid.NameSpaceURL, []byte(str))
|
|
uuidString := uuid.String()
|
|
return strings.Split(uuidString, "-")[0]
|
|
}
|