feat: Get by name (#6)

* feat: Get by name

Signed-off-by: Carlos A Becker <caarlos0@users.noreply.github.com>

* fix: variant

Signed-off-by: Carlos A Becker <caarlos0@users.noreply.github.com>

Author: caarlos0

main.go

@@ -1,6 +1,9 @@
 package catppuccingo
 
-import "image/color"
+import (
+	"image/color"
+	"strings"
+)
 
 // Flavour is an interface implemented by all Catppuccin variations.
 type Flavour interface {
@@ -49,3 +52,18 @@ func (c Color) RGBA() (r uint32, g uint32, b uint32, a uint32) {
 }
 
 var _ color.Color = Color{}
+
+// Variant returns the Theme variant by name.
+func Variant(flavour string) Theme {
+	for _, t := range []Theme{
+		Mocha,
+		Frappe,
+		Macchiato,
+		Latte,
+	} {
+		if strings.EqualFold(t.Name(), flavour) {
+			return t
+		}
+	}
+	return nil
+}

main_test.go

@@ -1,8 +1,38 @@
 package catppuccingo
 
-import (
-	"testing"
-)
+import "testing"
+
+func TestVariant(t *testing.T) {
+	for s, ss := range map[string]string{
+		"moCha":     Mocha.Name(),
+		"Mocha":     Mocha.Name(),
+		"MOCHA":     Mocha.Name(),
+		"mocha":     Mocha.Name(),
+		"Frappe":    Frappe.Name(),
+		"frappe":    Frappe.Name(),
+		"MaccHiaTo": Macchiato.Name(),
+		"latte":     Latte.Name(),
+	} {
+		t.Run(s, func(t *testing.T) {
+			if Variant(s).Name() != ss {
+				t.Errorf("expected %q, got %q", ss, Variant(s).Name())
+			}
+		})
+	}
+	t.Run("invalid", func(t *testing.T) {
+		v := Variant("invalid")
+		if v != nil {
+			t.Errorf("expected nil, got %v", v)
+		}
+	})
+}
+
+func TestColor(t *testing.T) {
+	r, g, b, a := Mocha.Base().RGBA()
+	if 30 != r || 30 != g || b != 46 || a != 1 {
+		t.Fatalf("expected rgba=%d %d %d %d, got %d %d %d %d", 30, 30, 46, 1, r, g, b, a)
+	}
+}
 
 func TestFlavours(t *testing.T) {
 	for _, flavour := range []Flavour{