types/enum/ide.go - main - githubmirror/gitness

types/enum/ide.go
89 строк2 KB
Enver Biševac
feat: [CODE-4627]: add modernize tool (#4668)
29 окт 2025, 17:06
29 окт 2025, 17:069e6cc38
// Copyright 2023 Harness, Inc.
//
// 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 enum

import (
	"encoding/json"
	"fmt"
)

type IDEType string

func (i IDEType) Enum() []any {
	return toInterfaceSlice(ideTypes)
}

func (i *IDEType) String() string {
	if i == nil {
		return ""
	}
	return string(*i)
}

var ideTypes = []IDEType{IDETypeVSCode, IDETypeVSCodeWeb, IDETypeCursor, IDETypeWindsurf,
	IDETypeIntelliJ, IDETypePyCharm, IDETypeGoland, IDETypeWebStorm, IDETypeCLion, IDETypePHPStorm,
	IDETypeRubyMine, IDETypeRider}

var jetBrainsIDESet = map[IDEType]struct{}{
	IDETypeIntelliJ: {},
	IDETypePyCharm:  {},
	IDETypeGoland:   {},
	IDETypeWebStorm: {},
	IDETypeCLion:    {},
	IDETypePHPStorm: {},
	IDETypeRubyMine: {},
	IDETypeRider:    {},
}

const (
	IDETypeVSCode    IDEType = "vs_code"
	IDETypeVSCodeWeb IDEType = "vs_code_web"
	// AI-based IDEs.
	IDETypeCursor   IDEType = "cursor"
	IDETypeWindsurf IDEType = "windsurf"
	// all jetbrains IDEs.
	IDETypeIntelliJ IDEType = "intellij"
	IDETypePyCharm  IDEType = "pycharm"
	IDETypeGoland   IDEType = "goland"
	IDETypeWebStorm IDEType = "webstorm"
	IDETypeCLion    IDEType = "clion"
	IDETypePHPStorm IDEType = "phpstorm"
	IDETypeRubyMine IDEType = "rubymine"
	IDETypeRider    IDEType = "rider"
)

func IsJetBrainsIDE(t IDEType) bool {
	_, exist := jetBrainsIDESet[t]
	return exist
}

func (i *IDEType) UnmarshalJSON(data []byte) error {
	var s string
	if err := json.Unmarshal(data, &s); err != nil {
		return err
	}
	// Accept empty IDE type without failing JSON decode
	if s == "" {
		*i = ""
		return nil
	}
	for _, v := range ideTypes {
		if IDEType(s) == v {
			*i = v
			return nil
		}
	}
	return fmt.Errorf("invalid IDEType: %s", s)
}