import React from 'react'; import { StackNavigationOptions, StackNavigationProp } from '@react-navigation/stack'; import { CompositeNavigationProp, RouteProp } from '@react-navigation/core'; import { FlatList, Text, View } from 'react-native'; import { Q } from '@nozbe/watermelondb'; import { connect } from 'react-redux'; import { dequal } from 'dequal'; import { FormTextInput } from '../../containers/TextInput'; import ActivityIndicator from '../../containers/ActivityIndicator'; import Markdown from '../../containers/markdown'; import Message from '../../containers/message'; import scrollPersistTaps from '../../lib/methods/helpers/scrollPersistTaps'; import I18n from '../../i18n'; import StatusBar from '../../containers/StatusBar'; import log from '../../lib/methods/helpers/log'; import { themes } from '../../lib/constants'; import { TSupportedThemes, withTheme } from '../../theme'; import { getUserSelector } from '../../selectors/login'; import SafeAreaView from '../../containers/SafeAreaView'; import * as HeaderButton from '../../containers/HeaderButton'; import database from '../../lib/database'; import { sanitizeLikeString } from '../../lib/database/utils'; import getThreadName from '../../lib/methods/getThreadName'; import getRoomInfo, { IRoomInfoResult } from '../../lib/methods/getRoomInfo'; import styles from './styles'; import { InsideStackParamList, ChatsStackParamList } from '../../stacks/types'; import { compareServerVersion, debounce, isIOS } from '../../lib/methods/helpers'; import { IMessageFromServer, IUser, TMessageModel, IUrl, IAttachment, ISubscription, SubscriptionType, TSubscriptionModel, TGetCustomEmoji, ICustomEmoji } from '../../definitions'; import { Services } from '../../lib/services'; import KeyboardView from '../../containers/KeyboardView'; const QUERY_SIZE = 50; interface ISearchMessagesViewState { loading: boolean; messages: (IMessageFromServer | TMessageModel)[]; searchText: string; } export interface IRoomInfoParam { room?: ISubscription; member?: any; rid: string; t: SubscriptionType; joined?: boolean; } interface INavigationOption { navigation: CompositeNavigationProp< StackNavigationProp, StackNavigationProp >; route: RouteProp; } interface ISearchMessagesViewProps extends INavigationOption { user: IUser; baseUrl: string; serverVersion: string; customEmojis: { [key: string]: ICustomEmoji; }; theme: TSupportedThemes; useRealName: boolean; } class SearchMessagesView extends React.Component { private offset: number; private rid: string; private t: SubscriptionType; private encrypted: boolean | undefined; private room?: IRoomInfoResult; static navigationOptions = ({ navigation, route }: INavigationOption) => { const options: StackNavigationOptions = { title: I18n.t('Search') }; const showCloseModal = route.params?.showCloseModal; if (showCloseModal) { options.headerLeft = () => ; } return options; }; constructor(props: ISearchMessagesViewProps) { super(props); this.state = { loading: false, messages: [], searchText: '' }; this.offset = 0; this.rid = props.route.params.rid; this.t = props.route.params?.t; this.encrypted = props.route.params?.encrypted; } async componentDidMount() { this.room = (await getRoomInfo(this.rid)) ?? undefined; } shouldComponentUpdate(nextProps: ISearchMessagesViewProps, nextState: ISearchMessagesViewState) { const { loading, searchText, messages } = this.state; const { theme } = this.props; if (nextProps.theme !== theme) { return true; } if (nextState.loading !== loading) { return true; } if (nextState.searchText !== searchText) { return true; } if (!dequal(nextState.messages, messages)) { return true; } return false; } componentWillUnmount() { this.searchDebounced?.stop?.(); } // Handle encrypted rooms search messages searchMessages = async (searchText: string): Promise<(IMessageFromServer | TMessageModel)[]> => { if (!searchText) { return []; } // If it's a encrypted, room we'll search only on the local stored messages if (this.encrypted) { const db = database.active; const messagesCollection = db.get('messages'); const likeString = sanitizeLikeString(searchText); return messagesCollection .query( // Messages of this room Q.where('rid', this.rid), // Message content is like the search text Q.where('msg', Q.like(`%${likeString}%`)) ) .fetch(); } // If it's not a encrypted room, search messages on the server const result = await Services.searchMessages(this.rid, searchText, QUERY_SIZE, this.offset); if (result.success) { const urlRenderMessages = result.messages?.map(message => { if (message.urls && message.urls.length > 0) { message.urls = message.urls?.map((url, index) => { if (url.meta) { return { _id: index, title: url.meta.pageTitle, description: url.meta.ogDescription, image: url.meta.ogImage, url: url.url } as IUrl; } return {} as IUrl; }); } return message; }); this.offset += QUERY_SIZE; return urlRenderMessages; } return []; }; getMessages = async (searchText: string, debounced?: boolean) => { try { const messages = await this.searchMessages(searchText); this.setState(prevState => ({ messages: debounced ? messages : [...prevState.messages, ...messages], loading: false })); } catch (e) { this.setState({ loading: false }); log(e); } }; search = (searchText: string) => { this.offset = 0; this.setState({ searchText, loading: true, messages: [] }); this.searchDebounced(searchText); }; searchDebounced = debounce(async (searchText: string) => { await this.getMessages(searchText, true); }, 1000); getCustomEmoji: TGetCustomEmoji = name => { const { customEmojis } = this.props; const emoji = customEmojis[name]; if (emoji) { return emoji; } return null; }; showAttachment = (attachment: IAttachment) => { const { navigation } = this.props; navigation.navigate('AttachmentView', { attachment }); }; navToRoomInfo = (navParam: IRoomInfoParam) => { const { navigation, user } = this.props; if (navParam.rid === user.id) { return; } navigation.navigate('RoomInfoView', navParam); }; jumpToMessage = async ({ item }: { item: IMessageFromServer | TMessageModel }) => { const { navigation } = this.props; let params: { rid: string; jumpToMessageId: string; t: SubscriptionType; room: TSubscriptionModel | undefined; tmid?: string; name?: string; } = { rid: this.rid, jumpToMessageId: item._id, t: this.t, room: this.room as TSubscriptionModel }; if ('tmid' in item && item.tmid) { navigation.pop(); params = { ...params, tmid: item.tmid, name: await getThreadName(this.rid, item.tmid as string, item._id), t: SubscriptionType.THREAD }; navigation.push('RoomView', params); } else { navigation.navigate('RoomView', params); } }; onEndReached = async () => { const { serverVersion } = this.props; const { searchText, messages, loading } = this.state; if ( messages.length < this.offset || this.encrypted || loading || compareServerVersion(serverVersion, 'lowerThan', '3.17.0') ) { return; } this.setState({ loading: true }); await this.getMessages(searchText); }; renderEmpty = () => { const { theme } = this.props; return ( {I18n.t('No_results_found')} ); }; renderItem = ({ item }: { item: IMessageFromServer | TMessageModel }) => { const message = item as TMessageModel; const { user, baseUrl, theme, useRealName } = this.props; return ( this.jumpToMessage({ item })} jumpToMessage={() => this.jumpToMessage({ item })} rid={message.rid} /> ); }; renderList = () => { const { messages, loading, searchText } = this.state; const { theme } = this.props; if (!loading && messages.length === 0 && searchText.length) { return this.renderEmpty(); } return ( item._id} onEndReached={this.onEndReached} ListFooterComponent={loading ? : null} onEndReachedThreshold={0.5} removeClippedSubviews={isIOS} {...scrollPersistTaps} /> ); }; render() { const { theme } = this.props; return ( {this.renderList()} ); } } const mapStateToProps = (state: any) => ({ serverVersion: state.server.version, baseUrl: state.server.server, user: getUserSelector(state), useRealName: state.settings.UI_Use_Real_Name, customEmojis: state.customEmojis }); export default connect(mapStateToProps)(withTheme(SearchMessagesView));