dev.pellet.server.extension.PelletBufferExtensions.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pellet-server Show documentation

An opinionated Kotlin web framework, with best-practices built-in

There is a newer version: 0.0.16

package dev.pellet.server.extension

import dev.pellet.server.buffer.PelletBuffer
import java.nio.charset.Charset

internal fun PelletBuffer.trimTrailing(byte: Byte): PelletBuffer {
    val limit = this.limit()

    if (limit < 1) {
        return this
    }
    if (this[limit - 1] == byte) {
        return this.limit(limit - 1)
    }

    return this
}

internal fun PelletBuffer.stringifyAndClear(charset: Charset = Charsets.US_ASCII): String {
    assert(this.position() == 0)
    val string = if (this.hasArray()) {
        String(this.array(), 0, this.limit(), charset)
    } else {
        val byteArray = ByteArray(this.limit())
        this.byteBuffer.get(byteArray, 0, this.limit())
        String(byteArray, 0, this.limit(), charset)
    }
    this.clear()
    return string
}

// Returns the next position of a byte, by ignoring bytes before the current position()
internal fun PelletBuffer.nextPositionOfOrNull(needle: Byte): Int? {
    val position = this.position()
    val limit = this.limit()
    for (i in position.until(limit)) {
        val storedByte = this.get(i)
        if (needle == storedByte) {
            return i
        }
    }
    return null
}

// Advances position by count, up to and including the current limit
internal fun PelletBuffer.advance(count: Int): PelletBuffer {
    val newPosition = this.position() + count
    return if (newPosition > this.limit()) {
        this.position(this.limit())
    } else {
        this.position(newPosition)
    }
}