HD
63 lines
1.7 KiB
TypeScript
63 lines
1.7 KiB
TypeScript
import { useState, useEffect } from "react";
|
|
import { useLazyRemoveFavoriteQuery, useFavoriteMessageMutation } from "../../app/services/message";
|
|
import { useAppSelector } from "../../app/store";
|
|
|
|
export default function useFavMessage({ cid = null, uid = null }) {
|
|
const [removeFav] = useLazyRemoveFavoriteQuery();
|
|
const [addFav] = useFavoriteMessageMutation();
|
|
const [favorites, setFavorites] = useState([]);
|
|
const { favs = [] } = useAppSelector((store) => {
|
|
return { favs: store.favorites };
|
|
});
|
|
|
|
const addFavorite = async (mid = []) => {
|
|
const mids = Array.isArray(mid) ? mid.map((i) => +i) : [+mid];
|
|
if (mids.length == 0) return;
|
|
const { error = null } = await addFav(mids);
|
|
return !error;
|
|
};
|
|
|
|
const removeFavorite = (id) => {
|
|
if (!id) return;
|
|
removeFav(id);
|
|
};
|
|
|
|
const isFavorited = (mid = null) => {
|
|
if (!mid) return false;
|
|
let mids = [];
|
|
favorites.forEach((f) => {
|
|
if (f.messages.length == 1) {
|
|
const ids = f.messages.map((m) => m.from_mid);
|
|
mids = [...mids, ...ids];
|
|
}
|
|
});
|
|
return mids.findIndex((i) => i == mid) > -1;
|
|
};
|
|
|
|
useEffect(() => {
|
|
let filtereds = [];
|
|
filtereds = cid
|
|
? favs.filter((f) => {
|
|
if (!f.messages) return false;
|
|
return f.messages.every((m) => m.source.gid == cid);
|
|
})
|
|
: favs;
|
|
filtereds = uid
|
|
? filtereds.filter((f) => {
|
|
if (!f.messages) return false;
|
|
return f.messages.every((m) => m.source.uid == uid);
|
|
})
|
|
: filtereds;
|
|
|
|
setFavorites(filtereds);
|
|
}, [cid, uid, favs]);
|
|
|
|
// console.log("filtered", cid, uid, favs);
|
|
return {
|
|
isFavorited,
|
|
addFavorite,
|
|
removeFavorite,
|
|
favorites
|
|
};
|
|
}
|