package main

import (
	"fmt"

	"github.com/insomnius/rb"
)

func main() {
	// Working with rb.String
	str := rb.String("Hello, World!")
	fmt.Println("Original:", str)             // Original: Hello, World!
	fmt.Println("Downcased:", str.Downcase()) // Downcased: hello, world!
	fmt.Println("Capitalized:", str.Capitalize()) // Capitalized: Hello, world!
	fmt.Println("Reversed:", str.Reverse())   // Reversed: !dlroW ,olleH

	str.EnforceDowncase()
	fmt.Println("Enforce Downcase:", str) // Enforce Downcase: hello, world!

	// Ruby-style Array methods
	arr := rb.Array[rb.String]{"ruby", "go", "ruby", "python"}
	fmt.Println("Total Elements:", arr.Count())     // Total Elements: 4
	fmt.Println("Count 'ruby':", arr.Count("ruby")) // Count 'ruby': 2
	fmt.Println("Count with Predicate:", arr.Count(func(s rb.String) bool {
		return s.Length() > 2
	})) // Count with Predicate: 4

	// Array transformations
	mapped := arr.Map(func(s rb.String) rb.String {
		return s.Upcase()
	})
	fmt.Println("Mapped:", mapped) // Mapped: [RUBY GO RUBY PYTHON]

	selected := arr.Select(func(s rb.String) bool {
		return s.Length() > 3
	})
	fmt.Println("Selected:", selected) // Selected: [ruby ruby python]

	// Working with rb.Integer
	num := rb.Integer(5)
	fmt.Println("Is Prime:", num.IsPrime())        // Is Prime: true
	fmt.Println("Factorial:", num.Factorial())     // Factorial: 120
	fmt.Println("Divisors:", num.Divisors())       // Divisors: [1 5]

	// Integer iteration
	num.Times(func(i rb.Integer) {
		fmt.Printf("Count: %d\n", i)
	})

	// Working with rb.Float
	pi := rb.Float(3.14159)
	fmt.Println("Ceiled:", pi.Ceil())    // Ceiled: 4
	fmt.Println("Floored:", pi.Floor())  // Floored: 3
	fmt.Println("Rounded:", pi.Round())  // Rounded: 3

	// Working with rb.Boolean
	flag := rb.Boolean(true)
	flag.IfTrue(func() { fmt.Println("It's true!") })
	result := flag.Ternary("yes", "no")
	fmt.Println("Ternary result:", result) // Ternary result: yes

	// Working with rb.Hash
	hash := rb.Hash[rb.String, rb.Integer]{"a": 1, "b": 2, "c": 3}
	fmt.Println("Keys:", hash.Keys())           // Keys: [a b c]
	fmt.Println("Values:", hash.Values())       // Values: [1 2 3]
	fmt.Println("Has key 'a':", hash.HasKey("a")) // Has key 'a': true

	// Working with rb.Range
	rng := rb.NewRange(rb.Integer(1), rb.Integer(5))
	fmt.Println("Range size:", rng.Size())      // Range size: 5
	fmt.Println("Includes 3:", rng.Include(3)) // Includes 3: true
}