package api

import (
	"net/http"

	"github.com/jfraeys/fetch_ml/internal/api/ws"
	"github.com/jfraeys/fetch_ml/internal/audit"
	"github.com/jfraeys/fetch_ml/internal/auth"
	"github.com/jfraeys/fetch_ml/internal/config"
	"github.com/jfraeys/fetch_ml/internal/experiment"
	"github.com/jfraeys/fetch_ml/internal/jupyter"
	"github.com/jfraeys/fetch_ml/internal/logging"
	"github.com/jfraeys/fetch_ml/internal/queue"
	"github.com/jfraeys/fetch_ml/internal/storage"
)

// Re-export WebSocket types from ws package for backward compatibility
// Deprecated: Use api/ws package directly

// NewWSHandler creates a new WebSocket handler (re-export from ws package)
// Deprecated: Use ws.NewHandler instead
func NewWSHandler(
	authConfig *auth.Config,
	logger *logging.Logger,
	expManager *experiment.Manager,
	dataDir string,
	taskQueue queue.Backend,
	db *storage.DB,
	jupyterServiceMgr *jupyter.ServiceManager,
	securityCfg *config.SecurityConfig,
	auditLogger *audit.Logger,
) http.Handler {
	return ws.NewHandler(authConfig, logger, expManager, dataDir, taskQueue, db, jupyterServiceMgr, securityCfg, auditLogger)
}

// WebSocket opcodes (re-exported from ws package)
// Deprecated: Use ws package constants directly
const (
	OpcodeQueueJob             = ws.OpcodeQueueJob
	OpcodeStatusRequest        = ws.OpcodeStatusRequest
	OpcodeCancelJob            = ws.OpcodeCancelJob
	OpcodePrune                = ws.OpcodePrune
	OpcodeDatasetList          = ws.OpcodeDatasetList
	OpcodeDatasetRegister      = ws.OpcodeDatasetRegister
	OpcodeDatasetInfo          = ws.OpcodeDatasetInfo
	OpcodeDatasetSearch        = ws.OpcodeDatasetSearch
	OpcodeLogMetric            = ws.OpcodeLogMetric
	OpcodeGetExperiment        = ws.OpcodeGetExperiment
	OpcodeQueueJobWithTracking = ws.OpcodeQueueJobWithTracking
	OpcodeQueueJobWithSnapshot = ws.OpcodeQueueJobWithSnapshot
	OpcodeQueueJobWithArgs     = ws.OpcodeQueueJobWithArgs
	OpcodeQueueJobWithNote     = ws.OpcodeQueueJobWithNote
	OpcodeAnnotateRun          = ws.OpcodeAnnotateRun
	OpcodeSetRunNarrative      = ws.OpcodeSetRunNarrative
	OpcodeStartJupyter         = ws.OpcodeStartJupyter
	OpcodeStopJupyter          = ws.OpcodeStopJupyter
	OpcodeRemoveJupyter        = ws.OpcodeRemoveJupyter
	OpcodeRestoreJupyter       = ws.OpcodeRestoreJupyter
	OpcodeListJupyter          = ws.OpcodeListJupyter
	OpcodeListJupyterPackages  = ws.OpcodeListJupyterPackages
	OpcodeValidateRequest      = ws.OpcodeValidateRequest
	OpcodeGetLogs              = ws.OpcodeGetLogs
	OpcodeStreamLogs           = ws.OpcodeStreamLogs
)