--- /dev/null
+/*\r
+ * Copyright 2007 Daniel Spiewak.\r
+ * Copyright 2013 gitblit.com.\r
+ *\r
+ * Licensed under the Apache License, Version 2.0 (the "License");\r
+ * you may not use this file except in compliance with the License.\r
+ * You may obtain a copy of the License at\r
+ *\r
+ * http://www.apache.org/licenses/LICENSE-2.0\r
+ *\r
+ * Unless required by applicable law or agreed to in writing, software\r
+ * distributed under the License is distributed on an "AS IS" BASIS,\r
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
+ * See the License for the specific language governing permissions and\r
+ * limitations under the License.\r
+ */\r
+package com.gitblit.wicket.charting;\r
+\r
+import java.awt.Color;\r
+import java.awt.Dimension;\r
+import java.io.Serializable;\r
+import java.lang.reflect.InvocationTargetException;\r
+import java.util.concurrent.locks.ReadWriteLock;\r
+import java.util.concurrent.locks.ReentrantReadWriteLock;\r
+\r
+import org.apache.wicket.markup.ComponentTag;\r
+import org.apache.wicket.markup.html.WebComponent;\r
+import org.wicketstuff.googlecharts.ChartDataEncoding;\r
+import org.wicketstuff.googlecharts.IChartAxis;\r
+import org.wicketstuff.googlecharts.IChartData;\r
+import org.wicketstuff.googlecharts.IChartFill;\r
+import org.wicketstuff.googlecharts.IChartGrid;\r
+import org.wicketstuff.googlecharts.IChartProvider;\r
+import org.wicketstuff.googlecharts.IFillArea;\r
+import org.wicketstuff.googlecharts.ILineStyle;\r
+import org.wicketstuff.googlecharts.ILinearGradientFill;\r
+import org.wicketstuff.googlecharts.ILinearStripesFill;\r
+import org.wicketstuff.googlecharts.IRangeMarker;\r
+import org.wicketstuff.googlecharts.IShapeMarker;\r
+import org.wicketstuff.googlecharts.ISolidFill;\r
+import org.wicketstuff.googlecharts.Range;\r
+\r
+/**\r
+ * This is a fork of org.wicketstuff.googlecharts.Chart whose only purpose\r
+ * is to build https urls instead of http urls.\r
+ * \r
+ * @author Daniel Spiewak\r
+ * @author James Moger\r
+ */\r
+public class SecureChart extends WebComponent implements Serializable {\r
+\r
+ private static final long serialVersionUID = 6286305912682861488L;\r
+ private IChartProvider provider;\r
+ private StringBuilder url;\r
+ private final ReadWriteLock lock = new ReentrantReadWriteLock();\r
+\r
+ public SecureChart(String id, IChartProvider provider) {\r
+ super(id);\r
+\r
+ this.provider = provider;\r
+ }\r
+\r
+ public void invalidate() {\r
+ lock.writeLock().lock();\r
+ try {\r
+ url = null;\r
+ } finally {\r
+ lock.writeLock().unlock();\r
+ }\r
+ }\r
+\r
+ public CharSequence constructURL() {\r
+ lock.writeLock().lock();\r
+ try {\r
+ if (url != null) {\r
+ return url;\r
+ }\r
+\r
+ url = new StringBuilder("https://chart.googleapis.com/chart?");\r
+\r
+ addParameter(url, "chs", render(provider.getSize()));\r
+ addParameter(url, "chd", render(provider.getData()));\r
+ addParameter(url, "cht", render(provider.getType()));\r
+ addParameter(url, "chbh", render(provider.getBarWidth(), provider.getBarGroupSpacing()));\r
+ addParameter(url, "chtt", render(provider.getTitle()));\r
+ addParameter(url, "chdl", render(provider.getLegend()));\r
+ addParameter(url, "chco", render(provider.getColors()));\r
+\r
+ IChartFill bgFill = provider.getBackgroundFill();\r
+ IChartFill fgFill = provider.getChartFill();\r
+\r
+ StringBuilder fillParam = new StringBuilder();\r
+\r
+ if (bgFill != null) {\r
+ fillParam.append("bg,").append(render(bgFill));\r
+ }\r
+\r
+ if (fgFill != null) {\r
+ if (fillParam.length() > 0) {\r
+ fillParam.append('|');\r
+ }\r
+\r
+ fillParam.append("c,").append(render(fgFill));\r
+ }\r
+\r
+ if (fillParam.toString().trim().equals("")) {\r
+ fillParam = null;\r
+ }\r
+\r
+ addParameter(url, "chf", fillParam);\r
+\r
+ IChartAxis[] axes = provider.getAxes();\r
+ addParameter(url, "chxt", renderTypes(axes));\r
+ addParameter(url, "chxl", renderLabels(axes));\r
+ addParameter(url, "chxp", renderPositions(axes));\r
+ addParameter(url, "chxr", renderRanges(axes));\r
+ addParameter(url, "chxs", renderStyles(axes));\r
+\r
+ addParameter(url, "chg", render(provider.getGrid()));\r
+ addParameter(url, "chm", render(provider.getShapeMarkers()));\r
+ addParameter(url, "chm", render(provider.getRangeMarkers()));\r
+ addParameter(url, "chls", render(provider.getLineStyles()));\r
+ addParameter(url, "chm", render(provider.getFillAreas()));\r
+ addParameter(url, "chl", render(provider.getPieLabels()));\r
+\r
+ return url;\r
+ } finally {\r
+ lock.writeLock().unlock();\r
+ }\r
+ }\r
+\r
+ private void addParameter(StringBuilder url, CharSequence param, CharSequence value) {\r
+ if (value == null || value.length() == 0) {\r
+ return;\r
+ }\r
+\r
+ if (url.charAt(url.length() - 1) != '?') {\r
+ url.append('&');\r
+ }\r
+\r
+ url.append(param).append('=').append(value);\r
+ }\r
+ \r
+ private CharSequence convert(ChartDataEncoding encoding, double value, double max) {\r
+ switch (encoding) {\r
+ case TEXT:\r
+ return SecureChartDataEncoding.TEXT.convert(value, max);\r
+ case EXTENDED:\r
+ return SecureChartDataEncoding.EXTENDED.convert(value, max);\r
+ case SIMPLE:\r
+ default:\r
+ return SecureChartDataEncoding.SIMPLE.convert(value, max);\r
+ }\r
+ }\r
+\r
+ private CharSequence render(Dimension dim) {\r
+ if (dim == null) {\r
+ return null;\r
+ }\r
+\r
+ return new StringBuilder().append(dim.width).append('x').append(dim.height);\r
+ }\r
+\r
+ private CharSequence render(IChartData data) {\r
+ if (data == null) {\r
+ return null;\r
+ }\r
+\r
+ ChartDataEncoding encoding = data.getEncoding();\r
+\r
+ StringBuilder back = new StringBuilder();\r
+ back.append(render(encoding)).append(':');\r
+\r
+ for (double[] set : data.getData()) {\r
+ if (set == null || set.length == 0) {\r
+ back.append(convert(encoding, -1, data.getMax()));\r
+ } else {\r
+ for (double value : set) {\r
+ back.append(convert(encoding, value, data.getMax())).append(encoding.getValueSeparator());\r
+ }\r
+\r
+ if (back.substring(back.length() - encoding.getValueSeparator().length(),\r
+ back.length()).equals(encoding.getValueSeparator())) {\r
+ back.setLength(back.length() - encoding.getValueSeparator().length());\r
+ }\r
+ }\r
+\r
+ back.append(encoding.getSetSeparator());\r
+ }\r
+\r
+ if (back.substring(back.length() - encoding.getSetSeparator().length(),\r
+ back.length()).equals(encoding.getSetSeparator())) {\r
+ back.setLength(back.length() - encoding.getSetSeparator().length());\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence render(Enum<?> value) {\r
+ if (value == null) {\r
+ return null;\r
+ }\r
+\r
+ try {\r
+ Object back = value.getClass().getMethod("getRendering").invoke(value);\r
+\r
+ if (back != null) {\r
+ return back.toString();\r
+ }\r
+ } catch (IllegalArgumentException e) {\r
+ } catch (SecurityException e) {\r
+ } catch (IllegalAccessException e) {\r
+ } catch (InvocationTargetException e) {\r
+ } catch (NoSuchMethodException e) {\r
+ }\r
+\r
+ return null;\r
+ }\r
+\r
+ private CharSequence render(int barWidth, int groupSpacing) {\r
+ if (barWidth == -1) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder(barWidth);\r
+\r
+ if (groupSpacing >= 0) {\r
+ back.append(',').append(groupSpacing);\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence render(String[] values) {\r
+ if (values == null) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ for (String value : values) {\r
+ CharSequence toRender = render(value);\r
+ if (toRender == null) {\r
+ toRender = "";\r
+ }\r
+\r
+ back.append(toRender).append('|');\r
+ }\r
+\r
+ if (back.length() > 0) {\r
+ back.setLength(back.length() - 1);\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence render(String value) {\r
+ if (value == null) {\r
+ return value;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ for (char c : value.toCharArray()) {\r
+ if (c == ' ') {\r
+ back.append('+');\r
+ } else {\r
+ back.append(c);\r
+ }\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence render(Color[] values) {\r
+ if (values == null) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ for (Color value : values) {\r
+ CharSequence toRender = render(value);\r
+ if (toRender == null) {\r
+ toRender = "";\r
+ }\r
+\r
+ back.append(toRender).append(',');\r
+ }\r
+\r
+ if (back.length() > 0) {\r
+ back.setLength(back.length() - 1);\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence render(Color value) {\r
+ if (value == null) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ {\r
+ String toPad = Integer.toHexString(value.getRed());\r
+\r
+ if (toPad.length() == 1) {\r
+ back.append(0);\r
+ }\r
+ back.append(toPad);\r
+ }\r
+\r
+ {\r
+ String toPad = Integer.toHexString(value.getGreen());\r
+\r
+ if (toPad.length() == 1) {\r
+ back.append(0);\r
+ }\r
+ back.append(toPad);\r
+ }\r
+\r
+ {\r
+ String toPad = Integer.toHexString(value.getBlue());\r
+\r
+ if (toPad.length() == 1) {\r
+ back.append(0);\r
+ }\r
+ back.append(toPad);\r
+ }\r
+\r
+ {\r
+ String toPad = Integer.toHexString(value.getAlpha());\r
+\r
+ if (toPad.length() == 1) {\r
+ back.append(0);\r
+ }\r
+ back.append(toPad);\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence render(IChartFill fill) {\r
+ if (fill == null) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ if (fill instanceof ISolidFill) {\r
+ ISolidFill solidFill = (ISolidFill) fill;\r
+\r
+ back.append("s,");\r
+ back.append(render(solidFill.getColor()));\r
+ } else if (fill instanceof ILinearGradientFill) {\r
+ ILinearGradientFill gradientFill = (ILinearGradientFill) fill;\r
+\r
+ back.append("lg,").append(gradientFill.getAngle()).append(',');\r
+\r
+ Color[] colors = gradientFill.getColors();\r
+ double[] offsets = gradientFill.getOffsets();\r
+ for (int i = 0; i < colors.length; i++) {\r
+ back.append(render(colors[i])).append(',').append(offsets[i]).append(',');\r
+ }\r
+\r
+ back.setLength(back.length() - 1);\r
+ } else if (fill instanceof ILinearStripesFill) {\r
+ ILinearStripesFill stripesFill = (ILinearStripesFill) fill;\r
+\r
+ back.append("ls,").append(stripesFill.getAngle()).append(',');\r
+\r
+ Color[] colors = stripesFill.getColors();\r
+ double[] widths = stripesFill.getWidths();\r
+ for (int i = 0; i < colors.length; i++) {\r
+ back.append(render(colors[i])).append(',').append(widths[i]).append(',');\r
+ }\r
+\r
+ back.setLength(back.length() - 1);\r
+ } else {\r
+ return null;\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence renderTypes(IChartAxis[] axes) {\r
+ if (axes == null) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ for (IChartAxis axis : axes) {\r
+ back.append(render(axis.getType())).append(',');\r
+ }\r
+\r
+ if (back.length() > 0) {\r
+ back.setLength(back.length() - 1);\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence renderLabels(IChartAxis[] axes) {\r
+ if (axes == null) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ for (int i = 0; i < axes.length; i++) {\r
+ if (axes[i] == null || axes[i].getLabels() == null) {\r
+ continue;\r
+ }\r
+\r
+ back.append(i).append(":|");\r
+\r
+ for (String label : axes[i].getLabels()) {\r
+ if (label == null) {\r
+ back.append('|');\r
+ continue;\r
+ }\r
+\r
+ back.append(render(label)).append('|');\r
+ }\r
+\r
+ if (i == axes.length - 1) {\r
+ back.setLength(back.length() - 1);\r
+ }\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence renderPositions(IChartAxis[] axes) {\r
+ if (axes == null) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ for (int i = 0; i < axes.length; i++) {\r
+ if (axes[i] == null || axes[i].getPositions() == null) {\r
+ continue;\r
+ }\r
+\r
+ back.append(i).append(',');\r
+\r
+ for (double position : axes[i].getPositions()) {\r
+ back.append(position).append(',');\r
+ }\r
+\r
+ back.setLength(back.length() - 1);\r
+\r
+ back.append('|');\r
+ }\r
+\r
+ if (back.length() > 0) {\r
+ back.setLength(back.length() - 1);\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence renderRanges(IChartAxis[] axes) {\r
+ if (axes == null) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ for (int i = 0; i < axes.length; i++) {\r
+ if (axes[i] == null || axes[i].getRange() == null) {\r
+ continue;\r
+ }\r
+\r
+ back.append(i).append(',');\r
+\r
+ Range range = axes[i].getRange();\r
+ back.append(range.getStart()).append(',').append(range.getEnd()).append('|');\r
+ }\r
+\r
+ if (back.length() > 0) {\r
+ back.setLength(back.length() - 1);\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence renderStyles(IChartAxis[] axes) {\r
+ if (axes == null) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ for (int i = 0; i < axes.length; i++) {\r
+ if (axes[i] == null || axes[i].getColor() == null\r
+ || axes[i].getFontSize() < 0 || axes[i].getAlignment() == null) {\r
+ continue;\r
+ }\r
+\r
+ back.append(i).append(',');\r
+ back.append(render(axes[i].getColor())).append(',');\r
+ back.append(axes[i].getFontSize()).append(',');\r
+ back.append(render(axes[i].getAlignment())).append('|');\r
+ }\r
+\r
+ if (back.length() > 0) {\r
+ back.setLength(back.length() - 1);\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence render(IChartGrid grid) {\r
+ if (grid == null) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ back.append(grid.getXStepSize()).append(',');\r
+ back.append(grid.getYStepSize());\r
+\r
+ if (grid.getSegmentLength() >= 0) {\r
+ back.append(',').append(grid.getSegmentLength());\r
+ back.append(',').append(grid.getBlankLength());\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence render(IShapeMarker[] markers) {\r
+ if (markers == null) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ for (IShapeMarker marker : markers) {\r
+ back.append(render(marker.getType())).append(',');\r
+ back.append(render(marker.getColor())).append(',');\r
+ back.append(marker.getIndex()).append(',');\r
+ back.append(marker.getPoint()).append(',');\r
+ back.append(marker.getSize()).append('|');\r
+ }\r
+\r
+ if (back.length() > 0) {\r
+ back.setLength(back.length() - 1);\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence render(IRangeMarker[] markers) {\r
+ if (markers == null) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ for (IRangeMarker marker : markers) {\r
+ back.append(render(marker.getType())).append(',');\r
+ back.append(render(marker.getColor())).append(',');\r
+ back.append(0).append(',');\r
+ back.append(marker.getStart()).append(',');\r
+ back.append(marker.getEnd()).append('|');\r
+ }\r
+\r
+ if (back.length() > 0) {\r
+ back.setLength(back.length() - 1);\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence render(IFillArea[] areas) {\r
+ if (areas == null) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ for (IFillArea area : areas) {\r
+ back.append(render(area.getType())).append(',');\r
+ back.append(render(area.getColor())).append(',');\r
+ back.append(area.getStartIndex()).append(',');\r
+ back.append(area.getEndIndex()).append(',');\r
+ back.append(0).append('|');\r
+ }\r
+\r
+ if (back.length() > 0) {\r
+ back.setLength(back.length() - 1);\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ private CharSequence render(ILineStyle[] styles) {\r
+ if (styles == null) {\r
+ return null;\r
+ }\r
+\r
+ StringBuilder back = new StringBuilder();\r
+\r
+ for (ILineStyle style : styles) {\r
+ if (style == null) {\r
+ back.append('|');\r
+ continue;\r
+ }\r
+\r
+ back.append(style.getThickness()).append(',');\r
+ back.append(style.getSegmentLength()).append(',');\r
+ back.append(style.getBlankLength()).append('|');\r
+ }\r
+\r
+ if (back.length() > 0) {\r
+ back.setLength(back.length() - 1);\r
+ }\r
+\r
+ return back;\r
+ }\r
+\r
+ @Override\r
+ protected void onComponentTag(ComponentTag tag) {\r
+ checkComponentTag(tag, "img");\r
+ super.onComponentTag(tag);\r
+\r
+ tag.put("src", constructURL());\r
+ }\r
+}\r