com.zeoflow.depot.processor.EntityOrViewProcessor.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of depot-compiler Show documentation

The Depot persistence library provides an abstraction layer over SQLite to allow for more robust database access while using the full power of SQLite.

The newest version!

/*
 * Copyright (C) 2021 ZeoFlow SRL
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.zeoflow.depot.processor

import com.zeoflow.depot.compiler.processing.XTypeElement
import com.zeoflow.depot.vo.EntityOrView
import com.zeoflow.depot.vo.Fields
import com.squareup.javapoet.TypeName

interface EntityOrViewProcessor {
    fun process(): EntityOrView
}

/**
 * A no-op implementation of [EntityOrViewProcessor] that just prints a processor error for use of
 * an invalid class as [EntityOrView].
 */
private class NonEntityOrViewProcessor(
    val context: Context,
    val element: XTypeElement,
    private val referenceStack: LinkedHashSet
) : EntityOrViewProcessor {

    override fun process(): EntityOrView {
        context.logger.e(element, ProcessorErrors.NOT_ENTITY_OR_VIEW)
        // Parse this as a Pojo in case there are more errors.
        PojoProcessor.createFor(
            context = context,
            element = element,
            bindingScope = FieldProcessor.BindingScope.READ_FROM_CURSOR,
            parent = null,
            referenceStack = referenceStack
        ).process()
        return object : EntityOrView {
            override val fields: Fields = Fields()
            override val tableName: String
                get() = typeName.toString()
            override val typeName: TypeName
                get() = element.type.typeName
        }
    }
}

@Suppress("FunctionName")
fun EntityOrViewProcessor(
    context: Context,
    element: XTypeElement,
    referenceStack: LinkedHashSet = LinkedHashSet()
): EntityOrViewProcessor {
    return when {
        element.hasAnnotation(com.zeoflow.depot.Entity::class) ->
            EntityProcessor(context, element, referenceStack)
        element.hasAnnotation(com.zeoflow.depot.DatabaseView::class) ->
            DatabaseViewProcessor(context, element)
        else ->
            NonEntityOrViewProcessor(context, element, referenceStack)
    }
}