chrome-extension/entrypoints/popup/pages/DesignPic.tsx

import React, { useState, useEffect } from 'react'
import { Button, Table, Input, Message, Loading } from '@alifd/next';
import { CustomMessage } from '@/components/CustomMessage'

export default function DesignPic() {
  const [uploadUrl, setUploadUrl] = useState<string>("");
  // 添加状态管理表格数据
  const [tableData, setTableData] = useState<any[]>([]);
  const [loading, setLoading] = useState(false);
  const [isPage, setIsPage] = useState(true);
  const [btnLoading, setBtnLoading] = useState(false);

  useEffect(() => {
    is_temu_design()
  }, [])

  const is_temu_design = async () => {
    try {
      // 获取当前活动标签页
      const [tab] = await browser.tabs.query({ active: true, currentWindow: true });

      // 获取页面域名
      if (tab.url) {
        const url = tab.url
        console.log('当前页面域名:', url);
        if (url.includes('main/customize-goods')) {
          // temu定制页面
          setIsPage(false)
        }
      }
    } catch (error) {
      console.error('获取当前页面域名时出错:', error);
    }
  }

  const pic_render = (value: any, index: any, record: any) => {
    return (
      <>
        {
          record.file ? <>
            <div className='flex items-center justify-center'><img className='size-20' src={record.file} /></div></> : <>
            <span>暂无图片</span>
          </>
        }
      </>
    );
  }

  const text_render = (value: any, index: any, record: any) => {
    return (
      <p>{record.text}</p>
    );
  }


  // 修改图片名称渲染函数
  const name_render = (value: any, index: number, record: any) => {
    return <>
      <div>{record.order_no}</div>
    </>
  };

  // 修改图片标签渲染函数
  const label_render = (value: any, index: number, record: any) => {
    return <>
      <div>{record.region}</div>
    </>
  };

  // 完善删除功能
  const handleDelete = (index: number, record: any) => {
    // 方案1: 直接删除
    const newData = [...tableData];
    newData.splice(index, 1);
    setTableData(newData);

    // 方案2: 带确认的删除 (可选)
    // if (window.confirm(`确定要删除 "${record.title.name}" 吗?`)) {
    //   const newData = [...tableData];
    //   newData.splice(index, 1);
    //   setTableData(newData);
    //   Message.success('删除成功');
    // }
  };

  // 删除操作渲染函数
  const option_render = (value: any, index: number, record: any) => {
    return (
      <a
        className='text-red-600 cursor-pointer'
        onClick={() => handleDelete(index, record)}
      >
        删除
      </a>
    );
  };

  // 添加行索引以确保正确删除
  const dataSourceWithIndex = tableData.map((item, index) => ({
    ...item,
    index
  }));

  const getPicData = async () => {
    const [tab] = await browser.tabs.query({ active: true, currentWindow: true });
    // 获取页面域名
    if (tab.url) {
      const url = tab.url
      if (!url.includes('main/customize-goods')) {
        // temu定制页面
        CustomMessage.error('当前页面不是定制页面')
        return
      }
    }
    try {
      setLoading(true)
      // 获取上传路径
      fetch('http://merchant.landwu.com/api/photo/getUploadUrl', {
        method: 'POST',
        headers: {
          'Content-Type': 'application/json',
        },
        body: JSON.stringify({
          api_token: localStorage.getItem('access_token'),
        }),
      }).then(resjson => {
        return resjson.json();
      }).then(res => {
        if (res.data) {
          const { data = {} } = res;
          const { url = null } = data;
          setUploadUrl(url)
        }
      }).catch(e => {
        setLoading(false)
      })

      setTableData([])
      setBtnLoading(true)
      // 获取当前活动标签页
      const [tab] = await browser.tabs.query({ active: true, currentWindow: true });

      if (tab.id) {
        // 发送消息到content script请求获取tbody数据
        const response = await browser.tabs.sendMessage(tab.id, { action: "getTbodyData" });
        const table_datas = []
        // 在控制台输出结果
        if (response.tables && response.tables.length > 0) {
          // console.log("找到的表格数据(包含背景图片):");
          // 显式声明类型
          let tables_arr: {
            good_info: any;
            sku_info: any;
            order_no: any;
            design_sku: any;
            design_info: { region: any; text: any; picture: any }[];
          }[] = [];
          let table_boj: {
            good_info: any;
            sku_info: any;
            order_no: any;
            design_sku: any;
            design_info: { region: any; text: any; picture: any }[];
          } = {
            good_info: null,
            sku_info: null,
            order_no: null,
            design_sku: null,
            design_info: []
          };

          response.tables.forEach((table_node: any, index: number) => {
            console.log(`表格 ${index + 1}:`, table_node);
            // 处理成接口参数

            // 处理成数组对象
            if (table_node.length > 4) {
              if (table_boj.design_info.length) {
                // 一个商品的定制信息
                tables_arr.push(table_boj)
                // 重置
                table_boj = {
                  good_info: null,
                  sku_info: null,
                  order_no: null,
                  design_sku: null,
                  design_info: []
                }
              }
              table_boj = {
                good_info: table_node[1],
                sku_info: table_node[2],
                order_no: table_node[3],
                design_sku: table_node[4],
                design_info: [{
                  region: table_node[9],
                  text: table_node[10],
                  picture: table_node[11],//11
                }]
              }
            } else {
              table_boj.design_info.push({
                region: table_node[0],
                text: table_node[1],
                picture: table_node[2],//2
              })
            }
            // 循环结束
            if (index == response.tables.length - 1) {
              if (table_boj.design_info.length) {
                // 一个商品的定制信息
                tables_arr.push(table_boj)
              }
            }
            // good_info  sku_info  design_info: region text picture 
          });

          console.log(tables_arr, "tables_arrddd")

          // 上传图片和文字的参数处理
          let pic_text_arr: any[] = [], img_values: any = {}
          tables_arr.forEach(item => {
            let order_no = "", data_params: any[] = []
            if (item.order_no && item.order_no != "-") {
              order_no = item.order_no.substring(0, 15); // 取前15位
            }
            if (item.design_info.length) {
              item.design_info.forEach(items => {
                const param: any = {}
                if (order_no || item.design_sku) {
                  let bg_pic = ""
                  if (items.picture && items.picture != '-') {
                    let bg_pic1 = (items.picture.backgroundImages as { url: string }[] | undefined)?.[0]?.url ?? '';
                    bg_pic = bg_pic1.split('&imageMogr2/thumbnail/60x60')[0]
                    if (!img_values[bg_pic]) {
                      img_values[bg_pic] = {
                        src: bg_pic,
                        width: '',
                        height: ''
                      }
                    }
                    if (bg_pic) {
                      param['width'] = 400;
                      param['height'] = 400;
                    }
                  } else {
                    bg_pic = ""
                  }
                  let des_text = ""
                  if (items.text && items.text != '-') {
                    const des_text1 = items.text.split(";}")[1]
                    des_text = des_text1.split("查看")[0]
                  } else {
                    des_text = ""
                  }
                  param['region'] = items.region
                  param['text'] = des_text
                  param['design_pic'] = bg_pic
                  data_params.push(param)
                }
              })
            }
            if (order_no || item.design_sku) {
              pic_text_arr.push({
                order_no,
                design_sku: item.design_sku,
                data_params
              })
            }
          })
          // console.log(img_values,'-img_valuesimg_values')
          let img_values1: any = {}, cIndex = 0
          Object.values(img_values).forEach((item, n) => {
            if (!img_values1[n]) {
              img_values1[n] = item
            }
          })

          const queueImg = () => {
            if (img_values1[cIndex]) {
              let img = new Image()
              img.onload = function () {
                img_values1[cIndex].width = img.width
                img_values1[cIndex].height = img.height
                img_values[img_values1[cIndex]?.src]['width'] = img.width
                img_values[img_values1[cIndex]?.src]['height'] = img.height
                cIndex++
                queueImg()
              }
              img.onerror = function () {
                cIndex++
                queueImg()
              }
              img.src = img_values1[cIndex]?.src

            } else {
              getImgWh()
            }
          }
          queueImg()
          function getImgWh() {
            setBtnLoading(false)
            setLoading(false)
            // 筛选备货单号存在的值
            let pic_text_arr1 = pic_text_arr.filter(item => item.order_no && item.order_no != "-")
            // 为存在图片的对象添加宽高
            pic_text_arr1.map(item => {
              item.data_params.map((items: any) => {
                if (items.design_pic) {
                  if (img_values[items.design_pic]) {
                    items['width'] = img_values[items.design_pic].width
                    items['height'] = img_values[items.design_pic].height
                  }
                }
              })
            })
            // 数据显示
            let table_data: any[] = []
            pic_text_arr1.forEach(item => {
              if (item.data_params.length) {
                item.data_params.forEach((items: any) => {
                  table_data.push({
                    order_no: item.order_no,
                    design_sku: item.design_sku,
                    file: items.design_pic,
                    region: items.region,
                    text: items.text,
                    width: items.width || "",
                    height: items.height || "",
                  })
                })
              }
            })
            console.log(pic_text_arr1, "pic_text_arr1aaa", img_values1)
            setTableData(table_data)
          }


        } else {
          console.log("页面上未找到表格数据");
          CustomMessage.error("页面上未找到表格数据");
        }
      }
    } catch (error) {
      setBtnLoading(false)
      setLoading(false)
      console.error("获取表格数据失败:", error);
      CustomMessage.error("获取表格数据失败");
    }
  };

  const uploadPic = () => {
    const token = localStorage.getItem("access_token");
    console.log(tableData, "tableData")
    let table_obj: { [key: string]: any } = {};
    tableData.forEach(item => {
      if (!table_obj[item.design_sku]) {
        table_obj[item.design_sku] = ""
      }
      if (table_obj[item.design_sku]) {
        const data_params = table_obj[item.design_sku].data_params
        data_params.push({
          design_pic: item.file,
          region: item.region,
          text: item.text,
          width: item.width,
          height: item.height,
        })
        table_obj[item.design_sku] = {
          ...table_obj[item.design_sku],
          data_params,
        }
      } else {
        table_obj[item.design_sku] = {
          order_no: item.order_no,
          design_sku: item.design_sku,
          data_params: [{
            design_pic: item.file,
            region: item.region,
            text: item.text,
            width: item.width,
            height: item.height,
          }]
        }
      }
    })
    console.log(table_obj, "table_obj", Object.values(table_obj))
    const table_params = Object.values(table_obj)
    let fetch_url = "", arr_index = 0
    if (!uploadUrl) {
      CustomMessage.warning("上传地址丢失,请再次点击获取定制数据!")
      return
    }
    if (uploadUrl.includes('http')) {
      fetch_url = `${uploadUrl}photo/temuUpload`
    } else {
      fetch_url = `http:${uploadUrl}photo/temuUpload`
    }
    const FetchFun = (params: any) => {
      // console.log(params, "params")
      params['api_token'] = token
      fetch(fetch_url, {
        method: 'POST',
        headers: {
          'Content-Type': 'application/json',
        },
        body: JSON.stringify(params),
      }).then(resjson => {
        return resjson.json();
      }).then(res => {
        if (res.data.error && res.data.error.length) {
          // 未上传成功的 图片
          console.log(res.data.error, "arr_index", arr_index)
          res.data.error.forEach((item: any) => {
            CustomMessage.error(item.design_pic)
          })
        }
        arr_index++
        if (arr_index < table_params.length) {
          FetchFun(table_params[arr_index])
        } else {
          setLoading(false)
          setTableData([])
          CustomMessage.success("图片上传结束")
        }
      }).catch(err => {
        console.log(err, "err")
        arr_index++
        if (arr_index < table_params.length) {
          FetchFun(table_params[arr_index])
        } else {
          setLoading(false)
          setTableData([])
          CustomMessage.success("图片上传结束")
        }
      })
    }
    setLoading(true)
    FetchFun(table_params[arr_index])
  }

  return (
    <Loading visible={loading} fullScreen>
      <div className='w-3xl h-fit'>
        <Button
          type="primary"
          disabled={isPage || btnLoading}
          style={{ marginBottom: '10px', marginRight: '10px' }}
          onClick={() => {
            getPicData()
          }}>获取定制数据</Button>
        <Button
          disabled={(!tableData.length) || isPage || btnLoading}
          type="primary"
          style={{ marginBottom: '10px', marginRight: '10px' }}
          onClick={() => {
            uploadPic()
          }}>上传定制信息</Button>
        <span className='text-red-500'>ps:需在Temu定制商品内容页面获取数据,且上传图片过程中请勿关闭插件</span>
        <Table dataSource={dataSourceWithIndex}>
          <Table.Column title="图片" cell={pic_render} align="center" />
          <Table.Column title="备货单" cell={name_render} align="center" />
          <Table.Column title="定制区域" cell={label_render} align="center" />
          <Table.Column title="文字" cell={text_render} align="center" />
          <Table.Column title="操作" cell={option_render} align="center" />
        </Table>
        <p className='text-red-500'>ps:若获取不到数据,可能是插件未连接成功,请刷新页面</p>
      </div>
    </Loading>
  )
}