71 lines
2.1 KiB
TypeScript
71 lines
2.1 KiB
TypeScript
import { Button, Tooltip } from 'antd';
|
|
import { BookmarkCheck, Bookmark as BookmarkIcn, Vault } from 'lucide-react';
|
|
import React, { useEffect, useState } from 'react';
|
|
import { connect } from 'react-redux';
|
|
import { toast } from 'react-toastify';
|
|
import { useStore } from 'App/mstore';
|
|
import { observer } from 'mobx-react-lite';
|
|
|
|
interface Props {
|
|
sessionId: any;
|
|
isEnterprise: boolean;
|
|
}
|
|
|
|
function Bookmark(props: Props) {
|
|
const { sessionStore } = useStore();
|
|
const favorite = sessionStore.current.favorite;
|
|
const onToggleFavorite = sessionStore.toggleFavorite;
|
|
const { sessionId, isEnterprise } = props;
|
|
const [isFavorite, setIsFavorite] = useState(favorite);
|
|
|
|
const ADDED_MESSAGE = isEnterprise
|
|
? 'Session added to vault'
|
|
: 'Session added to your bookmarks';
|
|
const REMOVED_MESSAGE = isEnterprise
|
|
? 'Session removed from vault'
|
|
: 'Session removed from your bookmarks';
|
|
const TOOLTIP_TEXT_ADD = isEnterprise ? 'Add to vault' : 'Add to bookmarks';
|
|
const TOOLTIP_TEXT_REMOVE = isEnterprise
|
|
? 'Remove from vault'
|
|
: 'Remove from bookmarks';
|
|
|
|
useEffect(() => {
|
|
setIsFavorite(favorite);
|
|
}, [favorite]);
|
|
|
|
const toggleFavorite = async () => {
|
|
onToggleFavorite(sessionId).then(() => {
|
|
toast.success(isFavorite ? REMOVED_MESSAGE : ADDED_MESSAGE);
|
|
setIsFavorite(!isFavorite);
|
|
});
|
|
};
|
|
|
|
const icon = isEnterprise ? (
|
|
<Vault size={16} strokeWidth={1} />
|
|
) : isFavorite ? (
|
|
<BookmarkCheck size={16} strokeWidth={1} />
|
|
) : (
|
|
<BookmarkIcn size={16} strokeWidth={1} />
|
|
);
|
|
|
|
return (
|
|
<div onClick={toggleFavorite} className="w-full">
|
|
<Tooltip title={isFavorite ? TOOLTIP_TEXT_REMOVE : TOOLTIP_TEXT_ADD} placement='bottom'>
|
|
<Button
|
|
type={isFavorite ? 'primary' : undefined}
|
|
ghost={isFavorite}
|
|
size={'small'}
|
|
className={'flex items-center justify-center'}
|
|
>
|
|
{icon}
|
|
</Button>
|
|
</Tooltip>
|
|
</div>
|
|
);
|
|
}
|
|
|
|
export default connect(
|
|
(state: any) => ({
|
|
isEnterprise: state.getIn(['user', 'account', 'edition']) === 'ee',
|
|
}),
|
|
)(observer(Bookmark));
|