136 lines
4.5 KiB
Go
136 lines
4.5 KiB
Go
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)
|
|
}
|