apache-airflow-providers-edge3 1.5.0__py3-none-any.whl → 2.0.0rc1__py3-none-any.whl

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

@@ -0,0 +1,103 @@
+/*!
+ * 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, Input, Kbd, type ButtonProps } from "@chakra-ui/react";
+import { useState, useRef, type ChangeEvent } from "react";
+import { useHotkeys } from "react-hotkeys-hook";
+import { FiSearch } from "react-icons/fi";
+import { useDebouncedCallback } from "use-debounce";
+
+import { CloseButton, InputGroup, type InputGroupProps } from "./ui";
+
+const debounceDelay = 200;
+
+type Props = {
+  readonly buttonProps?: ButtonProps;
+  readonly defaultValue: string;
+  readonly groupProps?: InputGroupProps;
+  readonly hideAdvanced?: boolean;
+  readonly hotkeyDisabled?: boolean;
+  readonly onChange: (value: string) => void;
+  readonly placeHolder: string;
+};
+
+export const SearchBar = ({
+  buttonProps,
+  defaultValue,
+  groupProps,
+  hideAdvanced = false,
+  hotkeyDisabled = false,
+  onChange,
+  placeHolder,
+}: Props) => {
+  const handleSearchChange = useDebouncedCallback((val: string) => onChange(val), debounceDelay);
+  const searchRef = useRef<HTMLInputElement>(null);
+  const [value, setValue] = useState(defaultValue);
+
+  const onSearchChange = (event: ChangeEvent<HTMLInputElement>) => {
+    setValue(event.target.value);
+    handleSearchChange(event.target.value);
+  };
+
+  useHotkeys(
+    "mod+k",
+    () => {
+      searchRef.current?.focus();
+    },
+    { enabled: !hotkeyDisabled, preventDefault: true },
+  );
+
+  return (
+    <InputGroup
+      {...groupProps}
+      colorPalette="brand"
+      endElement={
+        <>
+          {value ? (
+            <CloseButton
+              aria-label="Clear search"
+              colorPalette="brand"
+              data-testid="clear-search"
+              onClick={() => {
+                setValue("");
+                onChange("");
+              }}
+              size="xs"
+            />
+          ) : undefined}
+          {hideAdvanced ? undefined : (
+            <Button fontWeight="normal" height={28} variant="ghost" {...buttonProps}>
+              Advanced
+            </Button>
+          )}
+          {!hotkeyDisabled && <Kbd size="sm">⌘K</Kbd>}
+        </>
+      }
+      startElement={<FiSearch />}
+    >
+      <Input
+        data-testid="search-workers"
+        onChange={onSearchChange}
+        placeholder={placeHolder}
+        pr={150}
+        ref={searchRef}
+        value={value}
+      />
+    </InputGroup>
+  );
+};

airflow/providers/edge3/plugins/www/src/components/ui/InputGroup.tsx

@@ -0,0 +1,57 @@
+/*!
+ * 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.
+ */
+
+/* eslint-disable @typescript-eslint/strict-boolean-expressions */
+
+/* eslint-disable @typescript-eslint/no-unsafe-argument */
+import type { BoxProps, InputElementProps } from "@chakra-ui/react";
+import { Group, InputElement } from "@chakra-ui/react";
+import { cloneElement, forwardRef } from "react";
+
+export type InputGroupProps = {
+  readonly children: React.ReactElement;
+  readonly endElement?: React.ReactNode;
+  readonly endElementProps?: InputElementProps;
+  readonly startElement?: React.ReactNode;
+  readonly startElementProps?: InputElementProps;
+} & BoxProps;
+
+export const InputGroup = forwardRef<HTMLDivElement, InputGroupProps>((props, ref) => {
+  const { children, endElement, endElementProps, startElement, startElementProps, ...rest } = props;
+
+  return (
+    <Group ref={ref} {...rest}>
+      {startElement ? (
+        <InputElement pointerEvents="none" {...startElementProps}>
+          {startElement}
+        </InputElement>
+      ) : undefined}
+      {cloneElement(children, {
+        ...(startElement && { ps: "calc(var(--input-height) - 6px)" }),
+        ...(endElement && { pe: "calc(var(--input-height) - 6px)" }),
+        ...children.props,
+      })}
+      {endElement ? (
+        <InputElement placement="end" {...endElementProps}>
+          {endElement}
+        </InputElement>
+      ) : undefined}
+    </Group>
+  );
+});

airflow/providers/edge3/plugins/www/src/components/ui/Select/Content.tsx

@@ -0,0 +1,37 @@
+/*!
+ * 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 { Select as ChakraSelect, Portal } from "@chakra-ui/react";
+import { forwardRef } from "react";
+
+type ContentProps = {
+  readonly portalled?: boolean;
+  readonly portalRef?: React.RefObject<HTMLElement>;
+} & ChakraSelect.ContentProps;
+
+export const Content = forwardRef<HTMLDivElement, ContentProps>((props, ref) => {
+  const { portalled = true, portalRef, ...rest } = props;
+
+  return (
+    <Portal container={portalRef} disabled={!portalled}>
+      <ChakraSelect.Positioner>
+        <ChakraSelect.Content {...rest} ref={ref} />
+      </ChakraSelect.Positioner>
+    </Portal>
+  );
+});

airflow/providers/edge3/plugins/www/src/components/ui/Select/Item.tsx

@@ -0,0 +1,34 @@
+/*!
+ * 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.
+ */
+
+/* eslint-disable @typescript-eslint/no-unsafe-assignment */
+import { Select as ChakraSelect } from "@chakra-ui/react";
+import { forwardRef } from "react";
+
+export const Item = forwardRef<HTMLDivElement, ChakraSelect.ItemProps>((props, ref) => {
+  const { children, item, ...rest } = props;
+
+  return (
+    // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+    <ChakraSelect.Item item={item} key={item.value} {...rest} ref={ref}>
+      {children}
+      <ChakraSelect.ItemIndicator />
+    </ChakraSelect.Item>
+  );
+});

