All files / apps/host/src/components/ChatBox useChatBox.ts

95.65% Statements 44/46
81.25% Branches 26/32
100% Functions 7/7
97.61% Lines 41/42

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148                2x                             13x   13x 13x   13x 11x   1x   11x   1x     11x 11x 11x       13x 5x 5x 5x   4x               4x 1x     3x   2x   2x 1x       1x     1x             1x       13x 4x 4x 4x   3x 3x                   2x       2x   2x   2x 1x       1x     1x             1x       13x                                        
import { useEffect, useRef, useState } from 'react';
import { launchImageLibrary } from 'react-native-image-picker';
import { Alert, Keyboard, Platform, TextInput } from 'react-native';
 
import { useChat } from '@/contexts/ChatContext';
 
import { requestCameraPermission, requestPhotoLibraryPermission } from '@/utils/permissions';
 
export const useChatBox = () => {
  const {
    messages,
    text,
    setText,
    isStreaming,
    isInitializing,
    processingActionIds,
    selectedImage,
    setSelectedImage,
    onSend,
    onReset,
    handleCancel,
    handleAction,
    initThread,
  } = useChat();
 
  const [keyboardHeight, setKeyboardHeight] = useState(0);
  const inputRef = useRef<TextInput>(null);
 
  useEffect(() => {
    const showSubscription = Keyboard.addListener(
      Platform.OS === 'ios' ? 'keyboardWillShow' : 'keyboardDidShow',
      e => setKeyboardHeight(e.endCoordinates.height),
    );
    const hideSubscription = Keyboard.addListener(
      Platform.OS === 'ios' ? 'keyboardWillHide' : 'keyboardDidHide',
      () => setKeyboardHeight(0),
    );
 
    return () => {
      showSubscription.remove();
      hideSubscription.remove();
    };
  }, []);
 
  const handlePickImage = async () => {
    try {
      const hasPermission = await requestPhotoLibraryPermission();
      if (!hasPermission) return;
 
      const result = await launchImageLibrary({
        mediaType: 'photo',
        quality: 0.8,
        maxWidth: 1280,
        maxHeight: 1280,
        includeBase64: true,
      });
 
      if (result.errorCode) {
        throw new Error(result.errorMessage || result.errorCode);
      }
 
      if (result.didCancel || !result.assets?.[0]?.uri) return;
 
      const asset = result.assets[0];
 
      if (asset.fileSize && asset.fileSize > 700 * 1024) {
        Alert.alert(
          'Image Size Limit Exceeded',
          'The selected image exceeds the 700KB size limit. Please choose a smaller image.',
        );
        return;
      }
 
      setSelectedImage({
        uri: asset.uri!,
        fileName: asset.fileName,
        type: asset.type,
        base64: asset.base64,
      });
    } catch (error) {
      console.error('Failed to pick image', error);
    }
  };
 
  const handleLaunchCamera = async () => {
    try {
      const hasPermission = await requestCameraPermission();
      if (!hasPermission) return;
 
      const { launchCamera } = await import('react-native-image-picker');
      const result = await launchCamera({
        mediaType: 'photo',
        quality: 0.7,
        maxWidth: 1280,
        maxHeight: 1280,
        saveToPhotos: false,
        presentationStyle: 'fullScreen',
        includeBase64: true,
      });
 
      Iif (result.errorCode) {
        throw new Error(result.errorMessage || result.errorCode);
      }
 
      Iif (result.didCancel || !result.assets?.[0]?.uri) return;
 
      const asset = result.assets[0];
 
      if (asset.fileSize && asset.fileSize > 700 * 1024) {
        Alert.alert(
          'Image Size Limit Exceeded',
          'The captured image exceeds the 700KB size limit. Please take another picture.',
        );
        return;
      }
 
      setSelectedImage({
        uri: asset.uri!,
        fileName: asset.fileName,
        type: asset.type,
        base64: asset.base64,
      });
    } catch (error) {
      console.error('Failed to launch camera', error);
    }
  };
 
  return {
    messages,
    text,
    setText,
    isStreaming,
    isInitializing,
    processingActionIds,
    selectedImage,
    setSelectedImage,
    inputRef,
    keyboardHeight,
    handlePickImage,
    handleLaunchCamera,
    onSend,
    onReset,
    handleCancel,
    handleAction,
    initThread,
  };
};