fix: reset loader when task is cancelled (#40)

This issue happens when:
- User runs a task with a loader
- Loading is in "loading" state
- Task gets halted
- Loader does not get cleaned up
- Loader stays in "loading" state

store/query.ts

@@ -93,6 +93,14 @@ export function loader<
  message: err.message,
  }),
  ]);
+ } finally {
+ const loaders = yield* select((s: any) =>
+ loaderSchema.selectByIds(s, { ids: [ctx.name, ctx.key] })
+ );
+ const ids = loaders
+ .filter((loader) => loader.status === "loading")
+ .map((loader) => loader.id);
+ yield* updateStore(loaderSchema.resetByIds(ids));
  }
  };
 }
@@ -112,44 +120,54 @@ export function loaderApi<
  },
 ) {
  return function* trackLoading(ctx: Ctx, next: Next) {
- yield* updateStore([
- loaderSchema.start({ id: ctx.name }),
- loaderSchema.start({ id: ctx.key }),
- ]);
- if (!ctx.loader) ctx.loader = {} as any;
+ try {
+ yield* updateStore([
+ loaderSchema.start({ id: ctx.name }),
+ loaderSchema.start({ id: ctx.key }),
+ ]);
+ if (!ctx.loader) ctx.loader = {} as any;
 
- yield* next();
+  yield* next();
 
- if (!ctx.response) {
- yield* updateStore(
- loaderSchema.resetByIds([ctx.name, ctx.key]),
- );
- return;
- }
+  if (!ctx.response) {
+  yield* updateStore(
+  loaderSchema.resetByIds([ctx.name, ctx.key]),
+  );
+  return;
+  }
 
- if (!ctx.loader) {
- ctx.loader = {};
- }
+ if (!ctx.loader) {
+ ctx.loader = {};
+ }
+
+ if (!ctx.response.ok) {
+ yield* updateStore([
+ loaderSchema.error({
+ id: ctx.name,
+ message: errorFn(ctx),
+ ...ctx.loader,
+ }),
+ loaderSchema.error({
+ id: ctx.key,
+ message: errorFn(ctx),
+ ...ctx.loader,
+ }),
+ ]);
+ return;
+ }
 
- if (!ctx.response.ok) {
  yield* updateStore([
- loaderSchema.error({
- id: ctx.name,
- message: errorFn(ctx),
- ...ctx.loader,
- }),
- loaderSchema.error({
- id: ctx.key,
- message: errorFn(ctx),
- ...ctx.loader,
- }),
+ loaderSchema.success({ id: ctx.name, ...ctx.loader }),
+ loaderSchema.success({ id: ctx.key, ...ctx.loader }),
  ]);
- return;
+ } finally {
+ const loaders = yield* select((s: any) =>
+ loaderSchema.selectByIds(s, { ids: [ctx.name, ctx.key] })
+ );
+ const ids = loaders
+ .filter((loader) => loader.status === "loading")
+ .map((loader) => loader.id);
+ yield* updateStore(loaderSchema.resetByIds(ids));
  }
-
- yield* updateStore([
- loaderSchema.success({ id: ctx.name, ...ctx.loader }),
- loaderSchema.success({ id: ctx.key, ...ctx.loader }),
- ]);
  };
 }