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

95.16% Statements 59/62
85.71% Branches 24/28
93.75% Functions 15/16
95.74% Lines 45/47

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                    2x 11x 11x 11x 11x 11x   11x 11x 11x 11x 11x 11x   11x 11x   11x     2x               11x 5x 5x 5x   4x               4x 1x     3x   2x   2x 1x       1x     1x             1x       11x 4x 4x 4x   3x 3x                   2x       2x   2x   2x 1x       1x     1x             1x       11x                                    
import { useRef } from 'react';
import { launchImageLibrary } from 'react-native-image-picker';
import { useKeyboardHandler } from 'react-native-keyboard-controller';
import { useSharedValue } from 'react-native-reanimated';
import { Alert, TextInput } from 'react-native';
 
import { useChatActionsSelector, useChatStateSelector } from '@/contexts/ChatContext';
 
import { requestCameraPermission, requestPhotoLibraryPermission } from '@/utils/permissions';
 
export const useChatBox = () => {
  const messages = useChatStateSelector(s => s.messages);
  const isStreaming = useChatStateSelector(s => s.isStreaming);
  const isInitializing = useChatStateSelector(s => s.isInitializing);
  const processingActionIds = useChatStateSelector(s => s.processingActionIds);
  const selectedImage = useChatStateSelector(s => s.selectedImage);
 
  const setSelectedImage = useChatActionsSelector(a => a.setSelectedImage);
  const onSend = useChatActionsSelector(a => a.onSend);
  const onReset = useChatActionsSelector(a => a.onReset);
  const handleCancel = useChatActionsSelector(a => a.handleCancel);
  const handleAction = useChatActionsSelector(a => a.handleAction);
  const initThread = useChatActionsSelector(a => a.initThread);
 
  const keyboardHeight = useSharedValue(0);
  const inputRef = useRef<TextInput>(null);
 
  useKeyboardHandler({
    onMove: e => {
      'worklet';
      keyboardHeight.value = e.height;
    },
    onEnd: e => {
      'worklet';
      keyboardHeight.value = e.height;
    },
  });
 
  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,
    isStreaming,
    isInitializing,
    processingActionIds,
    selectedImage,
    setSelectedImage,
    inputRef,
    keyboardHeight,
    handlePickImage,
    handleLaunchCamera,
    onSend,
    onReset,
    handleCancel,
    handleAction,
    initThread,
  };
};