takeScreenshot now returns string

README.md

@@ -48,7 +48,7 @@ const Example = () => {
 #### `UseScreenshotReturnType`
 
 - `isLoading` - Indicates if the screenshot is loading this is useful for some webpages that are big and have lots of content to render.
-- `takeScreenshot: (types: ImgType, options: OptionsType)` - A function to take screenshot.
+- `takeScreenshot: (types: ImgType, options: OptionsType): string | undefined` - A function to take screenshot.
 - `image` - Contains the dataUrl for the image.
 - `clear()` - Clear the image from the memory.
 

package.json

@@ -1,6 +1,6 @@
 {
   "name": "use-screenshot-hook",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "React hook to take screenshot for React components.",
   "author": "fayeed",
   "license": "MIT",

src/index.tsx

@@ -10,25 +10,25 @@ export const useScreenshot = (options?: UseScreenshotProps) => {
   const takeScreenshot = useCallback(
     async (type?: ImgType, options?: OptionsType) => {
       setLoading(true);
+      let tempImage: string | undefined;
 
       try {
-        let image: string;
-
-        // using body tag causes a wierd bug
         const body = document.getElementById("root")!;
 
         if (type === "jpg") {
-          image = await toJpeg(ref?.current || body, options);
+          tempImage = await toJpeg(ref?.current || body, options);
         } else {
-          image = await toPng(ref?.current || body, options);
+          tempImage = await toPng(ref?.current || body, options);
         }
 
-        setImage(image);
+        setImage(tempImage);
       } catch (e) {
         console.error(e);
-      }
+      } finally {
+        setLoading(false);
 
-      setLoading(false);
+        return tempImage;
+      }
     },
     []
   );