133 lines
3.4 KiB
JavaScript

import React, { useState } from "react"
import PropTypes from "prop-types"
import { List } from "immutable"
import { connect } from "react-redux"
import { addTorrent } from "../../actions/torrents"
import Dropdown from "react-bootstrap/Dropdown"
function buildMenuItems(torrents) {
if (!torrents || torrents.size === 0) {
return [];
}
const t = torrents.groupBy((el) => el.get("source"));
// Build the array of entries
let entries = [];
let dividerCount = t.size - 1;
for (let [source, torrentList] of t.entrySeq()) {
// Push the title
entries.push({
type: "header",
value: source,
});
// Push the torrents
for (let torrent of torrentList) {
entries.push({
type: "entry",
quality: torrent.get("quality"),
url: torrent.get("url"),
});
}
// Push the divider
if (dividerCount > 0) {
dividerCount--;
entries.push({ type: "divider" });
}
}
return entries;
}
const torrentsButton = ({
torrents,
search,
searching,
addTorrent,
url,
}) => {
/* eslint-disable */
const [show, setShow] = useState(false);
/* eslint-enable */
const entries = buildMenuItems(torrents);
const count = (torrents && torrents.size !== 0) ? torrents.size : 0;
const onSelect = (eventKey) => {
// Close the dropdown if the eventkey is not specified
if (eventKey === null) {
setShow(false);
}
}
const onToggle = (isOpen, event, metadata) => {
// Don't close on select
if (metadata && metadata.source !== "select") {
setShow(isOpen);
}
}
return (
<span className="mr-1 mb-1">
<Dropdown drop="up" show={show} onToggle={onToggle} onSelect={onSelect}>
<Dropdown.Toggle variant="secondary" bsPrefix="btn-sm w-md-100">
<i className="fa fa-magnet mr-1" />
Torrents
<span className="ml-1 badge badge-pill badge-info">
{count}
</span>
</Dropdown.Toggle>
<Dropdown.Menu>
<Dropdown.Item eventKey={1} onClick={search} >
<i className={`fa ${ searching ? "fa-spin" : "" } fa-refresh mr-1`} />
Automatic search
</Dropdown.Item>
<Dropdown.Item href={url} >
<i className="fa fa-search mr-1" />
Manual search
</Dropdown.Item>
{entries.length > 0 &&
<Dropdown.Divider />
}
{entries.map((e, index) => {
switch (e.type) {
case "header":
return (
<Dropdown.Header key={index}>
<span className="text-warning">{e.value}</span>
</Dropdown.Header>
);
case "divider":
return (
<Dropdown.Divider key={index}/>
);
case "entry":
return (
<Dropdown.Item key={index} onClick={() => addTorrent(e.url)}>
{e.quality}
</Dropdown.Item>
);
}
})}
</Dropdown.Menu>
</Dropdown>
</span>
);
}
torrentsButton.propTypes = {
torrents: PropTypes.instanceOf(List),
searching: PropTypes.bool,
search: PropTypes.func.isRequired,
addTorrent: PropTypes.func.isRequired,
url: PropTypes.string,
}
torrentsButton.defaultProps = {
torrents: List(),
}
export const TorrentsButton = connect(null, {addTorrent})(torrentsButton);