fix(FileOrganizer): fixed bug that skipped setting thumbnail size

ApryseSDK · Nov 24, 2020 · c25d758 · c25d758
1 parent 953a352
commit c25d758
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/components/FileOrganizer/FileOrganizer.stories.tsx b/src/components/FileOrganizer/FileOrganizer.stories.tsx
@@ -35,7 +35,13 @@ const Template: FC<TemplateProps> = ({
   customSizedThumbnail,
 }) => {
   // This is the index organizing function.
-  const [files, setFiles] = useState<FakeFile[]>([]);
+  const [files, setFiles] = useState<FakeFile[]>(() => {
+    const newFiles = [];
+    for (let index = 0; index < numFiles; index++) {
+      newFiles.push(createFile(index));
+    }
+    return newFiles;
+  });
   const handleOnMove = useCallback<NonNullable<FileOrganizerProps<FakeFile>['onMove']>>((fromIndex, toIndex) => {
     setFiles((prev) => {
       if (toIndex < 0 || toIndex >= prev.length) return prev;

diff --git a/src/components/FileOrganizer/FileOrganizer.tsx b/src/components/FileOrganizer/FileOrganizer.tsx
@@ -183,8 +183,8 @@ export function FileOrganizer<F extends ObjectWithId>({
 
   // Update size when getWidth ref changes (when hasFiles changes).
   useEffect(() => {
-    if (thumbnailSize) setSize(thumbnailSize);
-    if (files.length === 0) setSize(defaultSize);
+    if (thumbnailSize) return setSize(thumbnailSize);
+    if (files.length === 0) return setSize(defaultSize);
     setSize((prev) => {
       const { width, height } = getSize();
       if (prev.width === width && prev.height === height) return prev;