Add a composition for the task runs count by state api

src/compositions/index.ts

@@ -48,6 +48,7 @@ export * from './useTaskRunFavicon'
 export * from './useTaskRunResult'
 export * from './useTaskRuns'
 export * from './useTaskRunsCount'
+export * from './useTaskRunsCountByState'
 export * from './useTaskRunsHistory'
 export * from './useThemeTokens'
 export * from './useVariable'

src/compositions/useTaskRunsCountByState.ts

@@ -0,0 +1,58 @@
+import { SubscriptionOptions, useSubscriptionWithDependencies } from '@prefecthq/vue-compositions'
+import merge from 'lodash.merge'
+import { computed, MaybeRefOrGetter, toRef, toValue } from 'vue'
+import { useCan, useWorkspaceApi } from '@/compositions'
+import { TaskRunsFilter } from '@/models/Filters'
+import { UiApi } from '@/services'
+import { MaybeRef } from '@/types'
+import { Getter } from '@/types/reactivity'
+import { UseEntitySubscription } from '@/types/useEntitySubscription'
+
+export type UseTaskRunsCountByState = UseEntitySubscription<UiApi['getTaskRunsCountByState'], 'count'>
+
+export function useTaskRunsCountByState(filter: MaybeRefOrGetter<TaskRunsFilter | null | undefined>, options?: MaybeRef<SubscriptionOptions>): UseTaskRunsCountByState {
+  const api = useWorkspaceApi()
+  const can = useCan()
+
+  const getter: Getter<[TaskRunsFilter] | null> = () => {
+    if (!can.read.task_run) {
+      return null
+    }
+
+    const filterValue = toValue(filter)
+
+    if (!filterValue) {
+      return null
+    }
+
+    const base = getBaseFilter(filterValue)
+
+    // merge here is important to track changes to `filter` if it is a reactive
+    const parameter = merge({}, base, filterValue)
+
+    return [parameter]
+  }
+
+  const parameters = toRef(getter)
+  const subscription = useSubscriptionWithDependencies(api.ui.getTaskRunsCountByState, parameters, options)
+  const count = computed(() => subscription.response)
+
+  return {
+    subscription,
+    count,
+  }
+}
+
+function getBaseFilter(filter: TaskRunsFilter): TaskRunsFilter {
+  // makes sure that if subFlowRunsExists but was set to undefined (to get both all tasks)
+  // it doesn't get overridden to `false`
+  if ('subFlowRunsExist' in (filter.taskRuns ?? {})) {
+    return {}
+  }
+
+  return {
+    taskRuns: {
+      subFlowRunsExist: false,
+    },
+  }
+}