CSVServlet.java
4.88 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
/*
* Decompiled with CFR 0_118.
*
* Could not load the following classes:
* com.day.cq.commons.servlets.HtmlStatusResponseHelper
* com.day.cq.dam.api.Asset
* com.day.cq.dam.api.Rendition
* javax.servlet.ServletException
* javax.servlet.http.HttpServletResponse
* org.apache.felix.scr.annotations.Component
* org.apache.felix.scr.annotations.Properties
* org.apache.felix.scr.annotations.Property
* org.apache.felix.scr.annotations.Service
* org.apache.sling.api.SlingHttpServletRequest
* org.apache.sling.api.SlingHttpServletResponse
* org.apache.sling.api.request.RequestParameter
* org.apache.sling.api.resource.Resource
* org.apache.sling.api.resource.ValueMap
* org.apache.sling.api.servlets.SlingAllMethodsServlet
* org.slf4j.Logger
* org.slf4j.LoggerFactory
*/
package com.day.cq.dam.pim.impl.csv;
import com.day.cq.commons.servlets.HtmlStatusResponseHelper;
import com.day.cq.dam.api.Asset;
import com.day.cq.dam.api.Rendition;
import com.day.cq.dam.pim.PIMHelper;
import com.day.cq.dam.pim.impl.util.PIMUtils;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletResponse;
import org.apache.felix.scr.annotations.Component;
import org.apache.felix.scr.annotations.Properties;
import org.apache.felix.scr.annotations.Property;
import org.apache.felix.scr.annotations.Service;
import org.apache.sling.api.SlingHttpServletRequest;
import org.apache.sling.api.SlingHttpServletResponse;
import org.apache.sling.api.request.RequestParameter;
import org.apache.sling.api.resource.Resource;
import org.apache.sling.api.resource.ValueMap;
import org.apache.sling.api.servlets.SlingAllMethodsServlet;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@Component(metatype=0)
@Service
@Properties(value={@Property(name="sling.servlet.resourceTypes", value={"sling/servlet/default"}), @Property(name="sling.servlet.methods", value={"POST"}), @Property(name="sling.servlet.selectors", value={"csvedit"})})
public class CSVServlet
extends SlingAllMethodsServlet {
private static final Logger log = LoggerFactory.getLogger(CSVServlet.class);
protected void doPost(SlingHttpServletRequest request, SlingHttpServletResponse response) throws ServletException, IOException {
block10 : {
try {
Integer row = Integer.parseInt(request.getRequestParameter("row").getString());
String data = request.getRequestParameter("data").getString("UTF-8");
Resource resource = request.getResource();
Asset asset = (Asset)resource.adaptTo(Asset.class);
if (!asset.getMimeType().equals("text/csv")) break block10;
if (row < 0) {
asset.addRendition("original", PIMUtils.strToStream(data), "text/csv");
} else {
String line;
String encoding = "UTF-8";
try {
encoding = (String)((ValueMap)((Resource)asset.getOriginal().adaptTo(Resource.class)).getChild("jcr:content").adaptTo(ValueMap.class)).get("jcr:encoding", String.class);
}
catch (Exception e) {
log.warn("Unable to read encoding for " + asset.getPath());
}
InputStream dataStream = asset.getOriginal().getStream();
PIMHelper.skipBOM(dataStream, encoding);
BufferedReader reader = new BufferedReader(new InputStreamReader(dataStream, encoding));
StringBuilder csv = new StringBuilder(reader.readLine());
csv.append("\n");
int cnt = 0;
boolean found = false;
while ((line = reader.readLine()) != null) {
if (cnt == row) {
found = true;
csv.append(data);
} else {
csv.append(line);
}
csv.append("\n");
++cnt;
}
reader.close();
if (!found) {
csv.append(data);
}
asset.addRendition("original", PIMUtils.strToStream(csv.toString()), "text/csv");
}
HtmlStatusResponseHelper.createStatusResponse((boolean)true, (String)(asset.getName() + " saved successfully.")).send((HttpServletResponse)response, true);
}
catch (Exception e) {
log.warn("Error while saving CSV : " + e.getMessage());
HtmlStatusResponseHelper.createStatusResponse((boolean)false, (String)"Error while saving CSV").send((HttpServletResponse)response, true);
}
}
}
}