Fix selecting single job from evaluated job ref

5 files changed, 5 insertions, 13 deletions

diff --git a/src/Command/Extract.hs b/src/Command/Extract.hs
index 366128c..8dee537 100644
--- a/src/Command/Extract.hs
+++ b/src/Command/Extract.hs
@@ -6,7 +6,6 @@ import Control.Monad
 import Control.Monad.Except
 import Control.Monad.IO.Class
 
-import Data.Bifunctor
 import Data.Text qualified as T
 
 import System.Console.GetOpt
@@ -80,8 +79,7 @@ cmdExtract (ExtractCommand ExtractOptions {..} ExtractArguments {..}) = do
             _     -> return False
 
     forM_ extractArtifacts $ \( ref, aname ) -> do
-        [ jid ] <- either tfail (return . map jobId) =<<
-            return . either (Left . textEvalError) (first T.pack . jobsetJobsEither) =<<
+        jid <- either (tfail . textEvalError) (return . jobId) =<<
             liftIO (runEval (evalJobReference ref) einput)
 
         tpath <- if
diff --git a/src/Command/JobId.hs b/src/Command/JobId.hs
index b349ebe..173f543 100644
--- a/src/Command/JobId.hs
+++ b/src/Command/JobId.hs
@@ -5,7 +5,6 @@ module Command.JobId (
 import Control.Monad
 import Control.Monad.IO.Class
 
-import Data.Bifunctor
 import Data.Text (Text)
 import Data.Text qualified as T
 
@@ -53,8 +52,7 @@ cmdJobId :: JobIdCommand -> CommandExec ()
 cmdJobId (JobIdCommand JobIdOptions {..} ref) = do
     einput <- getEvalInput
     out <- getOutput
-    [ JobId ids ] <- either tfail (return . map jobId) =<<
-        return . either (Left . textEvalError) (first T.pack . jobsetJobsEither) =<<
+    JobId ids <- either (tfail . textEvalError) (return . jobId) =<<
         liftIO (runEval (evalJobReference ref) einput)
 
     outputMessage out $ textJobId $ JobId ids
diff --git a/src/Command/Log.hs b/src/Command/Log.hs
index 438c25e..25bfc06 100644
--- a/src/Command/Log.hs
+++ b/src/Command/Log.hs
@@ -4,7 +4,6 @@ module Command.Log (
 
 import Control.Monad.IO.Class
 
-import Data.Bifunctor
 import Data.Text (Text)
 import Data.Text qualified as T
 import Data.Text.Lazy qualified as TL
@@ -38,8 +37,7 @@ instance Command LogCommand where
 cmdLog :: LogCommand -> CommandExec ()
 cmdLog (LogCommand ref) = do
     einput <- getEvalInput
-    [ jid ] <- either tfail (return . map jobId) =<<
-        return . either (Left . textEvalError) (first T.pack . jobsetJobsEither) =<<
+    jid <- either (tfail . textEvalError) (return . jobId) =<<
         liftIO (runEval (evalJobReference ref) einput)
     output <- getOutput
     storageDir <- getStorageDir
diff --git a/src/Command/Run.hs b/src/Command/Run.hs
index 04c7273..b299931 100644
--- a/src/Command/Run.hs
+++ b/src/Command/Run.hs
@@ -181,7 +181,7 @@ argumentJobSource names = do
 refJobSource :: [ JobRef ] -> CommandExec JobSource
 refJobSource [] = emptyJobSource
 refJobSource refs = do
-    sets <- foldl' addJobToList [] <$> cmdEvalWith id (mapM evalJobReference refs)
+    sets <- foldl' addJobToList [] <$> cmdEvalWith id (mapM evalJobReferenceToSet refs)
     oneshotJobSource sets
   where
     addJobToList :: [ JobSet ] -> JobSet -> [ JobSet ]
diff --git a/src/Command/Shell.hs b/src/Command/Shell.hs
index dfff50a..16f366e 100644
--- a/src/Command/Shell.hs
+++ b/src/Command/Shell.hs
@@ -5,7 +5,6 @@ module Command.Shell (
 import Control.Monad
 import Control.Monad.IO.Class
 
-import Data.Bifunctor
 import Data.Maybe
 import Data.Text (Text)
 import Data.Text qualified as T
@@ -38,8 +37,7 @@ instance Command ShellCommand where
 cmdShell :: ShellCommand -> CommandExec ()
 cmdShell (ShellCommand ref) = do
     einput <- getEvalInput
-    [ job ] <- either tfail return =<<
-        return . either (Left . textEvalError) (first T.pack . jobsetJobsEither) =<<
+    job <- either (tfail . textEvalError) return =<<
         liftIO (runEval (evalJobReference ref) einput)
     sh <- fromMaybe "/bin/sh" <$> liftIO (lookupEnv "SHELL")
     storageDir <- getStorageDir