riffusion-app/components/Share.tsx

import { Dialog, Transition } from "@headlessui/react";
import { Fragment, useState } from "react";
import { FiShare } from "react-icons/fi";
import styled from "styled-components";

import { InferenceResult } from "../types";

interface ShareProps {
  inferenceResults: InferenceResult[];
  nowPlayingResult: InferenceResult;
}

const ModalContainer = styled.div`
  position: absolute;
  top: 0;
  left: 0;
  width: 100vw;
  height: 100vh;
  background: rgba(0, 0, 0, 0.5);
  display: flex;
  align-items: center;
  justify-content: center;
`;

export default function Share({
  inferenceResults,
  nowPlayingResult,
}: ShareProps) {
  const [open, setOpen] = useState(false);

  var classNameCondition = "";
  if (open) {
    classNameCondition =
      "fixed z-20 top-24 right-4 md:top-28 md:right-8 bg-sky-400 w-14 h-14 rounded-full drop-shadow-lg flex justify-center items-center text-white text-2xl hover:bg-sky-500 hover:drop-shadow-2xl";
  } else {
    classNameCondition =
      "fixed z-20 top-24 right-4 md:top-28 md:right-8 bg-slate-100 w-14 h-14 rounded-full drop-shadow-lg flex justify-center items-center text-sky-900 text-2xl hover:text-white hover:bg-sky-600 hover:drop-shadow-2xl";
  }

  // function to copy link to moment in song to the clipboard
  function copyLinkToClipboard(secondsAgo: number) {
    // use generateLink to generate the link
    const link = generateLink(secondsAgo);

    navigator.clipboard.writeText(link);
  }

  function getActiveResult() {
    if (!nowPlayingResult) {
      if (inferenceResults.length == 0) {
        return null;
      }
      return inferenceResults[0];
    } else {
      return nowPlayingResult;
    }
  }

  // function to generate a link to a the moment in the song based on the played clips, input variable is how many seconds ago
  function generateLink(secondsAgo: number) {
    var prompt;
    var seed;
    var denoising;
    var maskImageId;
    var seedImageId;
    var guidance;
    var numInferenceSteps;
    var alphaVelocity;

    if (!nowPlayingResult) {
      return window.location.href;
    } else {
      var selectedInput: InferenceResult["input"];
      if (secondsAgo == 0) {
        selectedInput = nowPlayingResult.input;
      } else {
        var selectedCounter = nowPlayingResult.counter - secondsAgo / 5;
        selectedInput = inferenceResults.find(
          (result) => result.counter == selectedCounter
        )?.input;

        if (!selectedInput) {
          // TODO: ideally don't show the button in this case...
          return window.location.href;
        }
      }

      // TODO: Consider only including in the link the things that are different from the default values
      prompt = selectedInput.start.prompt;
      seed = selectedInput.start.seed;
      denoising = selectedInput.start.denoising;
      maskImageId = selectedInput.mask_image_id;
      seedImageId = nowPlayingResult.input.seed_image_id;

      // TODO, selectively add these based on whether we give user option to change them
      // guidance = nowPlayingResult.input.guidance
      // numInferenceSteps = nowPlayingResult.input.num_inference_steps
      // alphaVelocity = nowPlayingResult.input.alpha_velocity
    }

    var baseUrl = window.location.origin + "/?";

    if (prompt != null) {
      var promptString = "&prompt=" + prompt;
    } else {
      promptString = "";
    }
    if (seed != null) {
      var seedString = "&seed=" + seed;
    } else {
      seedString = "";
    }
    if (denoising != null) {
      var denoisingString = "&denoising=" + denoising;
    } else {
      denoisingString = "";
    }
    if (maskImageId != null) {
      var maskImageIdString = "&maskImageId=" + maskImageId;
    } else {
      maskImageIdString = "";
    }
    if (seedImageId != null) {
      var seedImageIdString = "&seedImageId=" + seedImageId;
    } else {
      seedImageIdString = "";
    }
    if (guidance != null) {
      var guidanceString = "&guidance=" + guidance;
    } else {
      guidanceString = "";
    }
    if (numInferenceSteps != null) {
      var numInferenceStepsString = "&numInferenceSteps=" + numInferenceSteps;
    } else {
      numInferenceStepsString = "";
    }
    if (alphaVelocity != null) {
      var alphaVelocityString = "&alphaVelocity=" + alphaVelocity;
    } else {
      alphaVelocityString = "";
    }

    // Format strings to have + in place of spaces for ease of sharing, note this is only necessary for prompts currently
    promptString = promptString.replace(/ /g, "+");

    // create url string with the variables above combined
    var shareUrl =
      baseUrl +
      promptString +
      seedString +
      denoisingString +
      maskImageIdString +
      seedImageIdString +
      guidanceString +
      numInferenceStepsString +
      alphaVelocityString;

    return shareUrl;
  }

  return (
    <>
      <button
        title="Info"
        className={classNameCondition}
        onClick={() => setOpen(true)}
      >
        <FiShare />
      </button>

      <Transition appear show={open} as={Fragment}>
        <Dialog
          as="div"
          className="fixed inset-0 z-20 overflow-y-auto"
          onClose={() => setOpen(false)}
        >
          <div className="min-h-screen px-4 text-center">
            <Transition.Child
              as={Fragment}
              enter="ease-out duration-300"
              enterFrom="opacity-0"
              enterTo="opacity-100"
              leave="ease-in duration-200"
              leaveFrom="opacity-100"
              leaveTo="opacity-0"
            >
              <Dialog.Overlay className="fixed inset-0" />
            </Transition.Child>

            <span
              className="inline-block h-screen align-middle"
              aria-hidden="true"
            >
              &#8203;
            </span>
            <Transition.Child
              as={Fragment}
              enter="ease-out duration-300"
              enterFrom="opacity-0 scale-95"
              enterTo="opacity-100 scale-100"
              leave="ease-in duration-200"
              leaveFrom="opacity-100 scale-100"
              leaveTo="opacity-0 scale-95"
            >
              <ModalContainer>
                <div className="my-8 p-8 inline-block w-full max-w-md transform overflow-hidden rounded-2xl bg-white text-left align-middle shadow-xl transition-all">
                  <Dialog.Title
                    as="h1"
                    className="text-3xl font-medium leading-6 text-gray-900 pb-2"
                  >
                    Share your riff
                  </Dialog.Title>
                  <div className="mt-4">
                    {!getActiveResult() && (
                      <div className="pl-8">
                        <div className="animate-spin rounded-full h-16 w-16 border-b-2 border-sky-900"></div>
                      </div>
                    )}
                    {getActiveResult() && (
                      <img
                        src={getActiveResult()?.image}
                        alt="share image"
                        className="w-4/5 sm:w-3/4"
                      />
                    )}
                  </div>

                  <div className="mt-4">
                    <audio
                      controls
                      src={getActiveResult()?.audio}
                      className="w-4/5 sm:w-3/4"
                    >
                      Your browser does not support audio.
                    </audio>
                  </div>
                  <div className="mt-6">
                    <button
                      className="relative inline-flex items-center justify-center p-0.5 mb-2 mr-2 overflow-hidden text-sm font-medium text-gray-900 rounded-lg group bg-sky-500 group-hover:from-sky-600 group-hover:to-sky-500 hover:text-white"
                      onClick={() => {
                        setOpen(false);
                      }}
                    >
                      <span className="relative px-2 sm:px-5 py-2 transition-all ease-in duration-75 bg-white rounded-md group-hover:bg-opacity-0">
                        Cancel
                      </span>
                    </button>

                    <button
                      type="button"
                      className="sm:w-64 text-white bg-gradient-to-br from-purple-600 to-sky-500 hover:bg-gradient-to-bl font-medium rounded-lg text-sm px-4 sm:px-5 py-2.5 text-center mr-2 mb-2"
                      onClick={() => {
                        copyLinkToClipboard(0);
                        setOpen(false);
                      }}
                    >
                      🔗 Copy link to current moment
                    </button>
                  </div>
                </div>
              </ModalContainer>
            </Transition.Child>
          </div>
        </Dialog>
      </Transition>
    </>
  );
}