apps/ja
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 2 Wiki Activity

first commit

Browse Source
This commit is contained in:
Young Xu 2024-07-28 13:58:14 +08:00
commit 182d381084
8 changed files with 324 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored Normal file
View File

@ -0,0 +1 @@
.idea

16
go.mod Normal file
View File

@ -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
)

18
go.sum Normal file
View File

@ -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=

23
init.go Normal file
View File

@ -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())
}
}
}

17
init_test.go Normal file
View File

@ -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)
}

22
logger.go Normal file
View File

@ -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)
}

79
main.go Normal file
View File

@ -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)
}
}

148
utils.go Normal file
View File

@ -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())
}
}