package service import ( "context" "errors" "fmt" "next-terminal/server/env" "next-terminal/server/guacd" "next-terminal/server/model" "next-terminal/server/repository" "gorm.io/gorm" ) type propertyService struct { baseService } var deprecatedPropertyNames = []string{ guacd.EnableDrive, guacd.DrivePath, guacd.DriveName, guacd.DisableGlyphCaching, guacd.CreateRecordingPath, } var defaultProperties = map[string]string{ guacd.EnableRecording: "true", guacd.FontName: "menlo", guacd.FontSize: "12", guacd.ColorScheme: "gray-black", guacd.EnableWallpaper: "true", guacd.EnableTheming: "true", guacd.EnableFontSmoothing: "true", guacd.EnableFullWindowDrag: "true", guacd.EnableDesktopComposition: "true", guacd.EnableMenuAnimations: "true", guacd.DisableBitmapCaching: "false", guacd.DisableOffscreenCaching: "false", "cron-log-saved-limit": "360", "login-log-saved-limit": "360", "session-saved-limit": "360", "user-default-storage-size": "5120", } func (service propertyService) InitProperties() error { propertyMap := repository.PropertyRepository.FindAllMap(context.TODO()) for name, value := range defaultProperties { if err := service.CreateIfAbsent(propertyMap, name, value); err != nil { return err } } return nil } func (service propertyService) CreateIfAbsent(propertyMap map[string]string, name, value string) error { if len(propertyMap[name]) == 0 { property := model.Property{ Name: name, Value: value, } return repository.PropertyRepository.Create(context.TODO(), &property) } return nil } func (service propertyService) DeleteDeprecatedProperty() error { propertyMap := repository.PropertyRepository.FindAllMap(context.TODO()) for _, name := range deprecatedPropertyNames { if propertyMap[name] == "" { continue } if err := repository.PropertyRepository.DeleteByName(context.TODO(), name); err != nil { return err } } return nil } func (service propertyService) Update(item map[string]interface{}) error { return env.GetDB().Transaction(func(tx *gorm.DB) error { c := service.Context(tx) for key := range item { value := fmt.Sprintf("%v", item[key]) if value == "" { value = "-" } property := model.Property{ Name: key, Value: value, } if key == "enable-ldap" && value == "false" { if err := UserService.DeleteALlLdapUser(c); err != nil { return err } } _, err := repository.PropertyRepository.FindByName(c, key) if err != nil && errors.Is(err, gorm.ErrRecordNotFound) { if err := repository.PropertyRepository.Create(c, &property); err != nil { return err } } else { if err := repository.PropertyRepository.UpdateByName(c, &property, key); err != nil { return err } } } return nil }) }