图片上传增加token

pages/certification/index.page.tsx

@@ -12,9 +12,9 @@ import {
 } from "antd";
 import type { UploadChangeParam } from "antd/es/upload";
 import type { RcFile, UploadFile, UploadProps } from "antd/es/upload/interface";
-import { useContext, useState } from "react";
+import { useContext, useEffect, useState } from "react";
 import Layout from "~/components/layout";
-import api, { } from "./api";
+import api from "./api";
 import styles from "./index.module.scss";
 import gApi from "~/api";
 import Router from "next/router";
@@ -55,6 +55,11 @@ export default function Certification() {
     Array<EnterpriseOption>
   >([]);
   const [form] = Form.useForm();
+  const [token, setToken] = useState("");
+
+  useEffect(() => {
+    setToken(window.localStorage.getItem("token") || "");
+  }, []);
 
   //上传change事件
   const handleChange: UploadProps["onChange"] = (
@@ -91,9 +96,9 @@ export default function Certification() {
             });
           }
           setTimeout(() => {
-            if (Router.query.type == 'back'){
+            if (Router.query.type == "back") {
               Router.back();
-            }else{
+            } else {
               Router.push("/");
             }
           }, 1000);
@@ -213,6 +218,7 @@ export default function Certification() {
                 beforeUpload={beforeUpload}
                 onChange={handleChange}
                 maxCount={1}
+                headers={{ token: token }}
               >
                 {imageUrl ? (
                   <Image

pages/forum/components/publishMessage/index.tsx

@@ -2,12 +2,13 @@ import { PlusOutlined } from "@ant-design/icons";
 import { Form, Input, Modal, Upload, Image, Button, Row, Col } from "antd";
 import type { RcFile, UploadProps } from "antd/es/upload";
 import type { UploadFile } from "antd/es/upload/interface";
-import { useContext, useState } from "react";
+import { useContext, useEffect, useState } from "react";
 import gApi from "~/api";
 import NImage from "next/image";
 import api from "./api";
 import { useGeolocation } from "~/lib/hooks";
 import { UserContext } from "~/lib/userProvider";
+import { useToken } from "antd/es/theme/internal";
 
 type Props = {
   open: boolean;
@@ -33,6 +34,11 @@ export default function PublishMessage(props: Props) {
   const [form] = Form.useForm();
   const { userInfo, setNeedLogin } = useContext(UserContext);
   const position = useGeolocation();
+  const [token, setToken] = useState('');
+
+  useEffect(() => {
+    setToken(window.localStorage.getItem('token') || '');
+  }, [])
 
   //预览关闭
   const handlePreviewCancel = () => setPreviewOpen(false);
@@ -155,6 +161,7 @@ export default function PublishMessage(props: Props) {
             onPreview={handlePreview}
             onChange={handleChange}
             maxCount={1}
+            headers={{ token: token }}
           >
             {fileList.length >= 8 ? null : (
               <div>