change rotate image type

Author: Michael18811380328

src/ImagePreviewerLightbox/index.js

@@ -2,7 +2,7 @@ import React from 'react';
 import PropTypes from 'prop-types';
 import classnames from 'classnames';
 import Lightbox from '@seafile/react-image-lightbox';
-import { checkSVGImage, generateCurrentBaseImageUrl, getImageThumbnailUrl, isCustomAssetUrl, isDigitalSignsUrl, isInternalImg, needUseThumbnailImage } from '../utils/url';
+import { getFileSuffix, generateCurrentBaseImageUrl, getImageThumbnailUrl, isCustomAssetUrl, isDigitalSignsUrl, isInternalImg, needUseThumbnailImage } from '../utils/url';
 import { getLocale } from '../lang';
 
 import '@seafile/react-image-lightbox/style.css';
@@ -39,8 +39,7 @@ function ImagePreviewerLightbox(props) {
     console.log(error);
   }
 
-  // svg image is vectorgraph and can't rotate, external image can't rotate
-  const canRotateImage = onRotateImage && !readOnly && !checkSVGImage(URL) && isInternalImg(URL, server);
+  const canRotateImage = onRotateImage && !readOnly && ['gif', 'heic', 'heif'].includes(getFileSuffix(URL)) && isInternalImg(URL, server);
 
   let mainSrc = URL;
   if (needUseThumbnailImage(URL)) {

src/utils/url.js

@@ -34,10 +34,13 @@ export const checkImgExists = (url) => {
   });
 };
 
+export const getFileSuffix = (filename) => {
+  if (!filename || typeof filename !== 'string') return '';
+  return filename.substring(filename.lastIndexOf('.') + 1).toLowerCase();
+};
+
 export const checkSVGImage = (url) => {
-  if (!url || typeof url !== 'string') return false;
-  const isSVGImage = url.substring(url.lastIndexOf('.')).toLowerCase() === '.svg';
-  return isSVGImage;
+  return getFileSuffix(url) === 'svg';
 };
 
 export const isAIUrl = (url) => {

tests/utils/url.test.js

@@ -0,0 +1,14 @@
+import { getFileSuffix } from '../../src/utils/url';
+
+describe('getFileSuffix', () => {
+  it('should return the suffix of the filename in lowercase', () => {
+    expect(getFileSuffix('example.TXT')).toBe('txt');
+    expect(getFileSuffix('example.jpeg')).toBe('jpeg');
+    expect(getFileSuffix('example.heic')).toBe('heic');
+  });
+  it('should return an empty string if input is not a valid string', () => {
+    expect(getFileSuffix(null)).toBe('');
+    expect(getFileSuffix(123)).toBe('');
+    expect(getFileSuffix(undefined)).toBe('');
+  });
+});