Try to fix #16

src/components/Timeline/Shared/Actions.tsx

@@ -17,6 +17,7 @@ import { useQueryClient } from 'react-query'
 
 export interface Props {
   queryKey: QueryKeyTimeline
+  rootQueryKey?: QueryKeyTimeline
   status: Mastodon.Status
   accts: Mastodon.Account['acct'][] // When replying to conversations
   reblog: boolean
@@ -24,6 +25,7 @@ export interface Props {
 
 const TimelineActions: React.FC<Props> = ({
   queryKey,
+  rootQueryKey,
   status,
   accts,
   reblog
@@ -120,6 +122,7 @@ const TimelineActions: React.FC<Props> = ({
     mutation.mutate({
       type: 'updateStatusProperty',
       queryKey,
+      rootQueryKey,
       id: status.id,
       reblog,
       payload: {
@@ -139,6 +142,7 @@ const TimelineActions: React.FC<Props> = ({
     mutation.mutate({
       type: 'updateStatusProperty',
       queryKey,
+      rootQueryKey,
       id: status.id,
       reblog,
       payload: {
@@ -157,6 +161,7 @@ const TimelineActions: React.FC<Props> = ({
     mutation.mutate({
       type: 'updateStatusProperty',
       queryKey,
+      rootQueryKey,
       id: status.id,
       reblog,
       payload: {

src/components/Timeline/Shared/HeaderDefault.tsx

@@ -1,22 +1,27 @@
+import Icon from '@components/Icon'
+import { useNavigation } from '@react-navigation/native'
+import { QueryKeyTimeline } from '@utils/queryHooks/timeline'
 import { StyleConstants } from '@utils/styles/constants'
+import { useTheme } from '@utils/styles/ThemeManager'
 import React from 'react'
 import { Pressable, StyleSheet, View } from 'react-native'
 import HeaderSharedAccount from './HeaderShared/Account'
 import HeaderSharedApplication from './HeaderShared/Application'
 import HeaderSharedCreated from './HeaderShared/Created'
-import HeaderSharedVisibility from './HeaderShared/Visibility'
-import { QueryKeyTimeline } from '@utils/queryHooks/timeline'
 import HeaderSharedMuted from './HeaderShared/Muted'
-import { useNavigation } from '@react-navigation/native'
-import Icon from '@components/Icon'
-import { useTheme } from '@utils/styles/ThemeManager'
+import HeaderSharedVisibility from './HeaderShared/Visibility'
 
 export interface Props {
   queryKey?: QueryKeyTimeline
+  rootQueryKey?: QueryKeyTimeline
   status: Mastodon.Status
 }
 
-const TimelineHeaderDefault: React.FC<Props> = ({ queryKey, status }) => {
+const TimelineHeaderDefault: React.FC<Props> = ({
+  queryKey,
+  rootQueryKey,
+  status
+}) => {
   const navigation = useNavigation()
   const { theme } = useTheme()
 
@@ -38,6 +43,7 @@ const TimelineHeaderDefault: React.FC<Props> = ({ queryKey, status }) => {
           onPress={() =>
             navigation.navigate('Screen-Actions', {
               queryKey,
+              rootQueryKey,
               status,
               url: status.url || status.uri,
               type: 'status'

src/components/Timeline/Shared/Poll.tsx

@@ -10,6 +10,7 @@ import {
   QueryKeyTimeline,
   useTimelineMutation
 } from '@utils/queryHooks/timeline'
+import updateStatusProperty from '@utils/queryHooks/timeline/updateStatusProperty'
 import { StyleConstants } from '@utils/styles/constants'
 import { useTheme } from '@utils/styles/ThemeManager'
 import { maxBy } from 'lodash'
@@ -20,6 +21,7 @@ import { useQueryClient } from 'react-query'
 
 export interface Props {
   queryKey: QueryKeyTimeline
+  rootQueryKey?: QueryKeyTimeline
   statusId: Mastodon.Status['id']
   poll: NonNullable<Mastodon.Status['poll']>
   reblog: boolean
@@ -28,6 +30,7 @@ export interface Props {
 
 const TimelinePoll: React.FC<Props> = ({
   queryKey,
+  rootQueryKey,
   statusId,
   poll,
   reblog,
@@ -43,7 +46,19 @@ const TimelinePoll: React.FC<Props> = ({
   const queryClient = useQueryClient()
   const mutation = useTimelineMutation({
     queryClient,
-    onSuccess: true,
+    onSuccess: ({ body }, params) => {
+      const theParams = params as MutationVarsTimelineUpdateStatusProperty
+      queryClient.cancelQueries(queryKey)
+      rootQueryKey && queryClient.cancelQueries(rootQueryKey)
+
+      haptics('Success')
+      switch (theParams.payload.property) {
+        case 'poll':
+          theParams.payload.data = (body as unknown) as Mastodon.Poll
+          updateStatusProperty({ queryClient, ...theParams })
+          break
+      }
+    },
     onError: (err: any, params) => {
       const theParams = params as MutationVarsTimelineUpdateStatusProperty
       haptics('Error')
@@ -76,6 +91,7 @@ const TimelinePoll: React.FC<Props> = ({
                 mutation.mutate({
                   type: 'updateStatusProperty',
                   queryKey,
+                  rootQueryKey,
                   id: statusId,
                   reblog,
                   payload: {
@@ -102,6 +118,7 @@ const TimelinePoll: React.FC<Props> = ({
                 mutation.mutate({
                   type: 'updateStatusProperty',
                   queryKey,
+                  rootQueryKey,
                   id: statusId,
                   reblog,
                   payload: {