Avoid having default grey icons

3 files changed, 114 insertions, 3 deletions

diff --git a/sugar_network/node/routes.py b/sugar_network/node/routes.py
index 6a89e2c..17e7bbf 100644
--- a/sugar_network/node/routes.py
+++ b/sugar_network/node/routes.py
@@ -585,7 +585,7 @@ def load_bundle(volume, request, bundle_path):
 def _load_context_metadata(bundle, spec):
 
     def convert(data, w, h):
-        result = toolkit.svg_to_png(data.getvalue(), w, h)
+        result = toolkit.svg_to_png(data.getvalue(), w, h, spec['context'])
         return {'blob': result,
                 'mime_type': 'image/png',
                 'digest': hashlib.sha1(result.getvalue()).hexdigest(),
@@ -606,7 +606,7 @@ def _load_context_metadata(bundle, spec):
                 'mime_type': 'image/svg+xml',
                 'digest': hashlib.sha1(icon.getvalue()).hexdigest(),
                 },
-            'preview': convert(icon, 120, 120),
+            'preview': convert(icon, 140, 140),
             'icon': convert(icon, 55, 55),
             })
     except Exception:
diff --git a/sugar_network/toolkit/__init__.py b/sugar_network/toolkit/__init__.py
index cbe1b91..95d89dc 100644
--- a/sugar_network/toolkit/__init__.py
+++ b/sugar_network/toolkit/__init__.py
@@ -499,10 +499,14 @@ class mkdtemp(str):
         shutil.rmtree(self)
 
 
-def svg_to_png(data, w, h):
+def svg_to_png(data, w, h, replace_color=None):
     import rsvg
     import cairo
 
+    if replace_color:
+        from sugar_network.toolkit.sugar import color_svg
+        data = color_svg(data, replace_color)
+
     svg = rsvg.Handle(data=data)
     surface = cairo.ImageSurface(cairo.FORMAT_ARGB32, w, h)
     context = cairo.Context(surface)
@@ -517,6 +521,7 @@ def svg_to_png(data, w, h):
     result = StringIO()
     surface.write_to_png(result)
     result.seek(0)
+
     return result
 
 
