apache-airflow-providers-edge3 1.2.0__py3-none-any.whl → 1.3.0rc1__py3-none-any.whl

airflow/providers/edge3/plugins/www/src/components/AddQueueButton.tsx

@@ -0,0 +1,138 @@
+/*!
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+import {
+  Button,
+  CloseButton,
+  Dialog,
+  Input,
+  IconButton,
+  Portal,
+  Text,
+  useDisclosure,
+  VStack,
+} from "@chakra-ui/react";
+import { useUiServiceAddWorkerQueue } from "openapi/queries";
+import { useState } from "react";
+import { LuListPlus } from "react-icons/lu";
+
+interface AddQueueButtonProps {
+  onQueueUpdate: (toast: Record<string, string>) => void;
+  workerName: string;
+}
+
+export const AddQueueButton = ({ onQueueUpdate, workerName }: AddQueueButtonProps) => {
+  const { onClose, onOpen, open } = useDisclosure();
+  const [queueName, setQueueName] = useState("");
+
+  const addQueueMutation = useUiServiceAddWorkerQueue({
+    onError: (error) => {
+      onQueueUpdate({
+        description: `Unable to add queue to worker ${workerName}: ${error}`,
+        title: "Add Queue Failed",
+        type: "error",
+      });
+    },
+    onSuccess: () => {
+      onQueueUpdate({
+        description: `Queue "${queueName}" was added to worker ${workerName}.`,
+        title: "Queue Added",
+        type: "success",
+      });
+      onClose();
+      setQueueName("");
+    },
+  });
+
+  const handleAddQueue = () => {
+    if (!queueName.trim()) {
+      onQueueUpdate({
+        description: "Please enter a queue name.",
+        title: "Invalid Input",
+        type: "error",
+      });
+      return;
+    }
+
+    addQueueMutation.mutate({
+      queueName: queueName.trim(),
+      workerName,
+    });
+  };
+
+  return (
+    <>
+      <IconButton
+        size="sm"
+        variant="ghost"
+        onClick={onOpen}
+        aria-label="Add Queue"
+        title="Add Queue"
+        colorPalette="success"
+      >
+        <LuListPlus />
+      </IconButton>
+
+      <Dialog.Root onOpenChange={onClose} open={open} size="md">
+        <Portal>
+          <Dialog.Backdrop />
+          <Dialog.Positioner>
+            <Dialog.Content>
+              <Dialog.Header>
+                <Dialog.Title>Add Queue to {workerName}</Dialog.Title>
+              </Dialog.Header>
+              <Dialog.Body>
+                <VStack gap={4} align="stretch">
+                  <Text>Enter the name of the queue to add to this worker:</Text>
+                  <Input
+                    placeholder="Queue name"
+                    value={queueName}
+                    onChange={(e) => setQueueName(e.target.value)}
+                    onKeyDown={(e) => {
+                      if (e.key === "Enter") {
+                        handleAddQueue();
+                      }
+                    }}
+                  />
+                </VStack>
+              </Dialog.Body>
+              <Dialog.Footer>
+                <Dialog.ActionTrigger asChild>
+                  <Button variant="outline">Cancel</Button>
+                </Dialog.ActionTrigger>
+                <Button
+                  onClick={handleAddQueue}
+                  colorPalette="success"
+                  loading={addQueueMutation.isPending}
+                  loadingText="Adding queue..."
+                  disabled={!queueName.trim()}
+                >
+                  <LuListPlus />
+                  Add Queue
+                </Button>
+              </Dialog.Footer>
+              <Dialog.CloseTrigger asChild>
+                <CloseButton size="sm" />
+              </Dialog.CloseTrigger>
+            </Dialog.Content>
+          </Dialog.Positioner>
+        </Portal>
+      </Dialog.Root>
+    </>
+  );
+};

airflow/providers/edge3/plugins/www/src/components/MaintenanceEditCommentButton.tsx

@@ -0,0 +1,106 @@
+/*!
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+import { Button, CloseButton, Dialog, IconButton, Portal, Textarea, useDisclosure } from "@chakra-ui/react";
+import { useUiServiceUpdateWorkerMaintenance } from "openapi/queries";
+import { useState } from "react";
+import { FiEdit } from "react-icons/fi";
+
+interface MaintenanceEditCommentButtonProps {
+  onEditComment: (toast: Record<string, string>) => void;
+  workerName: string;
+}
+
+export const MaintenanceEditCommentButton = ({
+  onEditComment,
+  workerName,
+}: MaintenanceEditCommentButtonProps) => {
+  const { onClose, onOpen, open } = useDisclosure();
+  const [comment, setComment] = useState("");
+
+  const editCommentMutation = useUiServiceUpdateWorkerMaintenance({
+    onError: (error) => {
+      onEditComment({
+        description: `Unable to update comments for worker ${workerName}: ${error}`,
+        title: "Updating Comments failed",
+        type: "error",
+      });
+    },
+    onSuccess: () => {
+      onEditComment({
+        description: `Worker maintenance comments for ${workerName} were updated.`,
+        title: "Maintenance Comments updated",
+        type: "success",
+      });
+      onClose();
+    },
+  });
+
+  const editComment = () => {
+    editCommentMutation.mutate({ requestBody: { maintenance_comment: comment }, workerName });
+  };
+
+  return (
+    <>
+      <IconButton
+        size="sm"
+        variant="ghost"
+        aria-label="Edit Comments"
+        title="Edit Comments"
+        onClick={onOpen}
+        colorPalette="warning"
+      >
+        <FiEdit />
+      </IconButton>
+
+      <Dialog.Root onOpenChange={onClose} open={open} size="md">
+        <Portal>
+          <Dialog.Backdrop />
+          <Dialog.Positioner>
+            <Dialog.Content>
+              <Dialog.Header>
+                <Dialog.Title>Edit maintenance comments for worker {workerName}</Dialog.Title>
+              </Dialog.Header>
+              <Dialog.Body>
+                <Textarea
+                  placeholder="Change maintenance comment (required)"
+                  value={comment}
+                  onChange={(e) => setComment(e.target.value)}
+                  required
+                  maxLength={1024}
+                  size="sm"
+                />
+              </Dialog.Body>
+              <Dialog.Footer>
+                <Dialog.ActionTrigger asChild>
+                  <Button variant="outline">Cancel</Button>
+                </Dialog.ActionTrigger>
+                <Button onClick={editComment} disabled={!comment.trim()}>
+                  Update Comments
+                </Button>
+              </Dialog.Footer>
+              <Dialog.CloseTrigger asChild>
+                <CloseButton size="sm" />
+              </Dialog.CloseTrigger>
+            </Dialog.Content>
+          </Dialog.Positioner>
+        </Portal>
+      </Dialog.Root>
+    </>
+  );
+};

airflow/providers/edge3/plugins/www/src/components/MaintenanceEnterButton.tsx

@@ -0,0 +1,102 @@
+/*!
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+import { Button, CloseButton, Dialog, IconButton, Portal, Textarea, useDisclosure } from "@chakra-ui/react";
+import { useUiServiceRequestWorkerMaintenance } from "openapi/queries";
+import { useState } from "react";
+import { HiOutlineWrenchScrewdriver } from "react-icons/hi2";
+
+interface MaintenanceEnterButtonProps {
+  onEnterMaintenance: (toast: Record<string, string>) => void;
+  workerName: string;
+}
+
+export const MaintenanceEnterButton = ({ onEnterMaintenance, workerName }: MaintenanceEnterButtonProps) => {
+  const { onClose, onOpen, open } = useDisclosure();
+  const [comment, setComment] = useState("");
+
+  const enterMaintenanceMutation = useUiServiceRequestWorkerMaintenance({
+    onError: (error) => {
+      onEnterMaintenance({
+        description: `Unable to set worker ${workerName} to maintenance mode: ${error}`,
+        title: "Setting Maintenance Mode failed",
+        type: "error",
+      });
+    },
+    onSuccess: () => {
+      onEnterMaintenance({
+        description: `Worker ${workerName} was requested to be in maintenance mode.`,
+        title: "Maintenance Mode activated",
+        type: "success",
+      });
+    },
+  });
+
+  const enterMaintenance = () => {
+    enterMaintenanceMutation.mutate({ requestBody: { maintenance_comment: comment }, workerName });
+  };
+
+  return (
+    <>
+      <IconButton
+        size="sm"
+        variant="ghost"
+        aria-label="Enter Maintenance"
+        title="Enter Maintenance"
+        onClick={onOpen}
+        colorPalette="warning"
+      >
+        <HiOutlineWrenchScrewdriver />
+      </IconButton>
+
+      <Dialog.Root onOpenChange={onClose} open={open} size="md">
+        <Portal>
+          <Dialog.Backdrop />
+          <Dialog.Positioner>
+            <Dialog.Content>
+              <Dialog.Header>
+                <Dialog.Title>Set maintenance for worker {workerName}</Dialog.Title>
+              </Dialog.Header>
+              <Dialog.Body>
+                <Textarea
+                  placeholder="Enter maintenance comment (required)"
+                  value={comment}
+                  onChange={(e) => setComment(e.target.value)}
+                  required
+                  maxLength={1024}
+                  size="sm"
+                />
+              </Dialog.Body>
+              <Dialog.Footer>
+                <Dialog.ActionTrigger asChild>
+                  <Button variant="outline">Cancel</Button>
+                </Dialog.ActionTrigger>
+                <Button onClick={enterMaintenance} disabled={!comment.trim()}>
+                  Confirm Maintenance
+                </Button>
+              </Dialog.Footer>
+              <Dialog.CloseTrigger asChild>
+                <CloseButton size="sm" />
+              </Dialog.CloseTrigger>
+            </Dialog.Content>
+          </Dialog.Positioner>
+        </Portal>
+      </Dialog.Root>
+    </>
+  );
+};

airflow/providers/edge3/plugins/www/src/components/MaintenanceExitButton.tsx

@@ -0,0 +1,92 @@
+/*!
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+import { Button, CloseButton, Dialog, IconButton, Portal, Text, useDisclosure } from "@chakra-ui/react";
+import { useUiServiceExitWorkerMaintenance } from "openapi/queries";
+import { IoMdExit } from "react-icons/io";
+
+interface MaintenanceExitButtonProps {
+  onExitMaintenance: (toast: Record<string, string>) => void;
+  workerName: string;
+}
+
+export const MaintenanceExitButton = ({ onExitMaintenance, workerName }: MaintenanceExitButtonProps) => {
+  const { onClose, onOpen, open } = useDisclosure();
+
+  const exitMaintenanceMutation = useUiServiceExitWorkerMaintenance({
+    onError: (error) => {
+      onExitMaintenance({
+        description: `Unable to exit ${workerName} from maintenance mode: ${error}`,
+        title: "Exit Maintenance Mode failed",
+        type: "error",
+      });
+    },
+    onSuccess: () => {
+      onExitMaintenance({
+        description: `Worker ${workerName} was requested to exit maintenance mode.`,
+        title: "Maintenance Mode deactivated",
+        type: "success",
+      });
+      onClose();
+    },
+  });
+
+  const exitMaintenance = () => {
+    exitMaintenanceMutation.mutate({ workerName });
+  };
+
+  return (
+    <>
+      <IconButton
+        size="sm"
+        variant="ghost"
+        onClick={onOpen}
+        aria-label="Exit Maintenance"
+        title="Exit Maintenance"
+        colorPalette="warning"
+      >
+        <IoMdExit />
+      </IconButton>
+
+      <Dialog.Root onOpenChange={onClose} open={open} size="md">
+        <Portal>
+          <Dialog.Backdrop />
+          <Dialog.Positioner>
+            <Dialog.Content>
+              <Dialog.Header>
+                <Dialog.Title>Exit maintenance for worker {workerName}</Dialog.Title>
+              </Dialog.Header>
+              <Dialog.Body>
+                <Text>Are you sure you want to exit maintenance mode for worker {workerName}?</Text>
+              </Dialog.Body>
+              <Dialog.Footer>
+                <Dialog.ActionTrigger asChild>
+                  <Button variant="outline">No</Button>
+                </Dialog.ActionTrigger>
+                <Button onClick={exitMaintenance}>Yes</Button>
+              </Dialog.Footer>
+              <Dialog.CloseTrigger asChild>
+                <CloseButton size="sm" />
+              </Dialog.CloseTrigger>
+            </Dialog.Content>
+          </Dialog.Positioner>
+        </Portal>
+      </Dialog.Root>
+    </>
+  );
+};

airflow/providers/edge3/plugins/www/src/components/RemoveQueueButton.tsx

@@ -0,0 +1,151 @@
+/*!
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+import {
+  Button,
+  CloseButton,
+  Dialog,
+  IconButton,
+  Portal,
+  Text,
+  useDisclosure,
+  VStack,
+  For,
+} from "@chakra-ui/react";
+import { useUiServiceRemoveWorkerQueue } from "openapi/queries";
+import type { Worker } from "openapi/requests/types.gen";
+import { useState } from "react";
+import { LuListMinus } from "react-icons/lu";
+
+interface RemoveQueueButtonProps {
+  onQueueUpdate: (toast: Record<string, string>) => void;
+  worker: Worker;
+}
+
+export const RemoveQueueButton = ({ onQueueUpdate, worker }: RemoveQueueButtonProps) => {
+  const { onClose, onOpen, open } = useDisclosure();
+  const [selectedQueue, setSelectedQueue] = useState<string>("");
+
+  const removeQueueMutation = useUiServiceRemoveWorkerQueue({
+    onError: (error) => {
+      onQueueUpdate({
+        description: `Unable to remove queue from worker ${worker.worker_name}: ${error}`,
+        title: "Remove Queue Failed",
+        type: "error",
+      });
+    },
+    onSuccess: () => {
+      onQueueUpdate({
+        description: `Queue "${selectedQueue}" was removed from worker ${worker.worker_name}.`,
+        title: "Queue Removed",
+        type: "success",
+      });
+      onClose();
+      setSelectedQueue("");
+    },
+  });
+
+  const handleRemoveQueue = () => {
+    if (!selectedQueue) {
+      onQueueUpdate({
+        description: "Please select a queue to remove.",
+        title: "Invalid Selection",
+        type: "error",
+      });
+      return;
+    }
+
+    removeQueueMutation.mutate({
+      queueName: selectedQueue,
+      workerName: worker.worker_name,
+    });
+  };
+
+  const availableQueues = worker.queues || [];
+
+  // Don't render the button if there are no queues to remove
+  if (availableQueues.length === 0) {
+    return null;
+  }
+
+  return (
+    <>
+      <IconButton
+        size="sm"
+        variant="ghost"
+        onClick={onOpen}
+        aria-label="Remove Queue"
+        title="Remove Queue"
+        colorPalette="danger"
+      >
+        <LuListMinus />
+      </IconButton>
+
+      <Dialog.Root onOpenChange={onClose} open={open} size="md">
+        <Portal>
+          <Dialog.Backdrop />
+          <Dialog.Positioner>
+            <Dialog.Content>
+              <Dialog.Header>
+                <Dialog.Title>Remove Queue from {worker.worker_name}</Dialog.Title>
+              </Dialog.Header>
+              <Dialog.Body>
+                <VStack gap={4} align="stretch">
+                  <Text>Select a queue to remove from this worker:</Text>
+                  <VStack gap={2} align="stretch">
+                    <For each={availableQueues}>
+                      {(queue) => (
+                        <Button
+                          key={queue}
+                          variant={selectedQueue === queue ? "solid" : "outline"}
+                          colorPalette={selectedQueue === queue ? "blue" : "gray"}
+                          onClick={() => setSelectedQueue(queue)}
+                          justifyContent="flex-start"
+                        >
+                          {queue}
+                        </Button>
+                      )}
+                    </For>
+                  </VStack>
+                </VStack>
+              </Dialog.Body>
+              <Dialog.Footer>
+                <Dialog.ActionTrigger asChild>
+                  <Button variant="outline">Cancel</Button>
+                </Dialog.ActionTrigger>
+                <Button
+                  onClick={handleRemoveQueue}
+                  colorPalette="danger"
+                  loading={removeQueueMutation.isPending}
+                  loadingText="Removing queue..."
+                  disabled={!selectedQueue}
+                >
+                  <LuListMinus />
+                  Remove Queue
+                </Button>
+              </Dialog.Footer>
+              <Dialog.CloseTrigger asChild>
+                <CloseButton size="sm" />
+              </Dialog.CloseTrigger>
+            </Dialog.Content>
+          </Dialog.Positioner>
+        </Portal>
+      </Dialog.Root>
+    </>
+  );
+};

airflow/providers/edge3/plugins/www/src/components/WorkerDeleteButton.tsx

@@ -0,0 +1,104 @@
+/*!
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+import { Button, CloseButton, Dialog, IconButton, Portal, Text, useDisclosure } from "@chakra-ui/react";
+import { useUiServiceDeleteWorker } from "openapi/queries";
+import { FaRegTrashCan } from "react-icons/fa6";
+
+interface WorkerDeleteButtonProps {
+  onDelete: (toast: Record<string, string>) => void;
+  workerName: string;
+}
+
+export const WorkerDeleteButton = ({ onDelete, workerName }: WorkerDeleteButtonProps) => {
+  const { onClose, onOpen, open } = useDisclosure();
+
+  const deleteMutation = useUiServiceDeleteWorker({
+    onError: (error) => {
+      onDelete({
+        description: `Unable to delete worker ${workerName}: ${error}`,
+        title: "Delete Worker Failed",
+        type: "error",
+      });
+    },
+    onSuccess: () => {
+      onDelete({
+        description: `Worker ${workerName} has been deleted from the system.`,
+        title: "Worker Deleted",
+        type: "success",
+      });
+      onClose();
+    },
+  });
+
+  const handleDelete = () => {
+    deleteMutation.mutate({ workerName });
+  };
+
+  return (
+    <>
+      <IconButton
+        size="sm"
+        variant="ghost"
+        onClick={onOpen}
+        aria-label="Delete Worker"
+        title="Delete Worker"
+        colorPalette="danger"
+      >
+        <FaRegTrashCan />
+      </IconButton>
+
+      <Dialog.Root onOpenChange={onClose} open={open} size="md">
+        <Portal>
+          <Dialog.Backdrop />
+          <Dialog.Positioner>
+            <Dialog.Content>
+              <Dialog.Header>
+                <Dialog.Title>Delete worker {workerName}</Dialog.Title>
+              </Dialog.Header>
+              <Dialog.Body>
+                <Text>Are you sure you want to delete worker {workerName}?</Text>
+                <Text fontSize="sm" color="red.500" mt={2}>
+                  This will permanently remove the worker record from the system. This action cannot be
+                  undone.
+                </Text>
+              </Dialog.Body>
+              <Dialog.Footer>
+                <Dialog.ActionTrigger asChild>
+                  <Button variant="outline">Cancel</Button>
+                </Dialog.ActionTrigger>
+                <Button
+                  onClick={handleDelete}
+                  colorPalette="danger"
+                  loading={deleteMutation.isPending}
+                  loadingText="Deleting..."
+                >
+                  <FaRegTrashCan style={{ marginRight: "8px" }} />
+                  Delete Worker
+                </Button>
+              </Dialog.Footer>
+              <Dialog.CloseTrigger asChild>
+                <CloseButton size="sm" />
+              </Dialog.CloseTrigger>
+            </Dialog.Content>
+          </Dialog.Positioner>
+        </Portal>
+      </Dialog.Root>
+    </>
+  );
+};