commit 182d381084b40284268bb3378e240cc6f10f00c0 Author: Young Xu Date: Sun Jul 28 13:58:14 2024 +0800 first commit diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..485dee6 --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +.idea diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..0a294b9 --- /dev/null +++ b/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 +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..4353b61 --- /dev/null +++ b/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= diff --git a/init.go b/init.go new file mode 100644 index 0000000..f9e586d --- /dev/null +++ b/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()) + } + } +} diff --git a/init_test.go b/init_test.go new file mode 100644 index 0000000..c45138f --- /dev/null +++ b/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) +} diff --git a/logger.go b/logger.go new file mode 100644 index 0000000..f8768a1 --- /dev/null +++ b/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) +} diff --git a/main.go b/main.go new file mode 100644 index 0000000..0f16e9c --- /dev/null +++ b/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) + } +} diff --git a/utils.go b/utils.go new file mode 100644 index 0000000..8570d3b --- /dev/null +++ b/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()) + } +}