package com.fr.plugin.export.xml.core;

import com.fr.general.DeclareRecordType;
import com.fr.io.collection.ExportCollection;
import com.fr.io.exporter.CSVExporter;
import com.fr.plugin.export.xml.exporter.XmlExporter;
import com.fr.web.core.ReportSessionIDInfor;
import com.fr.web.core.reserve.DefaultOperate;
import com.fr.web.core.utils.ExportUtils;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * Created by richie on 16/1/19.
 */
public class XmlExportOperate extends DefaultOperate {

    @Override
    public void setContent(HttpServletRequest req, HttpServletResponse res, String fileName, boolean isEmbed) {
        res.setContentType("application/octet-stream");
        res.setHeader("extension", "xml");
        res.setHeader("Content-disposition", "attachment; filename=" + fileName + ".xml");
    }

    @Override
    public ExportCollection createCollection(HttpServletRequest req, HttpServletResponse res, ReportSessionIDInfor sessionIDInfor, String fileName) {
        ExportCollection collection = ExportCollection.create();
        collection.setExporter(new XmlExporter());
        collection.setRecordType(DeclareRecordType.EXPORT_TYPE_CSV);

        return collection;
    }
}