feat: repo picker (#226)

Author: jonathanlab

apps/array/src/renderer/features/repository-picker/components/RepositoryPicker.tsx

@@ -147,11 +147,15 @@ export function RepositoryPicker({
           variant="outline"
           size="1"
           color="gray"
-          style={{ width: "100%" }}
           disabled={!hasIntegration}
         >
           <Flex justify="between" align="center" gap="2" width="100%">
-            <Flex align="center" gap="2" style={{ minWidth: 0, flex: 1 }}>
+            <Flex
+              align="center"
+              gap="2"
+              width="250px"
+              style={{ minWidth: 0, flex: 1 }}
+            >
               <GitBranchIcon
                 size={16}
                 weight="regular"

apps/array/src/renderer/features/task-detail/components/TaskInput.tsx

@@ -1,5 +1,7 @@
 import { FolderPicker } from "@features/folder-picker/components/FolderPicker";
+import { RepositoryPicker } from "@features/repository-picker/components/RepositoryPicker";
 import { useSettingsStore } from "@features/settings/stores/settingsStore";
+import { useRepositoryIntegration } from "@hooks/useIntegrations";
 import { useSetHeaderContent } from "@hooks/useSetHeaderContent";
 import { Flex } from "@radix-ui/themes";
 import { useRegisteredFoldersStore } from "@renderer/stores/registeredFoldersStore";
@@ -28,12 +30,17 @@ export function TaskInput() {
   const [selectedDirectory, setSelectedDirectory] = useState(
     lastUsedDirectory || "",
   );
+  const [selectedRepository, setSelectedRepository] = useState<string | null>(
+    null,
+  );
   const [runMode, setRunMode] = useState<RunMode>(
     import.meta.env.DEV ? lastUsedRunMode : "local",
   );
   const [localWorkspaceMode, setLocalWorkspaceMode] =
     useState<LocalWorkspaceMode>(lastUsedLocalWorkspaceMode);
 
+  const { githubIntegration } = useRepositoryIntegration();
+
   useEffect(() => {
     if (view.folderId) {
       const folder = folders.find((f) => f.id === view.folderId);
@@ -60,6 +67,8 @@ export function TaskInput() {
   const { isCreatingTask, canSubmit, handleSubmit } = useTaskCreation({
     editor,
     selectedDirectory,
+    selectedRepository,
+    githubIntegrationId: githubIntegration?.id,
     workspaceMode: effectiveWorkspaceMode,
   });
 
@@ -113,12 +122,21 @@ export function TaskInput() {
         }}
       >
         <Flex gap="2" align="center">
-          <FolderPicker
-            value={selectedDirectory}
-            onChange={handleDirectoryChange}
-            placeholder="Select working directory..."
-            size="1"
-          />
+          {runMode === "cloud" ? (
+            <RepositoryPicker
+              value={selectedRepository}
+              onChange={setSelectedRepository}
+              placeholder="Select repository..."
+              size="1"
+            />
+          ) : (
+            <FolderPicker
+              value={selectedDirectory}
+              onChange={handleDirectoryChange}
+              placeholder="Select working directory..."
+              size="1"
+            />
+          )}
           {import.meta.env.DEV && (
             <RunModeSelect value={runMode} onChange={setRunMode} size="1" />
           )}
@@ -132,7 +150,9 @@ export function TaskInput() {
           onLocalWorkspaceModeChange={setLocalWorkspaceMode}
           canSubmit={canSubmit}
           onSubmit={handleSubmit}
-          hasDirectory={!!selectedDirectory}
+          hasDirectory={
+            runMode === "cloud" ? !!selectedRepository : !!selectedDirectory
+          }
         />
 
         <SuggestedTasks editor={editor} />

apps/array/src/renderer/features/task-detail/hooks/useTaskCreation.ts

@@ -22,6 +22,8 @@ const log = logger.scope("task-creation");
 interface UseTaskCreationOptions {
   editor: Editor | null;
   selectedDirectory: string;
+  selectedRepository?: string | null;
+  githubIntegrationId?: number;
   workspaceMode: WorkspaceMode;
 }
 
@@ -46,6 +48,8 @@ async function startAgentSession(
 export function useTaskCreation({
   editor,
   selectedDirectory,
+  selectedRepository,
+  githubIntegrationId,
   workspaceMode,
 }: UseTaskCreationOptions): UseTaskCreationReturn {
   const {
@@ -60,20 +64,22 @@ export function useTaskCreation({
   const { autoRunTasks } = useSettingsStore();
   const { clearDraft } = useTaskInputStore();
 
+  const isCloudMode = workspaceMode === "cloud";
   const canSubmit =
     !!editor &&
     isAuthenticated &&
     !!client &&
-    !!selectedDirectory &&
+    (isCloudMode ? !!selectedRepository : !!selectedDirectory) &&
     !isCreatingTask &&
     !editor.isEmpty;
 
   const handleSubmit = useCallback(async () => {
+    const isCloud = workspaceMode === "cloud";
     const canSubmit =
       !!editor &&
       isAuthenticated &&
       !!client &&
-      !!selectedDirectory &&
+      (isCloud ? !!selectedRepository : !!selectedDirectory) &&
       !isCreatingTask &&
       !editor.isEmpty;
 
@@ -91,7 +97,9 @@ export function useTaskCreation({
     const filePaths = extractFileMentions(editorJson);
 
     let repository: string | undefined;
-    if (selectedDirectory) {
+    if (selectedRepository) {
+      repository = selectedRepository;
+    } else if (selectedDirectory) {
       const detected = await window.electronAPI.detectRepo(selectedDirectory);
       if (detected) {
         repository = `${detected.organization}/${detected.repository}`;
@@ -102,6 +110,7 @@ export function useTaskCreation({
       {
         description: content,
         repository,
+        github_integration: isCloud ? githubIntegrationId : undefined,
         autoRun: autoRunTasks,
         createdFrom: "cli",
       },
@@ -192,6 +201,8 @@ export function useTaskCreation({
   }, [
     editor,
     selectedDirectory,
+    selectedRepository,
+    githubIntegrationId,
     workspaceMode,
     createTask,
     saveRepoPath,

apps/array/src/renderer/features/tasks/hooks/useTasks.ts

@@ -40,16 +40,19 @@ export function useCreateTask() {
       {
         description,
         repository,
+        github_integration,
       }: {
         description: string;
         repository?: string;
+        github_integration?: number;
         autoRun?: boolean;
         createdFrom?: "cli" | "command-menu";
       },
     ) =>
       client.createTask({
         description,
         repository,
+        github_integration,
       }) as unknown as Promise<Task>,
     {
       onSuccess: (_task, variables) => {