added additional 2 conversion settings : first symbol case and the whole text case

Author: EvilLord666

stringstyle_formatter.go

@@ -6,13 +6,20 @@ import (
 )
 
 type FormattingStyle string
+type CaseSetting int
 
 const (
 	Camel FormattingStyle = "camel"
 	Snake FormattingStyle = "snake"
 	Kebab FormattingStyle = "kebab"
 )
 
+const (
+	ToUpper   CaseSetting = 1
+	ToLower   CaseSetting = 2
+	NoChanges CaseSetting = 3
+)
+
 type styleInc struct {
 	Index int
 	Style FormattingStyle
@@ -23,7 +30,22 @@ var styles = map[rune]FormattingStyle{
 	'-': Kebab,
 }
 
-func SetStyle(text *string, style FormattingStyle) string {
+// SetStyle is a function that converts text with code to defined code style.
+/* Set text like a code style to on from FormattingStyle (Camel, Snake, or Kebab)
+ * conversion of abbreviations like JSON, USB, and so on is going like a regular text
+ * for current version, therefore they these abbreviations could be in a different
+ * case after conversion.
+ * Case settings apply in the following order : 1 - textCase, 2 - firstSymbol.
+ * If you are not applying textCase to text converting from Camel to Snake or Kebab
+ * result is lower case styled text. textCase does not apply to Camel style.
+ * Parameters:
+ * - text - pointer to text
+ * - style - new code style
+ * - firstSymbol - case settings for first symbol
+ * - textCase - case settings for whole text except first symbol
+ * Returns : new string with formatted line
+ */
+func SetStyle(text *string, style FormattingStyle, firstSymbol CaseSetting, textCase CaseSetting) string {
 	if text == nil {
 		return ""
 	}
@@ -65,10 +87,32 @@ func SetStyle(text *string, style FormattingStyle) string {
 		}
 	}
 	sb.WriteString((*text)[startIndex:])
+	result := strings.Builder{}
 	if style != Camel {
-		return strings.ToLower(sb.String())
+		switch textCase {
+		case ToUpper:
+			result.WriteString(strings.ToUpper(sb.String()[1:]))
+			break
+		case ToLower:
+			result.WriteString(strings.ToLower(sb.String()[1:]))
+			break
+		case NoChanges:
+			result.WriteString(sb.String()[1:])
+			break
+		}
+	} else {
+		result.WriteString(sb.String()[1:])
+	}
+
+	switch firstSymbol {
+	case ToUpper:
+		return strings.ToUpper(sb.String()[:1]) + result.String()
+	case ToLower:
+		return strings.ToLower(sb.String()[:1]) + result.String()
+	case NoChanges:
+		return sb.String()[:1] + result.String()
 	}
-	return sb.String()
+	return ""
 }
 
 // defineFormattingStyle

stringstyle_formatter_test.go

@@ -6,7 +6,7 @@ import (
 	"testing"
 )
 
-func TestSetFormattingStyle(t *testing.T) {
+func TestSetFormattingStyleWithoutCaseModification(t *testing.T) {
 	for name, test := range map[string]struct {
 		text     string
 		expected string
@@ -24,7 +24,7 @@ func TestSetFormattingStyle(t *testing.T) {
 		},
 		"lower-case-camel-to-snake-simple": {
 			text:     "mySuperFunc",
-			expected: "my_super_func",
+			expected: "my_Super_Func",
 			newStyle: stringFormatter.Snake,
 		},
 		"snake-to-camel-simple": {
@@ -34,7 +34,7 @@ func TestSetFormattingStyle(t *testing.T) {
 		},
 		"camel-to-snake-with-underscore-the-end": {
 			text:     "myVal_",
-			expected: "my_val_",
+			expected: "my_Val_",
 			newStyle: stringFormatter.Snake,
 		},
 		"mixed-to-camel-simple": {
@@ -54,12 +54,13 @@ func TestSetFormattingStyle(t *testing.T) {
 		},
 		"camel-with_abbreviation-to-snake": {
 			text:     "convertToJSON",
-			expected: "convert_to_json",
+			expected: "convert_To_JSON",
 			newStyle: stringFormatter.Snake,
 		},
 	} {
 		t.Run(name, func(t *testing.T) {
-			actual := stringFormatter.SetStyle(&test.text, test.newStyle)
+			actual := stringFormatter.SetStyle(&test.text, test.newStyle, stringFormatter.NoChanges,
+				stringFormatter.NoChanges)
 			assert.Equal(t, test.expected, actual)
 		})
 	}