From 3216dc9b8d876f24ed561e81c2a1fb91926cb57a Mon Sep 17 00:00:00 2001 From: Grégoire Duchêne Date: Thu, 26 Mar 2020 22:00:40 +0000 Subject: Add a few test cases to lib --- lib/lib.go | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'lib/lib.go') diff --git a/lib/lib.go b/lib/lib.go index b066e69..a35cb13 100644 --- a/lib/lib.go +++ b/lib/lib.go @@ -12,15 +12,16 @@ import ( ) func GetBody(pkg string) string { - dest := strings.TrimRight(os.Getenv("DEST"), "/") + "/" + getRepo(pkg) + dest := GetDest(os.Getenv("PREFIX"), os.Getenv("DEST"), pkg) return fmt.Sprintf(` go-import-redirect `, pkg, os.Getenv("VCS"), dest) } -func getRepo(pkg string) string { - prefix := strings.TrimRight(os.Getenv("PREFIX"), "/") - path := strings.TrimLeft(strings.TrimPrefix(pkg, prefix), "/") - return strings.Split(path, "/")[0] +func GetDest(srcPrefix, destPrefix, pkg string) string { + srcPrefix = strings.TrimRight(srcPrefix, "/") + destPrefix = strings.TrimRight(destPrefix, "/") + path := strings.TrimLeft(strings.TrimPrefix(pkg, srcPrefix), "/") + return destPrefix + "/" + strings.Split(path, "/")[0] } -- cgit v1.2.3-70-g09d2