From da2f95bfa39dc24a8a12750e28497618c67eb41d Mon Sep 17 00:00:00 2001
From: Rodrigo Graciano <rodrigolgraciano@gmail.com>
Date: Sun, 10 Mar 2019 20:34:19 -0400
Subject: [PATCH] BAEL-2760

---
 .../kotlin/com/baeldung/range/CustomColor.kt  | 56 +++++++++++++++++++
 .../com/baeldung/range/CustomColorTest.kt     | 41 ++++++++++++++
 2 files changed, 97 insertions(+)
 create mode 100644 core-kotlin-2/src/main/kotlin/com/baeldung/range/CustomColor.kt
 create mode 100644 core-kotlin-2/src/test/kotlin/com/baeldung/range/CustomColorTest.kt

diff --git a/core-kotlin-2/src/main/kotlin/com/baeldung/range/CustomColor.kt b/core-kotlin-2/src/main/kotlin/com/baeldung/range/CustomColor.kt
new file mode 100644
index 0000000000..b4fed13b18
--- /dev/null
+++ b/core-kotlin-2/src/main/kotlin/com/baeldung/range/CustomColor.kt
@@ -0,0 +1,56 @@
+package com.baeldung.range
+
+import java.lang.IllegalStateException
+
+class CustomColor(val rgb: Int): Comparable<CustomColor> {
+
+    override fun compareTo(other: CustomColor): Int {
+        return this.rgb.compareTo(other.rgb)
+    }
+
+    operator fun rangeTo(that: CustomColor) = ColorRange(this,that)
+
+    operator fun inc(): CustomColor {
+        return CustomColor(rgb + 1)
+    }
+
+    init {
+        if(rgb < 0x000000 || rgb > 0xFFFFFF){
+            throw IllegalStateException("RGB must be between 0 and 16777215")
+        }
+    }
+
+    override fun toString(): String {
+        return "CustomColor(rgb=$rgb)"
+    }
+}
+class ColorRange(override val start: CustomColor,
+                override val endInclusive: CustomColor) : ClosedRange<CustomColor>, Iterable<CustomColor>{
+
+    override fun iterator(): Iterator<CustomColor> {
+        return ColorIterator(start, endInclusive)
+    }
+}
+
+class ColorIterator(val start: CustomColor, val endInclusive: CustomColor) : Iterator<CustomColor> {
+
+    var initValue = start
+
+    override fun hasNext(): Boolean {
+        return initValue <= endInclusive
+    }
+
+    override fun next(): CustomColor {
+        return initValue++
+    }
+}
+
+fun main(args: Array<String>) {
+    val a = CustomColor(0xABCDEF)
+    val b = CustomColor(-1)
+    val c = CustomColor(0xABCDFF)
+
+    for(color in a..c){
+        println(color)
+    }
+}
\ No newline at end of file
diff --git a/core-kotlin-2/src/test/kotlin/com/baeldung/range/CustomColorTest.kt b/core-kotlin-2/src/test/kotlin/com/baeldung/range/CustomColorTest.kt
new file mode 100644
index 0000000000..8c8795ac42
--- /dev/null
+++ b/core-kotlin-2/src/test/kotlin/com/baeldung/range/CustomColorTest.kt
@@ -0,0 +1,41 @@
+package com.baeldung.range
+
+import org.junit.Test
+import java.lang.IllegalStateException
+import kotlin.test.assertFailsWith
+import kotlin.test.assertTrue
+
+class CustomColorTest {
+
+    @Test
+    fun testInvalidConstructor(){
+        assertFailsWith(IllegalStateException::class){
+            CustomColor(-1)
+        }
+    }
+
+    @Test
+    fun assertHas10Colors(){
+        assertTrue {
+            val a = CustomColor(1)
+            val b = CustomColor(10)
+            val range = a..b
+            for(cc in range){
+                println(cc)
+            }
+            range.toList().size == 10
+        }
+    }
+
+    @Test
+    fun assertContains0xCCCCCC(){
+        assertTrue {
+            val a = CustomColor(0xBBBBBB)
+            val b = CustomColor(0xDDDDDD)
+            val range = a..b
+            range.contains(CustomColor(0xCCCCCC))
+        }
+    }
+
+}
+