<commit_before>/*
 * Copyright (c) 2017. tangzx(<EMAIL>)
 *
 * 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.tang.intellij.lua.psi.search

import com.intellij.openapi.project.DumbService
import com.intellij.openapi.util.Ref
import com.intellij.util.Processor
import com.intellij.util.QueryExecutor
import com.tang.intellij.lua.comment.psi.LuaDocClassDef
import com.tang.intellij.lua.stubs.index.LuaSuperClassIndex

/**
 * LuaClassInheritorsSearchExecutor
 * Created by tangzx on 2017/3/28.
 */
class LuaClassInheritorsSearchExecutor : QueryExecutor<LuaDocClassDef, LuaClassInheritorsSearch.SearchParameters> {

    private fun processInheritors(searchParameters: LuaClassInheritorsSearch.SearchParameters, typeName: String, processor: Processor<LuaDocClassDef>): Boolean {
        var ret = true
        val processed = mutableListOf<LuaDocClassDef>()
        LuaSuperClassIndex.process(typeName, searchParameters.project, searchParameters.searchScope, Processor {
            processed.add(it)
            ret = processor.process(it)
            ret
        })
        if (ret && searchParameters.isDeep) {
            for (def in processed) {
                ret = processInheritors(searchParameters, def.name, processor)
                if (!ret) break
            }
        }
        return ret
    }

    override fun execute(searchParameters: LuaClassInheritorsSearch.SearchParameters, processor: Processor<LuaDocClassDef>): Boolean {
        val ref = Ref.create<Boolean>()
        DumbService.getInstance(searchParameters.project).runReadActionInSmartMode { ref.set(processInheritors(searchParameters, searchParameters.typeName, processor)) }
        return ref.get()
    }
}
<commit_msg>bug fix #87<commit_after>/*
 * Copyright (c) 2017. tangzx(<EMAIL>)
 *
 * 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.tang.intellij.lua.psi.search

import com.intellij.openapi.project.DumbService
import com.intellij.util.Processor
import com.intellij.util.QueryExecutor
import com.tang.intellij.lua.comment.psi.LuaDocClassDef
import com.tang.intellij.lua.stubs.index.LuaSuperClassIndex

/**
 * LuaClassInheritorsSearchExecutor
 * Created by tangzx on 2017/3/28.
 */
class LuaClassInheritorsSearchExecutor : QueryExecutor<LuaDocClassDef, LuaClassInheritorsSearch.SearchParameters> {

    private fun processInheritors(searchParameters: LuaClassInheritorsSearch.SearchParameters,
                                  typeName: String,
                                  processedNames: MutableSet<String>,
                                  processor: Processor<LuaDocClassDef>): Boolean {
        var ret = true
        // recursion guard!!
        if (!processedNames.add(typeName))
            return ret

        val processed = mutableListOf<LuaDocClassDef>()
        LuaSuperClassIndex.process(typeName, searchParameters.project, searchParameters.searchScope, Processor {
            processed.add(it)
            ret = processor.process(it)
            ret
        })
        if (ret && searchParameters.isDeep) {
            for (def in processed) {
                ret = processInheritors(searchParameters, def.name, processedNames, processor)
                if (!ret) break
            }
        }
        return ret
    }

    override fun execute(searchParameters: LuaClassInheritorsSearch.SearchParameters, processor: Processor<LuaDocClassDef>): Boolean {
        var ref = true
        DumbService.getInstance(searchParameters.project).runReadActionInSmartMode {
            ref = processInheritors(searchParameters, searchParameters.typeName, mutableSetOf(), processor)
        }
        return ref
    }
}

{'fraction_non_alphanumeric': 0.08277789722640293, 'fraction_numerical': 0.006880240808428295, 'mean_word_length': 3.157283288650581, 'pattern_counts': {'":': 0, '<': 15, '<?xml version=': 0, '>': 15, 'https://': 0, 'lorem ipsum': 0, 'www.': 2, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 3, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '5047429', 'n_tokens_mistral': 1285, 'n_tokens_neox': 1245, 'n_words': 335}