From dd581128fe9df5c8eb53dd2dc2fbe6d16b247dc5 Mon Sep 17 00:00:00 2001
From: Young Xu <xuthus5@gmail.com>
Date: Sun, 28 Jul 2024 14:11:45 +0800
Subject: [PATCH] fix: compatible with linux

---
 main.go  | 16 ++++++++--------
 utils.go |  5 +++++
 2 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/main.go b/main.go
index 0f16e9c..3e50602 100644
--- a/main.go
+++ b/main.go
@@ -33,35 +33,35 @@ func init() {
 					continue
 				}
 				if strings.Contains(entry.Name(), "CLion") {
-					InjectVmOption(entry.Name(), "clion64.exe.vmoptions", vmOptionDir)
+					InjectVmOption(entry.Name(), "clion64", vmOptionDir)
 					continue
 				}
 				if strings.Contains(entry.Name(), "GoLand") {
-					InjectVmOption(entry.Name(), "goland64.exe.vmoptions", vmOptionDir)
+					InjectVmOption(entry.Name(), "goland64", vmOptionDir)
 					continue
 				}
 				if strings.Contains(entry.Name(), "PyCharm") {
-					InjectVmOption(entry.Name(), "pycharm64.exe.vmoptions", vmOptionDir)
+					InjectVmOption(entry.Name(), "pycharm64", vmOptionDir)
 					continue
 				}
 				if strings.Contains(entry.Name(), "WebStorm") {
-					InjectVmOption(entry.Name(), "webstorm64.exe.vmoptions", vmOptionDir)
+					InjectVmOption(entry.Name(), "webstorm64", vmOptionDir)
 					continue
 				}
 				if strings.Contains(entry.Name(), "RustRover") {
-					InjectVmOption(entry.Name(), "rustrover64.exe.vmoptions", vmOptionDir)
+					InjectVmOption(entry.Name(), "rustrover64", vmOptionDir)
 					continue
 				}
 				if strings.Contains(entry.Name(), "DataGrip") {
-					InjectVmOption(entry.Name(), "datagrip64.exe.vmoptions", vmOptionDir)
+					InjectVmOption(entry.Name(), "datagrip64", vmOptionDir)
 					continue
 				}
 				if strings.Contains(entry.Name(), "IntelliJIdea") {
-					InjectVmOption(entry.Name(), "idea64.exe.vmoptions", vmOptionDir)
+					InjectVmOption(entry.Name(), "idea64", vmOptionDir)
 					continue
 				}
 				if strings.Contains(entry.Name(), "PhpStorm") {
-					InjectVmOption(entry.Name(), "PhpStorm64.exe.vmoptions", vmOptionDir)
+					InjectVmOption(entry.Name(), "PhpStorm64", vmOptionDir)
 					continue
 				}
 			}
diff --git a/utils.go b/utils.go
index 8570d3b..e70d6a4 100644
--- a/utils.go
+++ b/utils.go
@@ -124,6 +124,11 @@ func downloadFile(url string, filepath string) error {
 }
 
 func InjectVmOption(dirname, vmOptionFilename, vmOptionDir string) {
+	if runtime.GOOS == "windows" {
+		vmOptionFilename += ".exe.vmoptions"
+	} else {
+		vmOptionFilename += ".vmoptions"
+	}
 	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)