UNPKG

react-mention-input

Version:

A React component for input with @mention functionality.

188 lines (187 loc) 9.25 kB
var __assign = (this && this.__assign) || function () { __assign = Object.assign || function(t) { for (var s, i = 1, n = arguments.length; i < n; i++) { s = arguments[i]; for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p)) t[p] = s[p]; } return t; }; return __assign.apply(this, arguments); }; var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) { function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); } return new (P || (P = Promise))(function (resolve, reject) { function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } } function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } } function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); } step((generator = generator.apply(thisArg, _arguments || [])).next()); }); }; var __generator = (this && this.__generator) || function (thisArg, body) { var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype); return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g; function verb(n) { return function (v) { return step([n, v]); }; } function step(op) { if (f) throw new TypeError("Generator is already executing."); while (g && (g = 0, op[0] && (_ = 0)), _) try { if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t; if (y = 0, t) op = [op[0] & 2, t.value]; switch (op[0]) { case 0: case 1: t = op; break; case 4: _.label++; return { value: op[1], done: false }; case 5: _.label++; y = op[1]; op = [0]; continue; case 7: op = _.ops.pop(); _.trys.pop(); continue; default: if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; } if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; } if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; } if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; } if (t[2]) _.ops.pop(); _.trys.pop(); continue; } op = body.call(thisArg, _); } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; } if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true }; } }; import { useState, useEffect, useRef } from 'react'; // Module-level cache to persist blob URLs across component re-renders var blobUrlCache = new Map(); var fetchingUrls = new Set(); /** * Custom hook to handle protected image URLs that require authentication tokens in headers. * For protected URLs, it fetches the image with auth headers and converts it to a blob URL. * For non-protected URLs, it returns the URL as-is. * Uses a module-level cache to prevent blinking on re-renders. */ export var useProtectedImage = function (_a) { var url = _a.url, isProtected = _a.isProtected, getAuthHeaders = _a.getAuthHeaders; var _b = useState(function () { // Initialize from cache if available return url ? blobUrlCache.get(url) || null : null; }), blobUrl = _b[0], setBlobUrl = _b[1]; var previousUrlRef = useRef(null); var isProtectedRef = useRef(isProtected); var getAuthHeadersRef = useRef(getAuthHeaders); var mountedRef = useRef(true); // Update refs when props change (but don't trigger re-fetch) useEffect(function () { isProtectedRef.current = isProtected; getAuthHeadersRef.current = getAuthHeaders; }, [isProtected, getAuthHeaders]); useEffect(function () { mountedRef.current = true; return function () { mountedRef.current = false; }; }, []); useEffect(function () { // Always check cache first and restore if needed (synchronous) if (url && blobUrlCache.has(url) && !blobUrl) { var cachedBlobUrl = blobUrlCache.get(url); setBlobUrl(cachedBlobUrl); } // If URL hasn't changed, keep using cached blob URL if (url === previousUrlRef.current) { return; } var oldUrl = previousUrlRef.current; previousUrlRef.current = url; if (!url) { setBlobUrl(null); return; } // Check if URL is protected var shouldUseAuth = typeof isProtectedRef.current === 'boolean' ? isProtectedRef.current : isProtectedRef.current ? isProtectedRef.current(url) : false; // If not protected, use original URL if (!shouldUseAuth || !getAuthHeadersRef.current) { // Clear any cached blob URL for this URL if (blobUrlCache.has(url)) { var cachedBlobUrl = blobUrlCache.get(url); URL.revokeObjectURL(cachedBlobUrl); blobUrlCache.delete(url); } setBlobUrl(null); return; } // Check cache first - if we have a cached blob URL, use it immediately if (blobUrlCache.has(url)) { var cachedBlobUrl = blobUrlCache.get(url); setBlobUrl(cachedBlobUrl); return; } // If already fetching this URL, don't fetch again if (fetchingUrls.has(url)) { return; } // Fetch protected image with auth headers fetchingUrls.add(url); var fetchProtectedImage = function () { return __awaiter(void 0, void 0, void 0, function () { var headers, response, blob, newBlobUrl, error_1; return __generator(this, function (_a) { switch (_a.label) { case 0: _a.trys.push([0, 4, 5, 6]); return [4 /*yield*/, Promise.resolve(getAuthHeadersRef.current())]; case 1: headers = _a.sent(); return [4 /*yield*/, fetch(url, { method: 'GET', headers: __assign({}, headers), })]; case 2: response = _a.sent(); if (!response.ok) { throw new Error("Failed to fetch protected image: ".concat(response.statusText)); } return [4 /*yield*/, response.blob()]; case 3: blob = _a.sent(); newBlobUrl = URL.createObjectURL(blob); // Only update if URL hasn't changed during fetch and component is still mounted if (previousUrlRef.current === url && mountedRef.current) { // Cache the blob URL blobUrlCache.set(url, newBlobUrl); setBlobUrl(newBlobUrl); } else { // URL changed during fetch or component unmounted, revoke this blob URL URL.revokeObjectURL(newBlobUrl); } return [3 /*break*/, 6]; case 4: error_1 = _a.sent(); console.error('Error fetching protected image:', error_1); // On error, only clear if URL hasn't changed and component is mounted if (previousUrlRef.current === url && mountedRef.current) { setBlobUrl(null); } return [3 /*break*/, 6]; case 5: fetchingUrls.delete(url); return [7 /*endfinally*/]; case 6: return [2 /*return*/]; } }); }); }; fetchProtectedImage(); }, [url]); // Only depend on url // Cleanup: Don't revoke blob URLs on unmount - keep them in cache for reuse // They will be cleaned up when the URL changes or when the cache is cleared // Return blob URL if available, otherwise return original URL (or null) if (!url) { return null; } var isUrlProtected = typeof isProtectedRef.current === 'boolean' ? isProtectedRef.current : isProtectedRef.current ? isProtectedRef.current(url) : false; // For protected URLs, return cached blob URL or current blobUrl state // For non-protected URLs, return the original URL if (isUrlProtected) { return blobUrl || blobUrlCache.get(url) || null; } return url; };