vn-verdnaturachat/app/lib/hooks/useFrequentlyUsedEmoji.ts

44 lines
1.5 KiB
TypeScript

import { useEffect, useState } from 'react';
import { unstable_batchedUpdates } from 'react-native';
import { Q } from '@nozbe/watermelondb';
import database from '../database';
import { IEmoji } from '../../definitions';
import { DEFAULT_EMOJIS } from '../constants';
export const useFrequentlyUsedEmoji = (
withDefaultEmojis = false
): {
frequentlyUsed: IEmoji[];
loaded: boolean;
} => {
const [frequentlyUsed, setFrequentlyUsed] = useState<IEmoji[]>([]);
const [loaded, setLoaded] = useState(false);
useEffect(() => {
const getFrequentlyUsedEmojis = async () => {
const db = database.active;
const frequentlyUsedRecords = await db.get('frequently_used_emojis').query(Q.experimentalSortBy('count', Q.desc)).fetch();
let frequentlyUsedEmojis = frequentlyUsedRecords.map(item => {
if (item.isCustom) {
return { name: item.content, extension: item.extension! }; // if isCustom is true, extension is not null
}
return item.content;
});
if (withDefaultEmojis && frequentlyUsedEmojis.length < DEFAULT_EMOJIS.length) {
frequentlyUsedEmojis = frequentlyUsedEmojis
.concat(DEFAULT_EMOJIS.filter(de => !frequentlyUsedEmojis.find(fue => typeof fue === 'string' && fue === de)))
.slice(0, DEFAULT_EMOJIS.length);
}
// TODO: remove once we update to React 18
unstable_batchedUpdates(() => {
setFrequentlyUsed(frequentlyUsedEmojis);
setLoaded(true);
});
};
getFrequentlyUsedEmojis();
}, []);
return { frequentlyUsed, loaded };
};