Editor notification for missing annotation

SerializedNameNotification.kt

class SerializedNameNotification :
    EditorNotifications.Provider<EditorNotificationPanel>() {

    companion object {
        val KEY =
            Key.create<EditorNotificationPanel>("Add missing annotation?")
    }

    override fun getKey(): Key<EditorNotificationPanel> = KEY

    override fun createNotificationPanel(
        file: VirtualFile,
        fileEditor: FileEditor,
        project: Project
    ): EditorNotificationPanel? {
        val psiFile = file.toPsiFile(project)
        val ktFile = psiFile as? KtFile
        val psiClasses = ktFile?.classes
        psiClasses?.let { psiClassList ->
            psiClassList.forEach { psiClass ->
                (psiClass as? KtLightClassForSourceDeclaration)?.let {
                    ktLightClassForSourceDeclaration ->
                    val ktClass =
                        ktLightClassForSourceDeclaration.kotlinOrigin
                            as? KtClass
                    if (ktClass?.isData() == true) {
                        val param = ktClass.getMissingAnnotationParam()
                        param?.let {
                            removeNotification(fileEditor)

                            return createPanel(
                                psiFile = psiFile,
                                project = project,
                                name = ktClass.name.orEmpty(),
                                onAddClick = {
                                    addAnnotation(ktClass, project)
                                },
                                onIgnoreClick = {
                                    ignoreInspection(
                                        fileEditor,
                                        psiFile,
                                        project
                                    )
                                }
                            )
                        } ?: removeNotification(fileEditor)
                    }
                }
            }
        }
        return null
    }
}