WorkloadWeb/src/pages/PurchaseAdmin/PurchaseList/WorkingHours/index.js

import React, { useState, useEffect, useRef } from 'react';
import { Modal, Button, Calendar, Popover, Spin, Row, Col } from 'antd';
import AddModal from './AddModal';
import WorkList from './WorkList';
import { connect } from 'dva';
import moment from 'moment';

function List(props) {
  const { typeList, dispatch, loading, dataList, projectList, currentUser, allType } = props;
  const [visible, setVisible] = useState(false);
  const [current, setCurrent] = useState({
    date: moment(),
    list: [],
  });

  const onAuth = (item, workload) => {
    console.log(workload);
    Modal.confirm({
      title: '提示',
      content: '是否上报审批?',
      okText: '确认',
      cancelText: '取消',
      onOk() {
        console.log(item);
        let params = [
          {
            type_id: Number(item.type_id),
            // 父级id
            parent_type_id: allType[item.type_id].parent_id,
            data: [
              {
                id: item.id,
                project_id: item.project_id,
                workload: workload,
                day: item.time,
              },
            ],
          },
        ];
        dispatch({
          type: 'workload/addAuthWorkHours',
          payload: params,
          callback: list => {
            let time = current.date.format('YYYY-MM-DD');
            setCurrent({
              ...current,
              list: list.filter(item => item.time == time),
            });
          },
        });
      },
    });
  };

  const MultiAuth = () => {
    Modal.confirm({
      title: '提示',
      content: '是否上报全部审批?',
      okText: '确认',
      cancelText: '取消',
      onOk() {
        let params = [];
        for (let i = 0; i < dataList.length; i++) {
          const element = dataList[i];
          if (element.audit_state != 0) continue;
          var arr = params.find(arr => arr.type_id == element.type_id);
          if (arr) {
            console.log(arr);
            arr.parent_type_id = allType[element.type_id].parent_id;
            arr.data.push({
              id: element.id,
              project_id: element.project_id,
              workload: element.workload,
              day: element.time,
            });
          } else {
            let data = [];
            data.push({
              id: element.id,
              project_id: element.project_id,
              workload: element.workload,
              day: element.time,
            });
            params.push({
              type_id: Number(element.type_id),
              // 父级id
              parent_type_id: allType[element.type_id].parent_id,
              data: data,
            });
          }
        }
        console.log(params);
        dispatch({
          type: 'workload/addAuthWorkHours',
          payload: params,
        });
      },
    });
  };

  const currentDayAuth = () => {
    var list = current.list;
    Modal.confirm({
      title: '提示',
      content: '是否上报本日审批?',
      okText: '确认',
      cancelText: '取消',
      onOk() {
        let params = [];
        for (let i = 0; i < list.length; i++) {
          const element = list[i];
          if (element.audit_state != 0) continue;
          var arr = params.find(arr => arr.type_id == element.type_id);
          if (arr) {
            console.log(arr);
            arr.parent_type_id = allType[element.type_id].parent_id;
            arr.data.push({
              id: element.id,
              project_id: element.project_id,
              workload: element.workload,
              day: element.time,
            });
          } else {
            let data = [];
            data.push({
              id: element.id,
              project_id: element.project_id,
              workload: element.workload,
              day: element.time,
            });
            params.push({
              type_id: Number(element.type_id),
              // 父级id
              parent_type_id: allType[element.type_id].parent_id,
              data: data,
            });
          }
        }
        console.log(params);
        dispatch({
          type: 'workload/addAuthWorkHours',
          payload: params,
        });
      },
    });
  };

  const onSave = (item, workload) => {
    let params = [
      {
        type_id: Number(item.type_id),
        comment: '',
        // 父级id
        parent_type_id: allType[item.type_id].parent_id,
        data: [
          {
            id: item.id,
            project_id: Number(item.project_id),
            workload: workload,
            day: item.time,
          },
        ],
      },
    ];

    dispatch({
      type: 'workload/addWorkHours',
      payload: params,
      callback: list => {
        let time = current.date.format('YYYY-MM-DD');
        setCurrent({
          ...current,
          list: list.filter(item => item.time == time),
        });
      },
    });
  };

  const onAddWork = data => {
    dispatch({
      type: 'workload/addWorkHours',
      payload: data,
      callback: list => {
        setVisible(false);
        let time = current.date.format('YYYY-MM-DD');
        setCurrent({
          ...current,
          list: list.filter(item => item.time == time),
        });
      },
    });
  };

  const dateCellRender = value => {
    let current = value.format('YYYY-MM-DD');
    let list = dataList.filter(item => item.time == current);
    // let total = list.reduce((total, item) => total + item.workload, 0);
    let waitTotal = 0,
      successTotal = 0;
    list.forEach(item => {
      if (item.audit_state == 2) {
        successTotal += item.workload;
      } else {
        waitTotal += item.workload;
      }
    });
    if (list.length == 0) return;
    let content = (
      <div>
        {list.map(item => (
          <div key={item.type_id}>
            {getName(item)}-{item.TypeInfo?.name}:{item.workload}小时
          </div>
        ))}
      </div>
    );
    return (
      <Popover content={content}>
        {successTotal !== 0 && <div>已审批: {successTotal}</div>}
        {waitTotal !== 0 && <div style={{ color: '#e43d33' }}>待处理: {waitTotal}</div>}
      </Popover>
    );
  };

  const onChangeDate = value => {
    let time = value.format('YYYY-MM-DD');
    if (current.date.format('YYYY-MM') != value.format('YYYY-MM')) {
      const s_date = value.format('YYYY-MM') + '-01';
      const e_date = moment(s_date)
        .add('month', 1)
        .add('days', -1)
        .format('YYYY-MM-DD');
      dispatch({
        type: 'workload/queryWorkHours',
        payload: {
          pageSize: 9999,
          user_id: currentUser.ID,
          s_time: s_date + ' 00:00:00',
          e_time: e_date + ' 23:59:59',
        },
        callback: list => {
          setCurrent({
            date: value,
            list: list.filter(item => item.time == time),
          });
        },
      });
    } else {
      setCurrent({
        date: value,
        list: dataList.filter(item => item.time == time),
      });
    }
  };

  const getName = item => {
    let name;
    if (item.project_id == '0') {
      let pid = allType[item.type_id]?.parent_id;
      if (!pid) return '';
      name = allType[pid].name;
    } else {
      name = item.Project.Name;
    }
    return name;
  };

  useEffect(() => {
    if (!currentUser.ID) return;
    // 查询分类以及工时
    dispatch({
      type: 'workload/queryWorkType',
      callback: () => {
        const s_date = current.date.format('YYYY-MM') + '-01';
        const e_date = moment(s_date)
          .add('month', 1)
          .add('days', -1)
          .format('YYYY-MM-DD');
        dispatch({
          type: 'workload/queryWorkHours',
          payload: {
            pageSize: 9999,
            user_id: currentUser.ID,
            s_time: s_date + ' 00:00:00',
            e_time: e_date + ' 23:59:59',
          },
          callback: list => {
            let time = current.date.format('YYYY-MM-DD');
            setCurrent({
              ...current,
              list: list.filter(item => item.time == time),
            });
          },
        });
      },
    });
    // 查询项目列表
    // dispatch({
    //   type: 'workload/queryProject',
    // });
    return () => {
      // 清空查询数据
      dispatch({
        type: 'workload/save',
        payload: {
          filter: {},
          list: [],
        },
      });
    };
  }, [currentUser.ID]);

  // useEffect(() => {
  //   onChangeDate(current.date);
  // }, [dataList]);

  return (
    <div>
      <Spin spinning={loading}>
        <Row gutter={8}>
          <Col span={10}>
            <Calendar
              value={current.date}
              dateCellRender={dateCellRender}
              onChange={onChangeDate}
            />
          </Col>
          <Col span={14}>
            <div>
              <Button type="primary" style={{ marginBottom: 20 }} onClick={() => setVisible(true)}>
                新增
              </Button>
              <Button
                type="primary"
                style={{ marginBottom: 20, marginLeft: 20 }}
                onClick={() => MultiAuth()}
              >
                上报所有工时
              </Button>
              {current.list.length > 0 && (
                <Button
                  type="primary"
                  style={{ marginBottom: 20, marginLeft: 20, float: 'right' }}
                  onClick={() => currentDayAuth()}
                >
                  上报今日工时
                </Button>
              )}
            </div>

            <WorkList allType={allType} list={current.list} onAuth={onAuth} onSave={onSave} />
          </Col>
        </Row>
      </Spin>

      <AddModal
        typeList={typeList}
        visible={visible}
        onOk={onAddWork}
        time={current.date?.format('YYYY-MM-DD')}
        onCancel={() => setVisible(false)}
      />
    </div>
  );
}

export default connect(({ workload, user, loading }) => ({
  dataList: workload.dataList,
  typeList: workload.typeList,
  allType: workload.allType,
  currentUser: user.currentUser,
  loading: loading.models.workload,
}))(List);