@@ -0,0 +1,27 @@ | |||
/* | |||
* The MIT License (MIT) | |||
* | |||
* Copyright (c) 2018 Ma_124, <github.com/Ma124> | |||
* | |||
* Permission is hereby granted, free of charge, to any person obtaining a copy | |||
* of this software and associated documentation files (the "Software"), to deal | |||
* in the Software without restriction, including without limitation the rights | |||
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |||
* copies of the Software, and to permit persons to whom the Software is | |||
* furnished to do so, Subject to the following conditions: | |||
* | |||
* The above copyright notice and this permission notice shall be included in all | |||
* copies or Substantial portions of the Software. | |||
* | |||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |||
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |||
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |||
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |||
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |||
* SOFTWARE. | |||
*/ | |||
package strcase | |||
@@ -2,6 +2,7 @@ | |||
* The MIT License (MIT) | |||
* | |||
* Copyright (c) 2015 Ian Coleman | |||
* Copyright (c) 2018 Ma_124, <github.com/Ma124> | |||
* | |||
* Permission is hereby granted, free of charge, to any person obtaining a copy | |||
* of this software and associated documentation files (the "Software"), to deal | |||
@@ -31,6 +32,10 @@ import ( | |||
// Converts a string to snake_case | |||
func ToSnake(s string) string { | |||
return ToDelimited(s, '_') | |||
} | |||
func ToDelimited(s string, del uint8) string { | |||
s = addWordBoundariesToNumbers(s) | |||
s = strings.Trim(s, " ") | |||
n := "" | |||
@@ -44,16 +49,16 @@ func ToSnake(s string) string { | |||
} | |||
} | |||
if i > 0 && n[len(n)-1] != '_' && nextCaseIsChanged { | |||
if i > 0 && n[len(n)-1] != del && nextCaseIsChanged { | |||
// add underscore if next letter case type is changed | |||
if v >= 'A' && v <= 'Z' { | |||
n += "_" + string(v) | |||
n += string(del) + string(v) | |||
} else if v >= 'a' && v <= 'z' { | |||
n += string(v) + "_" | |||
n += string(v) + string(del) | |||
} | |||
} else if v == ' ' { | |||
// replace spaces with underscores | |||
n += "_" | |||
} else if v == ' ' || v == '_' || v == '-' { | |||
// replace spaces/underscores with delimiters | |||
n += string(del) | |||
} else { | |||
n = n + string(v) | |||
} | |||
@@ -2,6 +2,7 @@ | |||
* The MIT License (MIT) | |||
* | |||
* Copyright (c) 2015 Ian Coleman | |||
* Copyright (c) 2018 Ma_124, <github.com/Ma124> | |||
* | |||
* Permission is hereby granted, free of charge, to any person obtaining a copy | |||
* of this software and associated documentation files (the "Software"), to deal | |||
@@ -53,7 +54,38 @@ func TestToSnake(t *testing.T) { | |||
out := i[1] | |||
result := ToSnake(in) | |||
if result != out { | |||
t.Error("'" + result + "' != '" + out + "'") | |||
t.Error("'" + in + "'('" + result + "' != '" + out + "')") | |||
} | |||
} | |||
} | |||
func TestToDelimited(t *testing.T) { | |||
cases := [][]string{ | |||
[]string{"testCase", "test@case"}, | |||
[]string{"TestCase", "test@case"}, | |||
[]string{"Test Case", "test@case"}, | |||
[]string{" Test Case", "test@case"}, | |||
[]string{"Test Case ", "test@case"}, | |||
[]string{" Test Case ", "test@case"}, | |||
[]string{"test", "test"}, | |||
[]string{"test_case", "test@case"}, | |||
[]string{"Test", "test"}, | |||
[]string{"", ""}, | |||
[]string{"ManyManyWords", "many@many@words"}, | |||
[]string{"manyManyWords", "many@many@words"}, | |||
[]string{"AnyKind of_string", "any@kind@of@string"}, | |||
[]string{"numbers2and55with000", "numbers@2@and@55@with@000"}, | |||
[]string{"JSONData", "json@data"}, | |||
[]string{"userID", "user@id"}, | |||
[]string{"AAAbbb", "aa@abbb"}, | |||
[]string{"test-case", "test@case"}, | |||
} | |||
for _, i := range cases { | |||
in := i[0] | |||
out := i[1] | |||
result := ToDelimited(in, '@') | |||
if result != out { | |||
t.Error("'" + in + "' ('" + result + "' != '" + out + "')") | |||
} | |||
} | |||
} |