diff --git a/sugar_network/toolkit/sugar.py b/sugar_network/toolkit/sugar.py
new file mode 100644
index 0000000..2955165
--- /dev/null
+++ b/sugar_network/toolkit/sugar.py
@@ -0,0 +1,106 @@
+# Copyright (C) 2013 Aleksey Lim
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import re
+
+
+_STROKE_COLOR_RE = re.compile(r'<!ENTITY\s+stroke_color\s+"([^"]+)">')
+_FILL_COLOR_RE = re.compile(r'<!ENTITY\s+fill_color\s+"([^"]+)">')
+
+
+def color_svg(svg, colors):
+    orig_svg = svg
+    if isinstance(colors, basestring):
+        colors = XO_COLORS[hash(colors) % len(XO_COLORS)]
+
+    for regexp, new_color in [
+            (_STROKE_COLOR_RE, ('stroke_color', colors[0])),
+            (_FILL_COLOR_RE, ('fill_color', colors[1])),
+            ]:
+        parts = regexp.split(svg)
+        if len(parts) == 1:
+            continue
+        before, current_color, after = parts
+        current_color = zip(*[iter(current_color.strip('#'))] * 2)
+        if current_color.count(current_color[0]) < len(current_color):
+            # Preserve original colors if they are not monochrome
+            return orig_svg
+        svg = ''.join([before, '<!ENTITY %s "%s">' % new_color, after])
+
+    return svg
+
+
+XO_COLORS = [
+        ['#B20008', '#FF2B34'], ['#FF2B34', '#B20008'], ['#E6000A', '#FF2B34'],
+        ['#FF2B34', '#E6000A'], ['#FFADCE', '#FF2B34'], ['#9A5200', '#FF2B34'],
+        ['#FF2B34', '#9A5200'], ['#FF8F00', '#FF2B34'], ['#FF2B34', '#FF8F00'],
+        ['#FFC169', '#FF2B34'], ['#807500', '#FF2B34'], ['#FF2B34', '#807500'],
+        ['#BE9E00', '#FF2B34'], ['#FF2B34', '#BE9E00'], ['#F8E800', '#FF2B34'],
+        ['#008009', '#FF2B34'], ['#FF2B34', '#008009'], ['#00B20D', '#FF2B34'],
+        ['#FF2B34', '#00B20D'], ['#8BFF7A', '#FF2B34'], ['#00588C', '#FF2B34'],
+        ['#FF2B34', '#00588C'], ['#005FE4', '#FF2B34'], ['#FF2B34', '#005FE4'],
+        ['#BCCDFF', '#FF2B34'], ['#5E008C', '#FF2B34'], ['#FF2B34', '#5E008C'],
+        ['#7F00BF', '#FF2B34'], ['#FF2B34', '#7F00BF'], ['#D1A3FF', '#FF2B34'],
+        ['#9A5200', '#FF8F00'], ['#FF8F00', '#9A5200'], ['#C97E00', '#FF8F00'],
+        ['#FF8F00', '#C97E00'], ['#FFC169', '#FF8F00'], ['#807500', '#FF8F00'],
+        ['#FF8F00', '#807500'], ['#BE9E00', '#FF8F00'], ['#FF8F00', '#BE9E00'],
+        ['#F8E800', '#FF8F00'], ['#008009', '#FF8F00'], ['#FF8F00', '#008009'],
+        ['#00B20D', '#FF8F00'], ['#FF8F00', '#00B20D'], ['#8BFF7A', '#FF8F00'],
+        ['#00588C', '#FF8F00'], ['#FF8F00', '#00588C'], ['#005FE4', '#FF8F00'],
+        ['#FF8F00', '#005FE4'], ['#BCCDFF', '#FF8F00'], ['#5E008C', '#FF8F00'],
+        ['#FF8F00', '#5E008C'], ['#A700FF', '#FF8F00'], ['#FF8F00', '#A700FF'],
+        ['#D1A3FF', '#FF8F00'], ['#B20008', '#FF8F00'], ['#FF8F00', '#B20008'],
+        ['#FF2B34', '#FF8F00'], ['#FF8F00', '#FF2B34'], ['#FFADCE', '#FF8F00'],
+        ['#807500', '#F8E800'], ['#F8E800', '#807500'], ['#BE9E00', '#F8E800'],
+        ['#F8E800', '#BE9E00'], ['#FFFA00', '#EDDE00'], ['#008009', '#F8E800'],
+        ['#F8E800', '#008009'], ['#00EA11', '#F8E800'], ['#F8E800', '#00EA11'],
+        ['#8BFF7A', '#F8E800'], ['#00588C', '#F8E800'], ['#F8E800', '#00588C'],
+        ['#00A0FF', '#F8E800'], ['#F8E800', '#00A0FF'], ['#BCCEFF', '#F8E800'],
+        ['#5E008C', '#F8E800'], ['#F8E800', '#5E008C'], ['#AC32FF', '#F8E800'],
+        ['#F8E800', '#AC32FF'], ['#D1A3FF', '#F8E800'], ['#B20008', '#F8E800'],
+        ['#F8E800', '#B20008'], ['#FF2B34', '#F8E800'], ['#F8E800', '#FF2B34'],
+        ['#FFADCE', '#F8E800'], ['#9A5200', '#F8E800'], ['#F8E800', '#9A5200'],
+        ['#FF8F00', '#F8E800'], ['#F8E800', '#FF8F00'], ['#FFC169', '#F8E800'],
+        ['#008009', '#00EA11'], ['#00EA11', '#008009'], ['#00B20D', '#00EA11'],
+        ['#00EA11', '#00B20D'], ['#8BFF7A', '#00EA11'], ['#00588C', '#00EA11'],
+        ['#00EA11', '#00588C'], ['#005FE4', '#00EA11'], ['#00EA11', '#005FE4'],
+        ['#BCCDFF', '#00EA11'], ['#5E008C', '#00EA11'], ['#00EA11', '#5E008C'],
+        ['#7F00BF', '#00EA11'], ['#00EA11', '#7F00BF'], ['#D1A3FF', '#00EA11'],
+        ['#B20008', '#00EA11'], ['#00EA11', '#B20008'], ['#FF2B34', '#00EA11'],
+        ['#00EA11', '#FF2B34'], ['#FFADCE', '#00EA11'], ['#9A5200', '#00EA11'],
+        ['#00EA11', '#9A5200'], ['#FF8F00', '#00EA11'], ['#00EA11', '#FF8F00'],
+        ['#FFC169', '#00EA11'], ['#807500', '#00EA11'], ['#00EA11', '#807500'],
+        ['#BE9E00', '#00EA11'], ['#00EA11', '#BE9E00'], ['#F8E800', '#00EA11'],
+        ['#00588C', '#00A0FF'], ['#00A0FF', '#00588C'], ['#005FE4', '#00A0FF'],
+        ['#00A0FF', '#005FE4'], ['#BCCDFF', '#00A0FF'], ['#5E008C', '#00A0FF'],
+        ['#00A0FF', '#5E008C'], ['#9900E6', '#00A0FF'], ['#00A0FF', '#9900E6'],
+        ['#D1A3FF', '#00A0FF'], ['#B20008', '#00A0FF'], ['#00A0FF', '#B20008'],
+        ['#FF2B34', '#00A0FF'], ['#00A0FF', '#FF2B34'], ['#FFADCE', '#00A0FF'],
+        ['#9A5200', '#00A0FF'], ['#00A0FF', '#9A5200'], ['#FF8F00', '#00A0FF'],
+        ['#00A0FF', '#FF8F00'], ['#FFC169', '#00A0FF'], ['#807500', '#00A0FF'],
+        ['#00A0FF', '#807500'], ['#BE9E00', '#00A0FF'], ['#00A0FF', '#BE9E00'],
+        ['#F8E800', '#00A0FF'], ['#008009', '#00A0FF'], ['#00A0FF', '#008009'],
+        ['#00B20D', '#00A0FF'], ['#00A0FF', '#00B20D'], ['#8BFF7A', '#00A0FF'],
+        ['#5E008C', '#AC32FF'], ['#AC32FF', '#5E008C'], ['#7F00BF', '#AC32FF'],
+        ['#AC32FF', '#7F00BF'], ['#D1A3FF', '#AC32FF'], ['#B20008', '#AC32FF'],
+        ['#AC32FF', '#B20008'], ['#FF2B34', '#AC32FF'], ['#AC32FF', '#FF2B34'],
+        ['#FFADCE', '#AC32FF'], ['#9A5200', '#AC32FF'], ['#AC32FF', '#9A5200'],
+        ['#FF8F00', '#AC32FF'], ['#AC32FF', '#FF8F00'], ['#FFC169', '#AC32FF'],
+        ['#807500', '#AC32FF'], ['#AC32FF', '#807500'], ['#BE9E00', '#AC32FF'],
+        ['#AC32FF', '#BE9E00'], ['#F8E800', '#AC32FF'], ['#008009', '#AC32FF'],
+        ['#AC32FF', '#008009'], ['#00B20D', '#AC32FF'], ['#AC32FF', '#00B20D'],
+        ['#8BFF7A', '#AC32FF'], ['#00588C', '#AC32FF'], ['#AC32FF', '#00588C'],
+        ['#005FE4', '#AC32FF'], ['#AC32FF', '#005FE4'], ['#BCCDFF', '#AC32FF'],
+        ]