first commit

.gitignore

@@ -0,0 +1 @@
+.idea

go.mod

@@ -0,0 +1,16 @@
+module ja
+
+go 1.20
+
+require (
+	github.com/spf13/cobra v1.8.1
+	github.com/stretchr/testify v1.9.0
+)
+
+require (
+	github.com/davecgh/go-spew v1.1.1 // indirect
+	github.com/inconshreveable/mousetrap v1.1.0 // indirect
+	github.com/pmezard/go-difflib v1.0.0 // indirect
+	github.com/spf13/pflag v1.0.5 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)

go.sum

@@ -0,0 +1,18 @@
+github.com/cpuguy83/go-md2man/v2 v2.0.4/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
+github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
+github.com/spf13/cobra v1.8.1 h1:e5/vxKd/rZsfSJMUX1agtjeTDf+qv1/JdBF8gg5k9ZM=
+github.com/spf13/cobra v1.8.1/go.mod h1:wHxEcudfqmLYa8iTfL+OuZPbBZkmvliBWKIezN3kD9Y=
+github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
+github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
+github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
+github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

init.go

@@ -0,0 +1,23 @@
+package main
+
+import (
+	"os"
+	"path/filepath"
+)
+
+// https://gitee.com/ja-netfilter/ja-netfilter/releases/download/2022.2.0/ja-netfilter-2022.2.0.zip
+
+func CreateWorkDir() {
+	homeDir, err := os.UserHomeDir()
+	if err != nil {
+		panic("get user home dir failed: " + err.Error())
+	}
+
+	dir := filepath.Join(homeDir, ".ja")
+	if _, err := os.Stat(dir); os.IsNotExist(err) {
+		err = os.Mkdir(dir, 0755)
+		if err != nil {
+			panic("create workdir failed: " + err.Error())
+		}
+	}
+}

init_test.go

@@ -0,0 +1,17 @@
+package main
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCreateWorkDir(t *testing.T) {
+	homeDir, err := os.UserHomeDir()
+	assert.NoError(t, err)
+	t.Logf("homeDir: %s", homeDir)
+	join := filepath.Join(homeDir, ".ja")
+	t.Logf("join: %s", join)
+}

logger.go

@@ -0,0 +1,22 @@
+package main
+
+import (
+	"errors"
+	"fmt"
+	"os"
+)
+
+var logger = &Logger{}
+
+type Logger struct{}
+
+func (l *Logger) Printf(format string, args ...any) {
+	msg := fmt.Sprintf(format+"\n", args...)
+	_, _ = fmt.Fprint(os.Stdout, msg)
+}
+
+func (l *Logger) Errorf(format string, args ...any) error {
+	msg := fmt.Sprintf(format+"\n", args...)
+	_, _ = fmt.Fprint(os.Stderr, msg)
+	return errors.New(msg)
+}

main.go

@@ -0,0 +1,79 @@
+package main
+
+import (
+	"github.com/spf13/cobra"
+	"os"
+	"strings"
+)
+
+var ja *cobra.Command
+
+func init() {
+	var vmOptionDir string
+
+	ja = &cobra.Command{
+		Use:     "ja",
+		Short:   "jetbrains activation",
+		Example: "ja -d C:/Users/xuthu/AppData/Roaming/JetBrains",
+		PreRun: func(cmd *cobra.Command, args []string) {
+			// create home dir
+			CreateWorkDir()
+			// download ja-netfilter
+			DownloadJaNetfilter()
+			// unzip ja-netfilter
+			UnzipJaNetfilter()
+		},
+		Run: func(cmd *cobra.Command, args []string) {
+			dirEntries, err := os.ReadDir(vmOptionDir)
+			if err != nil {
+				panic("read dir failed: " + err.Error())
+			}
+			for _, entry := range dirEntries {
+				if !entry.IsDir() {
+					continue
+				}
+				if strings.Contains(entry.Name(), "CLion") {
+					InjectVmOption(entry.Name(), "clion64.exe.vmoptions", vmOptionDir)
+					continue
+				}
+				if strings.Contains(entry.Name(), "GoLand") {
+					InjectVmOption(entry.Name(), "goland64.exe.vmoptions", vmOptionDir)
+					continue
+				}
+				if strings.Contains(entry.Name(), "PyCharm") {
+					InjectVmOption(entry.Name(), "pycharm64.exe.vmoptions", vmOptionDir)
+					continue
+				}
+				if strings.Contains(entry.Name(), "WebStorm") {
+					InjectVmOption(entry.Name(), "webstorm64.exe.vmoptions", vmOptionDir)
+					continue
+				}
+				if strings.Contains(entry.Name(), "RustRover") {
+					InjectVmOption(entry.Name(), "rustrover64.exe.vmoptions", vmOptionDir)
+					continue
+				}
+				if strings.Contains(entry.Name(), "DataGrip") {
+					InjectVmOption(entry.Name(), "datagrip64.exe.vmoptions", vmOptionDir)
+					continue
+				}
+				if strings.Contains(entry.Name(), "IntelliJIdea") {
+					InjectVmOption(entry.Name(), "idea64.exe.vmoptions", vmOptionDir)
+					continue
+				}
+				if strings.Contains(entry.Name(), "PhpStorm") {
+					InjectVmOption(entry.Name(), "PhpStorm64.exe.vmoptions", vmOptionDir)
+					continue
+				}
+			}
+
+			logger.Printf("jetbrains activation success, access: https://jbls.ide-soft.com/")
+		},
+	}
+	ja.Flags().StringVarP(&vmOptionDir, "dir", "d", DefaultVmOptionDir(), "jetbrains vm option dir")
+}
+
+func main() {
+	if err := ja.Execute(); err != nil {
+		logger.Printf("running command failed: %v", err)
+	}
+}

utils.go

@@ -0,0 +1,148 @@
+package main
+
+import (
+	"archive/zip"
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"path/filepath"
+	"runtime"
+	"strings"
+)
+
+func GetHomeDir() string {
+	homeDir, err := os.UserHomeDir()
+	if err != nil {
+		panic("get user home dir failed: " + err.Error())
+	}
+	return homeDir
+}
+
+func GetWorkDir() string {
+	homeDir := GetHomeDir()
+	return filepath.Join(homeDir, ".ja")
+}
+
+func DefaultVmOptionDir() string {
+	homeDir := GetHomeDir()
+	if runtime.GOOS == "windows" {
+		return filepath.Join(homeDir, "AppData/Roaming/JetBrains")
+	}
+	return filepath.Join(homeDir, ".config/JetBrains")
+}
+
+func DownloadJaNetfilter() {
+	homeDir := GetWorkDir()
+	filename := filepath.Join(homeDir, "ja-netfilter-2022.2.0.zip")
+	if _, err := os.Stat(filename); os.IsNotExist(err) {
+		url := "https://gitee.com/ja-netfilter/ja-netfilter/releases/download/2022.2.0/ja-netfilter-2022.2.0.zip"
+		err := downloadFile(url, filename)
+		if err != nil {
+			panic("download ja-netfilter failed: " + err.Error())
+		}
+	}
+}
+
+func UnzipJaNetfilter() {
+	homeDir := GetWorkDir()
+	jaNetfilterDir := filepath.Join(homeDir, "ja-netfilter")
+	if _, err := os.Stat(jaNetfilterDir); os.IsNotExist(err) {
+		filename := filepath.Join(homeDir, "ja-netfilter-2022.2.0.zip")
+		reader, err := zip.OpenReader(filename)
+		if err != nil {
+			panic("open zip failed: " + err.Error())
+		}
+		defer reader.Close()
+
+		for _, file := range reader.File {
+			filePath := filepath.Join(homeDir, file.Name)
+			if !strings.HasPrefix(filePath, filepath.Clean(homeDir)+string(os.PathSeparator)) {
+				panic(fmt.Sprintf("illegal file path: %s", filePath))
+			}
+
+			if file.FileInfo().IsDir() {
+				if err := os.MkdirAll(filePath, os.ModePerm); err != nil {
+					panic("create dir failed: " + err.Error())
+				}
+				continue
+			}
+
+			if err := os.MkdirAll(filepath.Dir(filePath), os.ModePerm); err != nil {
+				panic("create dir failed: " + err.Error())
+			}
+
+			if err := unzipFile(file, filePath); err != nil {
+				panic("unzip file failed: " + err.Error())
+			}
+		}
+	}
+}
+
+func unzipFile(f *zip.File, fpath string) error {
+	rc, err := f.Open()
+	if err != nil {
+		return err
+	}
+	defer rc.Close()
+
+	outFile, err := os.OpenFile(fpath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
+	if err != nil {
+		return err
+	}
+	defer outFile.Close()
+
+	_, err = io.Copy(outFile, rc)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func downloadFile(url string, filepath string) error {
+	// 创建HTTP请求
+	resp, err := http.Get(url)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	// 创建文件
+	out, err := os.Create(filepath)
+	if err != nil {
+		return err
+	}
+	defer out.Close()
+
+	// 将HTTP响应的内容写入文件
+	_, err = io.Copy(out, resp.Body)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func InjectVmOption(dirname, vmOptionFilename, vmOptionDir string) {
+	logger.Printf("inject ide name: %s, vm option: %s", dirname, filepath.Join(vmOptionDir, dirname, vmOptionFilename))
+	var jaNetfilterFilename = "-javaagent:" + filepath.Join(GetWorkDir(), "ja-netfilter", "ja-netfilter.jar")
+	vmFile := filepath.Join(vmOptionDir, dirname, vmOptionFilename)
+	readFile, err := os.ReadFile(vmFile)
+	if err != nil {
+		panic("read vm file failed: " + err.Error())
+	}
+	var vmString = string(readFile)
+	if !strings.Contains(vmString, "--add-opens=java.base/jdk.internal.org.objectweb.asm=ALL-UNNAMED") {
+		vmString += "\n--add-opens=java.base/jdk.internal.org.objectweb.asm=ALL-UNNAMED"
+	}
+	if !strings.Contains(vmString, "--add-opens=java.base/jdk.internal.org.objectweb.asm.tree=ALL-UNNAMED") {
+		vmString += "\n--add-opens=java.base/jdk.internal.org.objectweb.asm.tree=ALL-UNNAMED"
+	}
+	if !strings.Contains(vmString, jaNetfilterFilename) {
+		vmString += "\n" + jaNetfilterFilename
+	}
+	err = os.WriteFile(vmFile, []byte(vmString), 0644)
+	if err != nil {
+		panic("write vm file failed: " + err.Error())
+	}
+}