bodyshop/client/src/utils/RenderTemplate.js

import { gql } from "@apollo/client";
import { notification } from "antd";
import axios from "axios";
import jsreport from "jsreport-browser-client-dist";
import _ from "lodash";
import moment from "moment";
import { auth } from "../firebase/firebase.utils";
import { setEmailOptions } from "../redux/email/email.actions";
import { store } from "../redux/store";
import client from "../utils/GraphQLClient";
import { TemplateList } from "./TemplateConstants";
const server = process.env.REACT_APP_REPORTS_SERVER_URL;
jsreport.serverUrl = server;

const Templates = TemplateList();

export default async function RenderTemplate(
  templateObject,
  bodyshop,
  renderAsHtml = false,
  renderAsExcel = false
) {
  //Query assets that match the template name. Must be in format <<templateName>>.query
  let { contextData, useShopSpecificTemplate } = await fetchContextData(
    templateObject
  );

  let reportRequest = {
    template: {
      name: useShopSpecificTemplate
        ? `/${bodyshop.imexshopid}/${templateObject.name}`
        : `/${templateObject.name}`,
      ...(renderAsHtml ? {} : { recipe: "chrome-pdf" }),
      ...(renderAsExcel ? { recipe: "html-to-xlsx" } : {}),
    },
    data: {
      ...contextData,
      ...templateObject.variables,
      ...templateObject.context,
      headerpath: `/${bodyshop.imexshopid}/header.html`,
      bodyshop: bodyshop,
      offset: moment().utcOffset(),
    },
  };

  try {
    const render = await jsreport.renderAsync(reportRequest);

    if (!renderAsHtml) {
      render.download(
        (Templates[templateObject.name] &&
          Templates[templateObject.name].title) ||
          ""
      );
    } else {
      let pdf;
      if (bodyshop.attach_pdf_to_email) {
        const pdfRequest = _.cloneDeep(reportRequest);
        pdfRequest.template.recipe = "chrome-pdf";
        const pdfRender = await jsreport.renderAsync(pdfRequest);
        pdf = pdfRender.toDataURI();
      }
      return new Promise((resolve, reject) => {
        resolve({
          pdf,
          filename:
            Templates[templateObject.name] &&
            Templates[templateObject.name].title,
          html: render.toString(),
        });
      });
    }
  } catch (error) {
    notification["error"]({ message: JSON.stringify(error) });
  }
}

export async function RenderTemplates(
  templateObjects,
  bodyshop,
  renderAsHtml = false
) {
  //Query assets that match the template name. Must be in format <<templateName>>.query
  let unsortedTemplatesAndData = [];
  let proms = [];
  templateObjects.forEach((template) => {
    proms.push(
      (async () => {
        let { contextData, useShopSpecificTemplate } = await fetchContextData(
          template
        );
        unsortedTemplatesAndData.push({
          templateObject: template,
          contextData,
          useShopSpecificTemplate,
        });
      })()
    );
  });
  await Promise.all(proms);

  //Re-order list to follow speedprint.
  // var templateAndData = _.sortBy(unsortedTemplatesAndData, function (item) {
  //   return templateObjects.findIndex(
  //     (template) => template.name === item.templateObject.name
  //   );
  // });

  unsortedTemplatesAndData.sort(function (a, b) {
    return (
      templateObjects.findIndex((x) => x.name === a.templateObject.name) -
      templateObjects.findIndex((x) => x.name === b.templateObject.name)
    );
  });
  const templateAndData = unsortedTemplatesAndData;

  let rootTemplate = templateAndData.shift();

  let reportRequest = {
    template: {
      name: rootTemplate.useShopSpecificTemplate
        ? `/${bodyshop.imexshopid}/${rootTemplate.templateObject.name}`
        : `/${rootTemplate.templateObject.name}`,
      ...(renderAsHtml ? {} : { recipe: "chrome-pdf" }),
      pdfOperations: templateAndData.map((template) => {
        return {
          template: {
            name: template.useShopSpecificTemplate
              ? `/${bodyshop.imexshopid}/${template.templateObject.name}`
              : `/${template.templateObject.name}`,
            ...(renderAsHtml ? {} : { recipe: "chrome-pdf" }),
          },
          type: "append",
          mergeWholeDocument: true,
          renderForEveryPage: true,
        };
      }),
    },
    data: {
      ...extend(
        rootTemplate.contextData,
        ...templateAndData.map((temp) => temp.contextData)
      ),

      // ...rootTemplate.templateObject.variables,
      // ...rootTemplate.templateObject.context,
      headerpath: `/${bodyshop.imexshopid}/header.html`,
      bodyshop: bodyshop,
      offset: moment().utcOffset(),
    },
  };

  try {
    const render = await jsreport.renderAsync(reportRequest);
    if (!renderAsHtml) {
      render.download("Speed Print");
    } else {
      return new Promise((resolve, reject) => {
        resolve(render.toString());
      });
    }
  } catch (error) {
    notification["error"]({ message: JSON.stringify(error) });
  }
}

