113 lines
3.0 KiB
Go
113 lines
3.0 KiB
Go
package injection
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
// Injector interface for text injection
|
|
type Injector interface {
|
|
Inject(ctx context.Context, text string) error
|
|
}
|
|
|
|
// Config for text injection
|
|
type Config struct {
|
|
Mode string // "clipboard", "type", "fallback"
|
|
AlwaysCopyClipboard bool // Always copy to clipboard regardless of mode
|
|
RestoreClipboard bool // Restore original clipboard after injection
|
|
WtypeTimeout time.Duration // Timeout for wtype commands
|
|
ClipboardTimeout time.Duration // Timeout for clipboard operations
|
|
}
|
|
|
|
// DefaultConfig returns sensible defaults for injection
|
|
func DefaultConfig() Config {
|
|
return Config{
|
|
Mode: "fallback",
|
|
AlwaysCopyClipboard: true,
|
|
RestoreClipboard: true,
|
|
WtypeTimeout: 5 * time.Second,
|
|
ClipboardTimeout: 3 * time.Second,
|
|
}
|
|
}
|
|
|
|
// injector implements the Injector interface
|
|
type injector struct {
|
|
config Config
|
|
}
|
|
|
|
// NewInjector creates a new injector with the given config
|
|
func NewInjector(config Config) Injector {
|
|
return &injector{
|
|
config: config,
|
|
}
|
|
}
|
|
|
|
// NewDefaultInjector creates an injector with default configuration
|
|
func NewDefaultInjector() Injector {
|
|
return NewInjector(DefaultConfig())
|
|
}
|
|
|
|
// Inject performs text injection based on the configured mode
|
|
func (i *injector) Inject(ctx context.Context, text string) error {
|
|
if text == "" {
|
|
return fmt.Errorf("cannot inject empty text")
|
|
}
|
|
|
|
// Always copy to clipboard if configured
|
|
var originalClipboard string
|
|
var err error
|
|
|
|
if i.config.AlwaysCopyClipboard || i.config.Mode == "clipboard" || i.config.Mode == "fallback" {
|
|
if err := checkClipboardAvailable(); err != nil {
|
|
return fmt.Errorf("clipboard tools not available: %w", err)
|
|
}
|
|
|
|
if i.config.RestoreClipboard {
|
|
originalClipboard, _ = getClipboard(ctx, i.config.ClipboardTimeout)
|
|
}
|
|
|
|
if err := setClipboard(ctx, text, i.config.ClipboardTimeout); err != nil {
|
|
return fmt.Errorf("failed to copy text to clipboard: %w", err)
|
|
}
|
|
}
|
|
|
|
// Handle different injection modes
|
|
switch i.config.Mode {
|
|
case "clipboard":
|
|
// Already handled above
|
|
return nil
|
|
|
|
case "type":
|
|
err = typeText(ctx, text, i.config.WtypeTimeout)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to type text: %w", err)
|
|
}
|
|
|
|
case "fallback":
|
|
// Try typing first, fallback to clipboard
|
|
err = typeText(ctx, text, i.config.WtypeTimeout)
|
|
if err != nil {
|
|
// Typing failed, but clipboard is already set from above
|
|
// Just log the typing error but don't fail the injection
|
|
return nil
|
|
}
|
|
|
|
default:
|
|
return fmt.Errorf("unsupported injection mode: %s", i.config.Mode)
|
|
}
|
|
|
|
// Restore original clipboard if configured and we have it
|
|
if i.config.RestoreClipboard && originalClipboard != "" {
|
|
// Restore after a short delay to ensure the text has been processed
|
|
go func() {
|
|
time.Sleep(100 * time.Millisecond)
|
|
restoreCtx, cancel := context.WithTimeout(context.Background(), i.config.ClipboardTimeout)
|
|
defer cancel()
|
|
setClipboard(restoreCtx, originalClipboard, i.config.ClipboardTimeout)
|
|
}()
|
|
}
|
|
|
|
return nil
|
|
}
|