src/main/kotlin/IPAddress.kt


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

/**
 * An IP address (v4 or v6).
 */
class IPAddress {
    private var address: Array<Byte>
    private var is4: Boolean

    /**
     * @throws NumberFormatException if the IP address follows neither the IPv4 syntax, nor the IPv6 syntax.
     */
    constructor(address: String) {
        val split4 = address.split('.')
        val split6 = address.split(':')
        var bytes: List<Byte> = emptyList()

        if (split4.size == 4) {
            bytes = split4.map { it.toByte() }
            this.is4 = true
        } else if (split6.size == 16) {
            bytes = split4.map { it.toByte(16) }
            this.is4 = false
        } else {
            throw NumberFormatException()
        }

        this.address = bytes.toTypedArray()
    }

    override fun toString(): String {
        val str: String
        if (this.is4) {
            str = this.address.joinToString(separator = ".") { it.toString() }
        } else {
            str = this.address.joinToString(separator = ":") { it.toString() }
        }
        return str
    }
}