export const GenerateDocument = async (
  template,
  messageOptions,
  sendType,
  jobid
) => {
  const bodyshop = store.getState().user.bodyshop;
  if (sendType === "e") {
    store.dispatch(
      setEmailOptions({
        jobid,
        messageOptions: {
          ...messageOptions,
          to: Array.isArray(messageOptions.to)
            ? messageOptions.to
            : [messageOptions.to],
        },
        template,
      })
    );
  } else if (sendType === "x") {
    console.log("excel");
    await RenderTemplate(template, bodyshop, false, true);
  } else {
    await RenderTemplate(template, bodyshop);
  }
};

export const GenerateDocuments = async (templates) => {
  const bodyshop = store.getState().user.bodyshop;
  await RenderTemplates(templates, bodyshop);
};

const fetchContextData = async (templateObject) => {
  const bodyshop = store.getState().user.bodyshop;

  jsreport.headers["Authorization"] =
    "Bearer " + (await auth.currentUser.getIdToken());

  const folders = await axios.get(`${server}/odata/folders`);
  const shopSpecificFolder = folders.data.value.find(
    (f) => f.name === bodyshop.imexshopid
  );

  const jsReportQueries = await axios.get(
    `${server}/odata/assets?$filter=name eq '${templateObject.name}.query'`
  );

  let templateQueryToExecute;
  let useShopSpecificTemplate = false;
  // let shopSpecificTemplate;

  if (shopSpecificFolder) {
    let shopSpecificTemplate = jsReportQueries.data.value.find(
      (f) => f?.folder?.shortid === shopSpecificFolder.shortid
    );
    if (shopSpecificTemplate) {
      useShopSpecificTemplate = true;
      templateQueryToExecute = atob(shopSpecificTemplate.content);
    }
  }

  if (!templateQueryToExecute) {
    const generalTemplate = jsReportQueries.data.value.find((f) => !f.folder);
    useShopSpecificTemplate = false;
    templateQueryToExecute = atob(generalTemplate.content);
  }

  let contextData = {};
  if (templateQueryToExecute) {
    const { data } = await client.query({
      query: gql(templateQueryToExecute),
      variables: { ...templateObject.variables },
    });
    contextData = data;
  }

  return { contextData, useShopSpecificTemplate };
};

//export const displayTemplateInWindow = (html) => {
//   try {
//     var newWin = window.open("", "_blank", "toolbar=0,location=0,menubar=0");
//     newWin.document.write(html);

//     setTimeout(function () {
//       newWin.document.close();
//       newWin.focus();
//       newWin.print();
//       newWin.close();
//     }, 500);
//   } catch (error) {
//     console.log("Unable to write to new window.", error);
//   }
// };

// export const displayTemplateInWindowNoprint = (html) => {
//   try {
//     var newWin = window.open("", "_blank", "toolbar=0,location=0,menubar=0");
//     newWin.document.write(html);

//     setTimeout(function () {
//       newWin.document.close();
//       newWin.focus();
//       //newWin.print();
//       //newWin.close();
//     }, 500);
//   } catch (error) {
//     console.log("Unable to write to new window.", error);
//   }
// };

function extend(o1, o2, o3) {
  var result = {},
    obj;

  for (var i = 0; i < arguments.length; i++) {
    obj = arguments[i];
    for (var key in obj) {
      if (Object.prototype.toString.call(obj[key]) === "[object Object]") {
        if (typeof result[key] === "undefined") {
          result[key] = {};
        }
        result[key] = extend(result[key], obj[key]);
      } else {
        result[key] = obj[key];
      }
    }
  }
  return result;
}