From d830091613973774d74735be5632fdccc0696d36 Mon Sep 17 00:00:00 2001
From: Andrew Plummer <plummer.andrew@gmail.com>
Date: Fri, 28 Jun 2024 01:19:22 +0900
Subject: [PATCH] added helper components to access private files

---
 services/web/src/components/PrivateFile.js  | 88 +++++++++++++++++++++
 services/web/src/components/PrivateImage.js | 65 +++++++++++++++
 2 files changed, 153 insertions(+)
 create mode 100644 services/web/src/components/PrivateFile.js
 create mode 100644 services/web/src/components/PrivateImage.js

diff --git a/services/web/src/components/PrivateFile.js b/services/web/src/components/PrivateFile.js
new file mode 100644
index 00000000..695c6193
--- /dev/null
+++ b/services/web/src/components/PrivateFile.js
@@ -0,0 +1,88 @@
+// Component for use with isPriivate uploads.
+
+import React from 'react';
+import PropTypes from 'prop-types';
+import { omit } from 'lodash';
+import { Button } from 'semantic';
+
+import { request } from 'utils/api';
+
+import { ExternalLink } from './Link';
+
+const ICONS = {
+  image: 'file-image',
+  video: 'file-video',
+  audio: 'file-audio',
+  text: 'file-lines',
+  pdf: 'file-pdf',
+  csv: 'file-excel',
+  document: 'file-excel',
+  application: 'file-pdf',
+  zip: 'file-archive',
+  'image/pdf': 'file-image',
+};
+
+export default class DownloadButton extends React.Component {
+  constructor(props) {
+    super(props);
+    this.state = {
+      url: null,
+    };
+  }
+
+  componentDidMount() {
+    this.load();
+  }
+
+  getIcon() {
+    const { upload } = this.props;
+    const [type] = upload.mimeType.split('/');
+    return ICONS[type];
+  }
+
+  load = async () => {
+    try {
+      this.setState({
+        error: null,
+        loading: true,
+      });
+      const { upload } = this.props;
+
+      const id = upload?.id || upload;
+
+      const { data: url } = await request({
+        method: 'GET',
+        path: `/1/uploads/${id}/url`,
+      });
+
+      this.setState({
+        url,
+        loading: false,
+      });
+    } catch (error) {
+      this.setState({
+        error,
+        loading: false,
+      });
+    }
+  };
+
+  render() {
+    const { upload } = this.props;
+    if (!upload) {
+      return null;
+    }
+    const { url } = this.state;
+    const props = omit(this.props, Object.keys(DownloadButton.propTypes));
+
+    return (
+      <ExternalLink href={url}>
+        <Button basic icon={this.getIcon()} {...props} />
+      </ExternalLink>
+    );
+  }
+}
+
+DownloadButton.propTypes = {
+  upload: PropTypes.object,
+};
diff --git a/services/web/src/components/PrivateImage.js b/services/web/src/components/PrivateImage.js
new file mode 100644
index 00000000..7a6c115f
--- /dev/null
+++ b/services/web/src/components/PrivateImage.js
@@ -0,0 +1,65 @@
+// Component for use with private uploads.
+
+import React from 'react';
+import { omit } from 'lodash';
+import PropTypes from 'prop-types';
+import { Image } from 'semantic';
+
+import { request } from 'utils/api';
+
+import { ExternalLink } from './Link';
+
+export default class PrivateImage extends React.Component {
+  constructor(props) {
+    super(props);
+    this.state = {
+      src: null,
+    };
+  }
+
+  componentDidMount() {
+    this.load();
+  }
+
+  load = async () => {
+    try {
+      this.setState({
+        error: null,
+        loading: true,
+      });
+      const { upload } = this.props;
+
+      const id = upload?.id || upload;
+
+      const { data: url } = await request({
+        method: 'GET',
+        path: `/1/uploads/${id}/url`,
+      });
+
+      this.setState({
+        src: url,
+        loading: false,
+      });
+    } catch (error) {
+      this.setState({
+        error,
+        loading: false,
+      });
+    }
+  };
+
+  render() {
+    const { src } = this.state;
+    const props = omit(this.props, PrivateImage.propTypes);
+    return (
+      <ExternalLink href={src}>
+        <Image {...props} src={src} />
+      </ExternalLink>
+    );
+  }
+}
+
+PrivateImage.propTypes = {
+  upload: PropTypes.oneOfType([PropTypes.string, PropTypes.object]).isRequired,
+  filename: PropTypes.string,
+};