airflow/providers/edge3/plugins/www/src/components/ui/Select/Root.tsx

@@ -0,0 +1,24 @@
+/*!
+ * 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 { Select as ChakraSelect } from "@chakra-ui/react";
+import { forwardRef } from "react";
+
+export const Root = forwardRef<HTMLDivElement, ChakraSelect.RootProps>((props, ref) => (
+  <ChakraSelect.Root {...props} positioning={{ sameWidth: true, ...props.positioning }} ref={ref} />
+));

airflow/providers/edge3/plugins/www/src/components/ui/Select/Trigger.tsx

@@ -0,0 +1,54 @@
+/*!
+ * 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 { Select as ChakraSelect } from "@chakra-ui/react";
+import { forwardRef } from "react";
+
+import { CloseButton } from "../CloseButton";
+
+type Props = {
+  readonly clearable?: boolean;
+  readonly isActive?: boolean;
+  readonly triggerProps?: ChakraSelect.TriggerProps;
+} & ChakraSelect.ControlProps;
+
+export const Trigger = forwardRef<HTMLButtonElement, Props>((props, ref) => {
+  const { children, clearable, isActive, triggerProps, ...rest } = props;
+
+  return (
+    <ChakraSelect.Control {...rest}>
+      <ChakraSelect.Trigger ref={ref} {...triggerProps}>
+        {children}
+      </ChakraSelect.Trigger>
+      <ChakraSelect.IndicatorGroup _rtl={{ bottom: 0, left: 0, right: "auto", top: 0 }}>
+        {clearable ? (
+          <ChakraSelect.ClearTrigger asChild>
+            <CloseButton
+              focusRingWidth="2px"
+              focusVisibleRing="inside"
+              pointerEvents="auto"
+              size="xs"
+              variant="plain"
+            />
+          </ChakraSelect.ClearTrigger>
+        ) : undefined}
+        <ChakraSelect.Indicator />
+      </ChakraSelect.IndicatorGroup>
+    </ChakraSelect.Control>
+  );
+});

airflow/providers/edge3/plugins/www/src/components/ui/Select/ValueText.tsx

@@ -0,0 +1,51 @@
+/*!
+ * 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 type { CollectionItem } from "@chakra-ui/react";
+import { Select as ChakraSelect } from "@chakra-ui/react";
+import { forwardRef } from "react";
+
+type ValueTextProps = {
+  readonly children?: (items: Array<CollectionItem>) => React.ReactNode;
+} & Omit<ChakraSelect.ValueTextProps, "children">;
+
+export const ValueText = forwardRef<HTMLSpanElement, ValueTextProps>((props, ref) => {
+  const { children, ...rest } = props;
+
+  return (
+    <ChakraSelect.ValueText {...rest} ref={ref}>
+      <ChakraSelect.Context>
+        {(select) => {
+          const items = select.selectedItems;
+
+          if (items.length === 0) {
+            return props.placeholder;
+          }
+          if (children) {
+            return children(items);
+          }
+          if (items.length === 1) {
+            return select.collection.stringifyItem(items[0]);
+          }
+
+          return `${items.length} selected`;
+        }}
+      </ChakraSelect.Context>
+    </ChakraSelect.ValueText>
+  );
+});

airflow/providers/edge3/plugins/www/src/components/ui/Select/index.ts

@@ -0,0 +1,34 @@
+/*!
+ * 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 { Select as ChakraSelect } from "@chakra-ui/react";
+
+import { Content } from "./Content";
+import { Item } from "./Item";
+import { Root } from "./Root";
+import { Trigger } from "./Trigger";
+import { ValueText } from "./ValueText";
+
+export const Select = {
+  ...ChakraSelect,
+  Content,
+  Item,
+  Root,
+  Trigger,
+  ValueText,
+};

airflow/providers/edge3/plugins/www/src/components/ui/index.ts

@@ -18,5 +18,8 @@
  */
 
 export * from "./Alert";
+export * from "./CloseButton";
 export * from "./createToaster";
+export * from "./InputGroup";
 export * from "./ScrollToAnchor";
+export * from "./Select";

airflow/providers/edge3/plugins/www/src/constants.ts

@@ -0,0 +1,43 @@
+/*!
+ * 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 { createListCollection } from "@chakra-ui/react";
+import type { EdgeWorkerState } from "../openapi-gen/requests/types.gen";
+import { $EdgeWorkerState } from "../openapi-gen/requests/schemas.gen";
+
+// Helper function to convert snake_case or space-separated strings to Title Case
+const toTitleCase = (str: string): string => {
+  return str
+    .split(/[\s_]+/)
+    .map((word) => word.charAt(0).toUpperCase() + word.slice(1).toLowerCase())
+    .join(" ");
+};
+
+export const workerStateOptions = createListCollection<{
+  label: string;
+  value: EdgeWorkerState | "all";
+}>({
+  items: [
+    { label: "All States", value: "all" },
+    ...$EdgeWorkerState.enum.map((state) => ({
+      label: toTitleCase(state),
+      value: state as EdgeWorkerState,
+    })),
+  ],
+});