Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

revert to using fetch, add token if needed (#5720)

Browse Source 
Co-authored-by: Mary Hipp <maryhipp@Marys-MacBook-Air.local>
This commit is contained in:
Mary Hipp Rogers 2024-02-14 10:04:12 -05:00 committed by GitHub
parent 9d6e4ff1fb
commit 34cc26a4ed
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
invokeai/frontend/web/src/common/hooks/useDownloadImage.ts
View File

@ -1,22 +1,28 @@
import { useStore } from '@nanostores/react';
import { useAppToaster } from 'app/components/Toaster'; import { useAppToaster } from 'app/components/Toaster';
import { $authToken } from 'app/store/nanostores/authToken';
import { useAppDispatch } from 'app/store/storeHooks'; import { useAppDispatch } from 'app/store/storeHooks';
import { imageDownloaded } from 'features/gallery/store/actions'; import { imageDownloaded } from 'features/gallery/store/actions';
import { useCallback } from 'react'; import { useCallback } from 'react';
import { useTranslation } from 'react-i18next'; import { useTranslation } from 'react-i18next';
import { useImageUrlToBlob } from './useImageUrlToBlob';
export const useDownloadImage = () => { export const useDownloadImage = () => {
const toaster = useAppToaster(); const toaster = useAppToaster();
const { t } = useTranslation(); const { t } = useTranslation();
const imageUrlToBlob = useImageUrlToBlob();
const dispatch = useAppDispatch(); const dispatch = useAppDispatch();
const authToken = useStore($authToken);
const downloadImage = useCallback( const downloadImage = useCallback(
async (image_url: string, image_name: string) => { async (image_url: string, image_name: string) => {
try { try {
const blob = await imageUrlToBlob(image_url); const requestOpts = authToken
? {
headers: {
Authorization: `Bearer ${authToken}`,
},
}
: {};
const blob = await fetch(image_url, requestOpts).then((resp) => resp.blob());
if (!blob) { if (!blob) {
throw new Error('Unable to create Blob'); throw new Error('Unable to create Blob');
} }
@ -40,7 +46,7 @@ export const useDownloadImage = () => {
}); });
} }
}, },
[t, toaster, imageUrlToBlob, dispatch] [t, toaster, dispatch, authToken]
); );
return { downloadImage }; return { downloadImage };