phodal/coca

package cmd

import (
    "encoding/json"
    "github.com/modernizing/coca/cmd/cmd_util"
    bs2 "github.com/modernizing/coca/pkg/application/bs"
    "github.com/modernizing/coca/pkg/domain/bs_domain"
    "github.com/modernizing/coca/pkg/infrastructure/string_helper"
    "github.com/spf13/cobra"
    "strings"
)

type BsCmdConfig struct {
    Path string
}

var (
    bsCmdConfig BsCmdConfig
)

var badsmellCmd = &cobra.Command{
    Use:   "bs",
    Short: "generate bad smell list and suggestions",
    Long:  ``,
    Run: func(cmd *cobra.Command, args []string) {
        importPath := bsCmdConfig.Path
        ignoreStr := cmd.Flag("ignore").Value.String()
        sortType := cmd.Flag("sort").Value.String()

        ignoreRules := strings.Split(ignoreStr, ",")

        bsApp := *bs2.NewBadSmellApp()
        nodeInfos := bsApp.AnalysisPath(importPath)

        bsModel, _ := json.MarshalIndent(nodeInfos, "", "\t")
        cmd_util.WriteToCocaFile("nodeInfos.json", string(bsModel))

        filterBs := bsApp.IdentifyBadSmell(nodeInfos, ignoreRules)
        filterBsModel, _ := json.MarshalIndent(filterBs, "", "\t")

        if sortType == "type" {
            sortSmells := bs_domain.SortSmellByType(filterBs, isSmellHaveSize)
            filterBsModel, _ = json.MarshalIndent(sortSmells, "", "\t")
        }

        cmd_util.WriteToCocaFile("bs.json", string(filterBsModel))
    },
}

func isSmellHaveSize(key string) bool {
    var smellList = []string{
        "largeClass",
        "repeatedSwitches",
        "longParameterList",
        "longMethod",
        "dataClass",
    }
    return string_helper.StringArrayContains(smellList, key)
}

func init() {
    rootCmd.AddCommand(badsmellCmd)

    badsmellCmd.PersistentFlags().StringVarP(&bsCmdConfig.Path, "path", "p", ".", "example -p core/main")
    badsmellCmd.PersistentFlags().StringP("ignore", "x", "", "-x=dataClass,lazyElement,longMethod,refusedBequest")
    badsmellCmd.PersistentFlags().StringP("sort", "s", "", "sort bad smell -s=type")
}