♻️ Refactor synchronization service

internal/syncer/resource/keybinding_adapter.go

@@ -0,0 +1,135 @@
+package resource
+
+import (
+	"context"
+	"fmt"
+	"voidraft/internal/models/ent"
+	"voidraft/internal/models/ent/keybinding"
+	"voidraft/internal/syncer/snapshot"
+)
+
+// KeyBindingAdapter 负责快捷键资源的快照导入导出。
+type KeyBindingAdapter struct {
+	client *ent.Client
+}
+
+// NewKeyBindingAdapter 创建快捷键适配器。
+func NewKeyBindingAdapter(client *ent.Client) *KeyBindingAdapter {
+	return &KeyBindingAdapter{client: client}
+}
+
+// Kind 返回适配器负责的资源类型。
+func (a *KeyBindingAdapter) Kind() string {
+	return "keybindings"
+}
+
+// Export 导出快捷键快照记录。
+func (a *KeyBindingAdapter) Export(ctx context.Context) ([]snapshot.Record, error) {
+	keyBindings, err := a.client.KeyBinding.Query().Order(keybinding.ByUUID()).All(exportContext(ctx))
+	if err != nil {
+		return nil, err
+	}
+
+	records := make([]snapshot.Record, 0, len(keyBindings))
+	for _, item := range keyBindings {
+		values := map[string]interface{}{
+			keybinding.FieldUUID:           item.UUID,
+			keybinding.FieldCreatedAt:      item.CreatedAt,
+			keybinding.FieldUpdatedAt:      item.UpdatedAt,
+			keybinding.FieldName:           item.Name,
+			keybinding.FieldType:           item.Type,
+			keybinding.FieldKey:            item.Key,
+			keybinding.FieldMacos:          item.Macos,
+			keybinding.FieldWindows:        item.Windows,
+			keybinding.FieldLinux:          item.Linux,
+			keybinding.FieldExtension:      item.Extension,
+			keybinding.FieldEnabled:        item.Enabled,
+			keybinding.FieldPreventDefault: item.PreventDefault,
+			keybinding.FieldScope:          item.Scope,
+		}
+		if item.DeletedAt != nil {
+			values[keybinding.FieldDeletedAt] = *item.DeletedAt
+		}
+
+		record, err := snapshot.NewRecord(a.Kind(), item.UUID, values, nil)
+		if err != nil {
+			return nil, fmt.Errorf("build keybinding record %s: %w", item.UUID, err)
+		}
+		records = append(records, record)
+	}
+
+	return records, nil
+}
+
+// Apply 将快照记录应用到本地快捷键表。
+func (a *KeyBindingAdapter) Apply(ctx context.Context, records []snapshot.Record) error {
+	applyCtx := importContext(ctx)
+
+	for _, record := range records {
+		found, err := a.client.KeyBinding.Query().Where(keybinding.UUIDEQ(record.ID)).First(applyCtx)
+		switch {
+		case ent.IsNotFound(err):
+			if err := a.create(applyCtx, record); err != nil {
+				return err
+			}
+		case err != nil:
+			return err
+		default:
+			if shouldApplyRecord(found.UpdatedAt, record) {
+				if err := a.update(applyCtx, found.ID, record); err != nil {
+					return err
+				}
+			}
+		}
+	}
+
+	return nil
+}
+
+// create 创建新的快捷键记录。
+func (a *KeyBindingAdapter) create(ctx context.Context, record snapshot.Record) error {
+	builder := a.client.KeyBinding.Create().
+		SetUUID(record.ID).
+		SetName(stringValue(record, keybinding.FieldName)).
+		SetType(stringValue(record, keybinding.FieldType)).
+		SetKey(stringValue(record, keybinding.FieldKey)).
+		SetMacos(stringValue(record, keybinding.FieldMacos)).
+		SetWindows(stringValue(record, keybinding.FieldWindows)).
+		SetLinux(stringValue(record, keybinding.FieldLinux)).
+		SetExtension(stringValue(record, keybinding.FieldExtension)).
+		SetEnabled(boolValue(record, keybinding.FieldEnabled)).
+		SetPreventDefault(boolValue(record, keybinding.FieldPreventDefault)).
+		SetScope(stringValue(record, keybinding.FieldScope)).
+		SetCreatedAt(stringValue(record, keybinding.FieldCreatedAt)).
+		SetUpdatedAt(stringValue(record, keybinding.FieldUpdatedAt))
+
+	if deletedAt := recordDeletedAtString(record); deletedAt != nil {
+		builder.SetDeletedAt(*deletedAt)
+	}
+
+	return builder.Exec(ctx)
+}
+
+// update 更新已有快捷键记录。
+func (a *KeyBindingAdapter) update(ctx context.Context, id int, record snapshot.Record) error {
+	builder := a.client.KeyBinding.UpdateOneID(id).
+		SetName(stringValue(record, keybinding.FieldName)).
+		SetType(stringValue(record, keybinding.FieldType)).
+		SetKey(stringValue(record, keybinding.FieldKey)).
+		SetMacos(stringValue(record, keybinding.FieldMacos)).
+		SetWindows(stringValue(record, keybinding.FieldWindows)).
+		SetLinux(stringValue(record, keybinding.FieldLinux)).
+		SetExtension(stringValue(record, keybinding.FieldExtension)).
+		SetEnabled(boolValue(record, keybinding.FieldEnabled)).
+		SetPreventDefault(boolValue(record, keybinding.FieldPreventDefault)).
+		SetScope(stringValue(record, keybinding.FieldScope)).
+		SetUpdatedAt(stringValue(record, keybinding.FieldUpdatedAt))
+
+	if deletedAt := recordDeletedAtString(record); deletedAt != nil {
+		builder.SetDeletedAt(*deletedAt)
+	} else {
+		builder.ClearDeletedAt()
+	}
+
+	return builder.Exec(ctx)
+}