lnreader/src/hooks/useChapter.ts

import { useCallback, useEffect, useState } from 'react';
import sanitize from 'sanitize-html';
import { defaultTo } from 'lodash';
import { getChapterFromDb } from '../database/queries/DownloadQueries';

import { sourceManager } from '../sources/sourceManager';

import { DownloadedChapter } from '../database/types';
import { SourceChapter } from '../sources/types';

type Chapter = SourceChapter | DownloadedChapter;

const useChapter = (
  sourceId: number,
  chapterId: number,
  chapterUrl: string,
  novelUrl: string,
) => {
  const [isLoading, setIsLoading] = useState<boolean>(true);
  const [chapter, setChapter] = useState<Chapter>();
  const [error, setError] = useState<string>();

  const getChapter = useCallback(async () => {
    try {
      let res: Chapter = await getChapterFromDb(chapterId);

      if (!res) {
        res = await sourceManager(sourceId).parseChapter(novelUrl, chapterUrl);
      }

      let chapterText = sanitize(
        defaultTo(
          res.chapterText,
          "Chapter is empty. Report if it's available in WebView.",
        ),
      );

      setChapter({ ...res, chapterText });
    } catch (err: unknown) {
      if (typeof err === 'string') {
        setError(err);
      } else if (err instanceof Error) {
        setError(err.message);
      }
    } finally {
      setIsLoading(false);
    }
  }, [sourceId, chapterId, chapterUrl, novelUrl]);

  useEffect(() => {
    getChapter();
  }, [getChapter]);

  return { isLoading, setIsLoading, chapter, error };
};

export default useChapter;