phodal/coca

package concept

import (
    languages2 "github.com/modernizing/coca/pkg/application/call/stop_words/languages"
    "github.com/modernizing/coca/pkg/domain/core_domain"
    "github.com/modernizing/coca/pkg/infrastructure/constants"
    "github.com/modernizing/coca/pkg/infrastructure/string_helper"
)

type ConceptAnalyser struct {
}

func NewConceptAnalyser() ConceptAnalyser {
    return ConceptAnalyser{}
}

func (c ConceptAnalyser) Analysis(clzs *[]core_domain.CodeDataStruct) string_helper.PairList {
    return buildMethodsFromDeps(*clzs)
}

func buildMethodsFromDeps(clzs []core_domain.CodeDataStruct) string_helper.PairList {
    var methodsName []string
    var methodStr string
    for _, clz := range clzs {
        for _, method := range clz.Functions {
            methodName := method.Name
            methodsName = append(methodsName, methodName)
            methodStr = methodStr + " " + methodName
        }
    }

    words := SegmentCamelcase(methodsName)
    words = removeNormalWords(words)

    wordCounts := string_helper.SortWord(words)
    return wordCounts
}

func removeNormalWords(words map[string]int) map[string]int {
    var newWords = words
    var stopwords = languages2.ENGLISH_STOP_WORDS
    stopwords = append(stopwords, constants.TechStopWords...)
    for _, normalWord := range stopwords {
        if newWords[normalWord] > 0 {
            delete(newWords, normalWord)
        }
    }

    return newWords
}