slash-e/extension/src/components/PullShortcutsButton.tsx

import { IconButton } from "@mui/joy";
import { useStorage } from "@plasmohq/storage/hook";
import axios from "axios";
import { useEffect } from "react";
import { toast } from "react-hot-toast";
import { ListShortcutsResponse } from "@/types/proto/api/v2/shortcut_service_pb";
import Icon from "./Icon";

const PullShortcutsButton = () => {
  const [domain] = useStorage("domain");
  const [accessToken] = useStorage("access_token");
  const [, setShortcuts] = useStorage("shortcuts");

  useEffect(() => {
    if (domain && accessToken) {
      handlePullShortcuts(true);
    }
  }, [domain, accessToken]);

  const handlePullShortcuts = async (silence = false) => {
    try {
      const {
        data: { shortcuts },
      } = await axios.get<ListShortcutsResponse>(`${domain}/api/v2/shortcuts`, {
        headers: {
          Authorization: `Bearer ${accessToken}`,
        },
      });
      setShortcuts(shortcuts);
      if (!silence) {
        toast.success("Shortcuts pulled");
      }
    } catch (error) {
      toast.error("Failed to pull shortcuts, error: " + error.message);
    }
  };

  return (
    <IconButton color="neutral" variant="plain" size="sm" onClick={() => handlePullShortcuts()}>
      <Icon.RefreshCcw className="w-4 h-auto" />
    </IconButton>
  );
};

export default PullShortcutsButton;