fb2bbbaae5
Phase 7 of the monorepo maintainability plan: New files created: - model/jobs.go - Job type, JobStatus constants, list.Item interface - model/messages.go - tea.Msg types (JobsLoadedMsg, StatusMsg, TickMsg, etc.) - model/styles.go - NewJobListDelegate(), JobListTitleStyle(), SpinnerStyle() - model/keys.go - KeyMap struct, DefaultKeys() function Modified files: - model/state.go - reduced from 226 to ~130 lines - Removed: Job, JobStatus, KeyMap, Keys, inline styles - Kept: State struct, domain re-exports, ViewMode, DatasetInfo, InitialState() - controller/commands.go - use model. prefix for message types - controller/controller.go - use model. prefix for message types - controller/settings.go - use model.SettingsContentMsg Deleted files: - controller/keys.go (moved to model/keys.go since State references KeyMap) Result: - No file >150 lines in model/ package - Single concern per file: state, jobs, messages, styles, keys - All 41 test packages pass
57 lines
1.6 KiB
Go
57 lines
1.6 KiB
Go
// Package network provides networking and server communication utilities
|
|
package network
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
// MLServer wraps SSHClient to provide a high-level interface for ML operations.
|
|
// It consolidates the TUI and worker implementations into a single reusable component.
|
|
type MLServer struct {
|
|
client SSHClient
|
|
addr string
|
|
}
|
|
|
|
// NewMLServer creates a new ML server connection.
|
|
// If host is empty, it creates a local mode client (no SSH).
|
|
func NewMLServer(host, user, sshKey string, port int, knownHosts string) (*MLServer, error) {
|
|
// Local mode: skip SSH entirely
|
|
if host == "" {
|
|
client, _ := NewSSHClient("", "", "", 0, "")
|
|
return &MLServer{client: *client, addr: "localhost"}, nil
|
|
}
|
|
|
|
client, err := NewSSHClient(host, user, sshKey, port, knownHosts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create SSH client: %w", err)
|
|
}
|
|
|
|
addr := fmt.Sprintf("%s:%d", host, port)
|
|
return &MLServer{client: *client, addr: addr}, nil
|
|
}
|
|
|
|
// Exec executes a command on the ML server.
|
|
// For local mode, it executes directly on the local machine.
|
|
func (s *MLServer) Exec(command string) (string, error) {
|
|
return s.client.Exec(command)
|
|
}
|
|
|
|
// ListDir lists files in a directory on the ML server.
|
|
func (s *MLServer) ListDir(path string) []string {
|
|
return s.client.ListDir(path)
|
|
}
|
|
|
|
// Addr returns the server address ("localhost" for local mode).
|
|
func (s *MLServer) Addr() string {
|
|
return s.addr
|
|
}
|
|
|
|
// IsLocal returns true if running in local mode (no SSH).
|
|
func (s *MLServer) IsLocal() bool {
|
|
return s.addr == "localhost"
|
|
}
|
|
|
|
// Close closes the SSH connection.
|
|
func (s *MLServer) Close() error {
|
|
return s.client.Close()
|
|
}
|