fix(jzero): fix gen when has go.work files

Author: jaronnie

cmd/jzero/internal/command/gen/genapi/patch_handler.go

@@ -282,7 +282,7 @@ func UpdateImportedModule(f *ast.File, fset *token.FileSet, workDir, module stri
 			imports := astutil.Imports(fset, f)
 			for _, imp := range imports {
 				for _, name := range imp {
-					if strings.HasPrefix(name.Path.Value, "\""+rootPkg) {
+					if strings.Contains(name.Path.Value, rootPkg) {
 						unQuote, _ := strconv.Unquote(name.Path.Value)
 						newImp := strings.Replace(unQuote, rootPkg, module, 1)
 						astutil.RewriteImport(fset, f, unQuote, newImp)

cmd/jzero/internal/command/gen/genrpc/patch_logic.go

@@ -176,7 +176,7 @@ func UpdateImportedModule(filepath, workDir, module string) error {
 			imports := astutil.Imports(fset, f)
 			for _, imp := range imports {
 				for _, name := range imp {
-					if strings.HasPrefix(name.Path.Value, "\""+rootPkg) {
+					if strings.Contains(name.Path.Value, rootPkg) {
 						unQuote, _ := strconv.Unquote(name.Path.Value)
 						newImp := strings.Replace(unQuote, rootPkg, module, 1)
 						astutil.RewriteImport(fset, f, unQuote, newImp)