66 "fmt"
77 "github.com/flipped-aurora/gin-vue-admin/server/utils/ast"
88 "github.com/pkg/errors"
9+ "os"
910 "path"
1011 "path/filepath"
1112 "strconv"
@@ -17,7 +18,7 @@ import (
1718 model "github.com/flipped-aurora/gin-vue-admin/server/model/system"
1819 request "github.com/flipped-aurora/gin-vue-admin/server/model/system/request"
1920 "github.com/flipped-aurora/gin-vue-admin/server/utils"
20-
21+ "github.com/tidwall/sjson"
2122 "go.uber.org/zap"
2223)
2324
@@ -79,12 +80,34 @@ func (s *autoCodeHistory) RollBack(ctx context.Context, info request.SysAutoHist
7980 if err != nil {
8081 global .GVA_LOG .Error ("ClearTag DeleteApiByIds:" , zap .Error (err ))
8182 }
83+ localPath := path .Join (global .GVA_CONFIG .AutoCode .Root , global .GVA_CONFIG .AutoCode .Server , "resource" , "lang" )
84+ files , err := s .readJSONFiles (localPath )
85+ if err != nil {
86+ return err
87+ }
88+ for _ , file := range files {
89+ err := s .reWriteI18nJson (file , "api" , history .Package , history .StructName )
90+ if err != nil {
91+ return err
92+ }
93+ }
8294 } // 清除API表
8395 if info .DeleteMenu {
8496 err = BaseMenuServiceApp .DeleteBaseMenu (int (history .MenuID ))
8597 if err != nil {
8698 return errors .Wrap (err , "删除菜单失败!" )
8799 }
100+ localPath := path .Join (global .GVA_CONFIG .AutoCode .Root , global .GVA_CONFIG .AutoCode .Server , "resource" , "lang" )
101+ files , err := s .readJSONFiles (localPath )
102+ if err != nil {
103+ return err
104+ }
105+ for _ , file := range files {
106+ err := s .reWriteI18nJson (file , "menu" , history .Package , history .StructName )
107+ if err != nil {
108+ return err
109+ }
110+ }
88111 } // 清除菜单表
89112 if info .DeleteTable {
90113 err = s .DropTable (history .BusinessDB , history .Table )
@@ -178,6 +201,17 @@ func (s *autoCodeHistory) RollBack(ctx context.Context, info request.SysAutoHist
178201 if err != nil {
179202 return errors .Wrap (err , "更新失败!" )
180203 }
204+ localPath := path .Join (global .GVA_CONFIG .AutoCode .Root , global .GVA_CONFIG .AutoCode .Web , "locales" )
205+ files , err := s .readJSONFiles (localPath )
206+ if err != nil {
207+ return err
208+ }
209+ for _ , file := range files {
210+ err := s .reWriteI18nJson (file , "web" , history .Package , history .StructName )
211+ if err != nil {
212+ return err
213+ }
214+ }
181215 return nil
182216}
183217
@@ -215,3 +249,75 @@ func (s *autoCodeHistory) DropTable(BusinessDb, tableName string) error {
215249 return global .GVA_DB .Exec ("DROP TABLE " + tableName ).Error
216250 }
217251}
252+
253+ func (s * autoCodeHistory ) readJSONFiles (dir string ) ([]string , error ) {
254+ var files []string
255+ err := filepath .Walk (dir , func (path string , info os.FileInfo , err error ) error {
256+ if err != nil {
257+ return err
258+ }
259+ if ! info .IsDir () && filepath .Ext (path ) == ".json" {
260+ files = append (files , path )
261+ }
262+ return nil
263+ })
264+ if err != nil {
265+ return nil , err
266+ }
267+ return files , nil
268+ }
269+
270+ func (s * autoCodeHistory ) reWriteI18nJson (file string , flag string , packageName , structName string ) error {
271+ data , err := os .ReadFile (file )
272+ if err != nil {
273+ return err
274+ }
275+ originJson := string (data )
276+ switch flag {
277+ case "api" :
278+ var err error
279+ originJson , err = sjson .Delete (originJson , fmt .Sprintf ("system.api.group.%s%s" , packageName , structName ))
280+ if err != nil {
281+ return err
282+ }
283+ originJson , err = sjson .Delete (originJson , fmt .Sprintf ("system.api.desc.add%s%s" , packageName , structName ))
284+ if err != nil {
285+ return err
286+ }
287+ originJson , err = sjson .Delete (originJson , fmt .Sprintf ("system.api.desc.delete%s%s" , packageName , structName ))
288+ if err != nil {
289+ return err
290+ }
291+ originJson , err = sjson .Delete (originJson , fmt .Sprintf ("system.api.desc.batch%s%s" , packageName , structName ))
292+ if err != nil {
293+ return err
294+ }
295+ originJson , err = sjson .Delete (originJson , fmt .Sprintf ("system.api.desc.update%s%s" , packageName , structName ))
296+ if err != nil {
297+ return err
298+ }
299+ originJson , err = sjson .Delete (originJson , fmt .Sprintf ("system.api.desc.find%s%s" , packageName , structName ))
300+ if err != nil {
301+ return err
302+ }
303+ originJson , err = sjson .Delete (originJson , fmt .Sprintf ("system.api.desc.list%s%s" , packageName , structName ))
304+ if err != nil {
305+ return err
306+ }
307+ case "menu" :
308+ originJson , err = sjson .Delete (originJson , fmt .Sprintf ("system.menu.%s%s" , packageName , structName ))
309+ if err != nil {
310+ return err
311+ }
312+ case "web" :
313+ originJson , err = sjson .Delete (originJson , fmt .Sprintf ("%s.%s" , packageName , structName ))
314+ if err != nil {
315+ return err
316+ }
317+ }
318+ err = os .WriteFile (file , []byte (originJson ), 0666 )
319+ if err != nil {
320+ return err
321+ }
322+ return nil
323+ }
0 commit comments