jsMain.styled.CssHolder.kt Maven / Gradle / Ivy

Go to download
package styled

import kotlinx.css.CssBuilder
import kotlinx.css.RuleSet
import kotlin.properties.ReadOnlyProperty
import kotlin.reflect.KProperty

abstract class AbstractCssHolder(protected val sheet: StyleSheet, protected vararg val ruleSets: RuleSet) {
    protected val classNamesToInject = mutableSetOf()
    val css by lazy {
        CssBuilder(allowClasses = false).apply {
            [email protected] { it() }
        }
    }

    fun scheduleToInject() {
        classNamesToInject.forEach { className ->
            scheduleToInject(className)
        }
    }

    fun scheduleToInject(className: String) {
        if (classNamesToInject.contains(className)) {
            GlobalStyles.scheduleToInject(".$className", css)
        }
    }

    fun removeInjected() {
        GlobalStyles.removeInjectedStyleSheet(classNamesToInject.map { className -> ".$className" })
    }
}


class CssHolder(sheet: StyleSheet, vararg ruleSets: RuleSet) : AbstractCssHolder(sheet, *ruleSets) {
    /**
     * Used to inject styleSheet properties with the same name but different CSS
     */
    private fun updateClassnameIfClashes(className: ClassName): ClassName {
        var newClassname = className
        var counter = 0
        while (GlobalStyles.usedStyleSheet.contains(".${newClassname}") && GlobalStyles.usedStyleSheet[".${newClassname}"] != sheet) {
            newClassname = className + counter
            counter++
        }
        return newClassname
    }

    operator fun provideDelegate(thisRef: Any?, providingProperty: KProperty<*>): ReadOnlyProperty {
        val className = updateClassnameIfClashes(sheet.getClassName(providingProperty))
        classNamesToInject.add(className)
        GlobalStyles.usedStyleSheet[".$className"] = sheet
        return ReadOnlyProperty { _, property ->
            {
                sheet.scheduleImports()
                if (sheet.isStatic) {
                    scheduleToInject(className)
                    +className
                }
                if (!sheet.isStatic || !allowClasses || isHolder) {
                    styleName.add(sheet.getClassName(property))
                    ruleSets.forEach { it() }
                }
            }
        }
    }
}