|
| 1 | +import { Menu, Transition } from '@headlessui/react'; |
| 2 | +import { Fragment } from 'react'; |
| 3 | +import { BsChevronDown, BsFillTrashFill } from 'react-icons/bs'; |
| 4 | +import { FiExternalLink } from 'react-icons/fi'; |
| 5 | +import { SiDropbox, SiGithub, SiGoogledrive, SiMega, SiMicrosoftonedrive } from 'react-icons/si'; |
| 6 | +import Logo from '../../../public/logo.svg'; |
| 7 | +import { isSelfHosted, toDirectDownloadLink } from '../../lib/download-util'; |
| 8 | +import { joinClasses } from '../../lib/util'; |
| 9 | +import { Link } from './link'; |
| 10 | + |
| 11 | +type DatasetDownloadButtonProps = { |
| 12 | + url: string; |
| 13 | + readonly?: boolean; |
| 14 | + onChange?: (url: string) => void; |
| 15 | +}; |
| 16 | + |
| 17 | +const hostFromUrl = (url: string): string => { |
| 18 | + try { |
| 19 | + const parsedUrl = new URL(url); |
| 20 | + const domainParts = parsedUrl.hostname.split('.'); |
| 21 | + const domainAndTld = domainParts.slice(domainParts.length - 2).join('.'); |
| 22 | + |
| 23 | + if (domainAndTld === 'github.com') { |
| 24 | + return 'GitHub'; |
| 25 | + } |
| 26 | + if (parsedUrl.hostname === 'drive.google.com') { |
| 27 | + return 'Google Drive'; |
| 28 | + } |
| 29 | + if (parsedUrl.hostname === 'cdn.discordapp.com') { |
| 30 | + return 'Discord'; |
| 31 | + } |
| 32 | + if (domainAndTld === '1drv.ms') { |
| 33 | + return 'OneDrive'; |
| 34 | + } |
| 35 | + if (domainAndTld === 'mega.nz') { |
| 36 | + return 'Mega'; |
| 37 | + } |
| 38 | + if (domainAndTld === 'mediafire.com') { |
| 39 | + return 'MediaFire'; |
| 40 | + } |
| 41 | + if (domainAndTld === 'pcloud.link') { |
| 42 | + return 'pCloud'; |
| 43 | + } |
| 44 | + if (domainAndTld === 'icedrive.net') { |
| 45 | + return 'Icedrive'; |
| 46 | + } |
| 47 | + if (domainAndTld === 'dropbox.com') { |
| 48 | + return 'Dropbox'; |
| 49 | + } |
| 50 | + return parsedUrl.hostname; |
| 51 | + } catch (e) { |
| 52 | + console.debug(e); |
| 53 | + return 'an unknown hoster'; |
| 54 | + } |
| 55 | +}; |
| 56 | + |
| 57 | +const iconFromHost = (host: string) => { |
| 58 | + switch (host) { |
| 59 | + case 'GitHub': |
| 60 | + return <SiGithub className="block" />; |
| 61 | + case 'Google Drive': |
| 62 | + return <SiGoogledrive className="block" />; |
| 63 | + case 'OneDrive': |
| 64 | + return <SiMicrosoftonedrive className="block" />; |
| 65 | + case 'Mega': |
| 66 | + return <SiMega className="block" />; |
| 67 | + case 'Dropbox': |
| 68 | + return <SiDropbox className="block" />; |
| 69 | + default: |
| 70 | + return <FiExternalLink className="block" />; |
| 71 | + } |
| 72 | +}; |
| 73 | + |
| 74 | +const isMirrorExternal = (url: string) => !isSelfHosted(url); |
| 75 | + |
| 76 | +export const DatasetDownloadButton = ({ url, readonly, onChange }: DatasetDownloadButtonProps) => { |
| 77 | + const isExternal = isMirrorExternal(url); |
| 78 | + const host = hostFromUrl(url); |
| 79 | + |
| 80 | + const showMenu = !readonly; |
| 81 | + |
| 82 | + return ( |
| 83 | + <div className="flex w-full flex-row gap-0.5 rounded-xl bg-accent-500 dark:bg-accent-400"> |
| 84 | + <Link |
| 85 | + external |
| 86 | + className={joinClasses( |
| 87 | + 'inline-flex h-20 w-full cursor-pointer items-center rounded-l-lg border-0 bg-accent-600 text-center text-lg font-medium text-white transition duration-100 ease-in-out hover:bg-accent-500 dark:bg-accent-500 dark:hover:bg-accent-600', |
| 88 | + !showMenu && 'rounded-r-lg' |
| 89 | + )} |
| 90 | + href={toDirectDownloadLink(url)} |
| 91 | + type="button" |
| 92 | + > |
| 93 | + <div className="w-full"> |
| 94 | + {isExternal ? ( |
| 95 | + <FiExternalLink |
| 96 | + className="mr-2 h-4 w-4" |
| 97 | + viewBox="0 0 22 22" |
| 98 | + /> |
| 99 | + ) : ( |
| 100 | + <svg |
| 101 | + className="mr-2 h-4 w-4 fill-current" |
| 102 | + viewBox="0 0 20 20" |
| 103 | + xmlns="http://www.w3.org/2000/svg" |
| 104 | + > |
| 105 | + <path d="M13 8V2H7v6H2l8 8 8-8h-5zM0 18h20v2H0v-2z" /> |
| 106 | + </svg> |
| 107 | + )} |
| 108 | + Visit Dataset Link |
| 109 | + {isExternal && ( |
| 110 | + <div className="text-center text-sm font-normal"> |
| 111 | + <span className="whitespace-nowrap px-1">Hosted by {host}</span> |
| 112 | + </div> |
| 113 | + )} |
| 114 | + </div> |
| 115 | + </Link> |
| 116 | + |
| 117 | + {showMenu && ( |
| 118 | + <Menu |
| 119 | + as="div" |
| 120 | + className="relative inline-block text-left" |
| 121 | + > |
| 122 | + <div> |
| 123 | + <Menu.Button |
| 124 | + aria-label="Edit dataset URL" |
| 125 | + className="inline-flex h-20 w-12 cursor-pointer items-center rounded-r-lg border-0 bg-accent-600 text-center align-middle text-lg font-medium text-white transition duration-100 ease-in-out hover:bg-accent-500 dark:bg-accent-500 dark:hover:bg-accent-600" |
| 126 | + > |
| 127 | + <BsChevronDown className="w-full" /> |
| 128 | + </Menu.Button> |
| 129 | + </div> |
| 130 | + <Transition |
| 131 | + as={Fragment} |
| 132 | + enter="transition ease-out duration-100" |
| 133 | + enterFrom="transform opacity-0 scale-95" |
| 134 | + enterTo="transform opacity-100 scale-100" |
| 135 | + leave="transition ease-in duration-75" |
| 136 | + leaveFrom="transform opacity-100 scale-100" |
| 137 | + leaveTo="transform opacity-0 scale-95" |
| 138 | + > |
| 139 | + <Menu.Items className="absolute right-0 z-50 mt-2 w-56 origin-top-right divide-y divide-gray-100 rounded-lg bg-fade-100 focus:outline-none dark:bg-fade-700"> |
| 140 | + <div className="flex flex-col divide-y rounded-lg p-2 shadow-lg"> |
| 141 | + {url !== '' && ( |
| 142 | + <Menu.Item |
| 143 | + as="a" |
| 144 | + className="flex cursor-pointer rounded-lg p-2 transition-colors duration-100 ease-in-out ui-active:bg-fade-300 ui-active:text-black ui-not-active:text-black dark:ui-active:bg-fade-600 dark:ui-active:text-white dark:ui-not-active:text-white" |
| 145 | + onClick={() => { |
| 146 | + const newUrl = prompt('Edit URL', url); |
| 147 | + if (newUrl !== null && onChange) { |
| 148 | + onChange(newUrl); |
| 149 | + } |
| 150 | + }} |
| 151 | + > |
| 152 | + {isExternal ? ( |
| 153 | + <div className="flex h-full w-full flex-row items-center gap-2 align-middle"> |
| 154 | + <div className="m-0 block h-full align-middle"> |
| 155 | + {iconFromHost(host)} |
| 156 | + </div> |
| 157 | + <div className="m-0 h-full align-middle">{host}</div> |
| 158 | + </div> |
| 159 | + ) : ( |
| 160 | + <Logo /> |
| 161 | + )} |
| 162 | + {!readonly && ( |
| 163 | + <button |
| 164 | + onClick={(e) => { |
| 165 | + e.stopPropagation(); |
| 166 | + if (onChange) { |
| 167 | + onChange(''); |
| 168 | + } |
| 169 | + }} |
| 170 | + > |
| 171 | + <BsFillTrashFill /> |
| 172 | + </button> |
| 173 | + )} |
| 174 | + </Menu.Item> |
| 175 | + )} |
| 176 | + {!readonly && url === '' && ( |
| 177 | + <Menu.Item |
| 178 | + as="a" |
| 179 | + className="cursor-pointer rounded-lg p-2 text-center transition-colors duration-100 ease-in-out ui-active:bg-fade-300 ui-active:text-black ui-not-active:text-black dark:ui-active:bg-fade-600 dark:ui-active:text-white dark:ui-not-active:text-white" |
| 180 | + onClick={() => { |
| 181 | + const newUrl = prompt('Enter a new URL'); |
| 182 | + if (newUrl && onChange) { |
| 183 | + onChange(newUrl); |
| 184 | + } |
| 185 | + }} |
| 186 | + > |
| 187 | + + Add URL |
| 188 | + </Menu.Item> |
| 189 | + )} |
| 190 | + </div> |
| 191 | + </Menu.Items> |
| 192 | + </Transition> |
| 193 | + </Menu> |
| 194 | + )} |
| 195 | + </div> |
| 196 | + ); |
| 197 | +}; |
0 commit comments