mirror of https://github.com/tailscale/tailscale/
util/strs: add new package for string utility funcs
Updates #5309 Change-Id: I677cc6e01050b6e10d8d6907d961b11c7a787a05 Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>pull/5311/head
parent
622d80c007
commit
01e8ef7293
@ -0,0 +1,36 @@
|
|||||||
|
// Copyright (c) 2022 Tailscale Inc & AUTHORS All rights reserved.
|
||||||
|
// Use of this source code is governed by a BSD-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
// Package strs contains string-related utility funcs.
|
||||||
|
package strs
|
||||||
|
|
||||||
|
import "strings"
|
||||||
|
|
||||||
|
// CutPrefix returns s without the provided leading prefix string
|
||||||
|
// and reports whether it found the prefix.
|
||||||
|
// If s doesn't start with prefix, CutPrefix returns s, false.
|
||||||
|
// If prefix is the empty string, CutPrefix returns s, true.
|
||||||
|
//
|
||||||
|
// TODO: remove this once Go 1.20 is out with it.
|
||||||
|
// See https://github.com/tailscale/tailscale/issues/5309
|
||||||
|
func CutPrefix(s, prefix string) (after string, found bool) {
|
||||||
|
if !strings.HasPrefix(s, prefix) {
|
||||||
|
return s, false
|
||||||
|
}
|
||||||
|
return s[len(prefix):], true
|
||||||
|
}
|
||||||
|
|
||||||
|
// CutSuffix returns s without the provided ending suffix string
|
||||||
|
// and reports whether it found the suffix.
|
||||||
|
// If s doesn't end with suffix, CutSuffix returns s, false.
|
||||||
|
// If suffix is the empty string, CutSuffix returns s, true.
|
||||||
|
//
|
||||||
|
// See https://github.com/tailscale/tailscale/issues/5309
|
||||||
|
// TODO: remove this once Go 1.20 is out with it.
|
||||||
|
func CutSuffix(s, suffix string) (before string, found bool) {
|
||||||
|
if !strings.HasSuffix(s, suffix) {
|
||||||
|
return s, false
|
||||||
|
}
|
||||||
|
return s[:len(s)-len(suffix)], true
|
||||||
|
}
|
@ -0,0 +1,30 @@
|
|||||||
|
// Copyright (c) 2022 Tailscale Inc & AUTHORS All rights reserved.
|
||||||
|
// Use of this source code is governed by a BSD-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package strs
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
func TestCut(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
fn func(string, string) (string, bool)
|
||||||
|
in1, in2 string
|
||||||
|
want string
|
||||||
|
wantOK bool
|
||||||
|
}{
|
||||||
|
{CutPrefix, "foo", "fo", "o", true},
|
||||||
|
{CutPrefix, "bar", "fo", "bar", false},
|
||||||
|
{CutSuffix, "foo", "o", "fo", true},
|
||||||
|
{CutSuffix, "bar", "fo", "bar", false},
|
||||||
|
}
|
||||||
|
for i, tt := range tests {
|
||||||
|
got, gotOK := tt.fn(tt.in1, tt.in2)
|
||||||
|
if got != tt.want {
|
||||||
|
t.Errorf("%d. got %q; want %q", i, got, tt.want)
|
||||||
|
}
|
||||||
|
if gotOK != tt.wantOK {
|
||||||
|
t.Errorf("%d. got %v; want %v", i, gotOK, tt.wantOK)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue