From d3419713c82922c560ef54eace9f91cc2ac10e68 Mon Sep 17 00:00:00 2001
From: Xevion <xevion@xevion.dev>
Date: Fri, 25 Aug 2023 23:55:32 -0500
Subject: [PATCH] Add manual target type selection ListBox, add targetType to
 onChange return, autodetect interface

---
 src/components/form/LookupInput.tsx | 142 ++++++++++++++++++++++++++--
 src/pages/index.tsx                 |   2 +-
 2 files changed, 135 insertions(+), 9 deletions(-)

diff --git a/src/components/form/LookupInput.tsx b/src/components/form/LookupInput.tsx
index 19c70d5..cedcaec 100644
--- a/src/components/form/LookupInput.tsx
+++ b/src/components/form/LookupInput.tsx
@@ -1,8 +1,15 @@
 import { useForm } from "react-hook-form";
 import type { FunctionComponent } from "react";
-import { onPromise, preventDefault } from "@/helpers";
+import { Fragment, useState } from "react";
+import { classNames, onPromise, preventDefault } from "@/helpers";
 import type { SubmitProps, TargetType } from "@/types";
-import { MagnifyingGlassIcon } from "@heroicons/react/20/solid";
+import { ObjectType } from "@/types";
+import {
+  CheckIcon,
+  ChevronUpDownIcon,
+  MagnifyingGlassIcon,
+} from "@heroicons/react/20/solid";
+import { Listbox, Transition } from "@headlessui/react";
 
 type LookupInputProps = {
   isLoading?: boolean;
@@ -10,7 +17,7 @@ type LookupInputProps = {
   onRegistry?: (type: TargetType) => Promise<any>;
   // When a user hits submit, this is called.
   onSubmit?: (props: SubmitProps) => Promise<any>;
-  onChange?: (target: string) => any;
+  onChange?: (target: { target: string; targetType: ObjectType | null }) => any;
 };
 
 const LookupInput: FunctionComponent<LookupInputProps> = ({
@@ -26,6 +33,36 @@ const LookupInput: FunctionComponent<LookupInputProps> = ({
     },
   });
 
+  const options: Record<ObjectType | "auto", string> = {
+    auto: "Autodetect",
+    domain: "Domain",
+    ip: "IP/CIDR",
+    tld: "TLD",
+    autnum: "AS Number",
+    entity: "Entity Handle",
+    registrar: "Registrar",
+    url: "URL",
+    json: "JSON",
+  };
+
+  const [selected, setSelected] = useState<ObjectType | "auto">("auto");
+
+  function getAutoTargetText(): string {
+    return "Autodetect";
+  }
+
+  /*
+   * Returns the type of object that the user has selected. Handles the extra "auto" option like null.
+   * @param value The value of the select element.
+   */
+  function getTargetType(value?: string | null): ObjectType | null {
+    // Pull the value from the select element if it's not provided (useful for eventing values).
+    if (value == null) value = selected;
+
+    if (value == "auto" || value == null) return null;
+    else return value as ObjectType;
+  }
+
   return (
     <form
       className="pb-3"
@@ -39,7 +76,7 @@ const LookupInput: FunctionComponent<LookupInputProps> = ({
         <label htmlFor="search" className="sr-only">
           Search
         </label>
-        <div className="relative">
+        <div className="relative flex">
           <div className="pointer-events-none absolute inset-y-0 left-0 flex items-center pl-3">
             <MagnifyingGlassIcon
               className="h-5 w-5 text-zinc-400"
@@ -47,20 +84,109 @@ const LookupInput: FunctionComponent<LookupInputProps> = ({
             />
           </div>
           <input
-            className="lg:py-4.5 block w-full rounded-md border border-transparent bg-zinc-700 py-2 pl-10 pr-3 text-sm placeholder-zinc-400 placeholder:translate-y-2 focus:text-zinc-200 focus:outline-none sm:text-sm md:py-3 md:text-base lg:text-lg"
+            className={classNames(
+              "lg:py-4.5 custom-select block w-full rounded-l-md border border-transparent",
+              "bg-zinc-700 py-2 pl-10 pr-3 text-sm placeholder-zinc-400 placeholder:translate-y-2 focus:text-zinc-200",
+              " focus:outline-none sm:text-sm md:py-3 md:text-base lg:text-lg"
+            )}
             disabled={isLoading}
             placeholder="A domain, an IP address, a TLD, an RDAP URL..."
             type="search"
             {...register("target", {
               required: true,
               onChange: () => {
-                if (onChange != undefined) onChange(getValues("target"));
+                if (onChange != undefined)
+                  onChange({
+                    target: getValues("target"),
+                    targetType: getTargetType(),
+                  });
               },
             })}
-          />
+          ></input>
+          <Listbox
+            value={selected}
+            onChange={(value) => {
+              setSelected(value);
+
+              if (onChange != undefined)
+                onChange({
+                  target: getValues("target"),
+                  targetType: getTargetType(value),
+                });
+            }}
+            disabled={isLoading}
+          >
+            <div className="relative min-w-[10rem]">
+              <Listbox.Button
+                className={classNames(
+                  "relative h-full w-full cursor-default rounded-r-lg bg-zinc-700 py-2 pl-3 pr-10",
+                  "text-left focus:outline-none focus-visible:border-indigo-500 sm:text-sm",
+                  "focus-visible:ring-2 focus-visible:ring-white focus-visible:ring-opacity-75 focus-visible:ring-offset-2 focus-visible:ring-offset-orange-300 "
+                )}
+              >
+                {/* Fetch special text for 'auto' mode, otherwise just use the options. */}
+                <span className="block">
+                  {selected == "auto" ? getAutoTargetText() : options[selected]}
+                </span>
+                <span className="pointer-events-none absolute inset-y-0 right-0 flex items-center pr-2">
+                  <ChevronUpDownIcon
+                    className="h-5 w-5 text-zinc-200"
+                    aria-hidden="true"
+                  />
+                </span>
+              </Listbox.Button>
+              <Transition
+                as={Fragment}
+                leave="transition ease-in duration-100"
+                leaveFrom="opacity-100"
+                leaveTo="opacity-0"
+              >
+                <Listbox.Options
+                  className={classNames(
+                    "scrollbar-thin absolute mt-1 max-h-60 w-full overflow-auto rounded-md bg-zinc-700 py-1",
+                    "text-zinc-200 shadow-lg ring-1 ring-black ring-opacity-5 focus:outline-none sm:text-sm"
+                  )}
+                >
+                  {Object.entries(options).map(([key, value]) => (
+                    <Listbox.Option
+                      key={key}
+                      className={({ active }) =>
+                        classNames(
+                          "relative cursor-default select-none py-2 pl-10 pr-4",
+                          active ? "bg-zinc-800 text-zinc-300" : null
+                        )
+                      }
+                      value={key}
+                    >
+                      {({ selected }) => (
+                        <>
+                          <span
+                            className={classNames(
+                              "block",
+                              selected ? "font-medium" : null
+                            )}
+                          >
+                            {value}
+                          </span>
+                          {selected ? (
+                            <span className="absolute inset-y-0 left-0 flex items-center pl-3 text-blue-500">
+                              <CheckIcon
+                                className="h-5 w-5"
+                                aria-hidden="true"
+                              />
+                            </span>
+                          ) : null}
+                        </>
+                      )}
+                    </Listbox.Option>
+                  ))}
+                </Listbox.Options>
+              </Transition>
+            </div>
+          </Listbox>
         </div>
       </div>
-      <div className="col p-0">
+      <div className="col">
         <div className="flex flex-wrap pt-3 pb-1 text-sm">
           <div className="whitespace-nowrap">
             <input
diff --git a/src/pages/index.tsx b/src/pages/index.tsx
index 28b9447..76714ae 100644
--- a/src/pages/index.tsx
+++ b/src/pages/index.tsx
@@ -38,7 +38,7 @@ const Index: NextPage = () => {
       <div className="mx-auto max-w-screen-sm px-5 lg:max-w-screen-md xl:max-w-screen-lg">
         <div className="dark container mx-auto w-full py-6 md:py-12 ">
           <LookupInput
-            onChange={(target) => {
+            onChange={({ target, targetType }) => {
               setTarget(target);
             }}
             onSubmit={async (props) => {