New way of taking other color schemes

Author: 2016rshah

app.rb

@@ -1,11 +1,12 @@
 require 'sinatra'
 require 'githubchart'
 
-COLOR_SCHEMES = {
-    default: ['#eeeeee', '#d6e685', '#8cc665', '#44a340', '#1e6823'],
-    halloween: ['#EEEEEE', '#FFEE4A', '#FFC501', '#FE9600', '#03001C'],
-    teal: ['#EEEEEE', "#7FFFD4", "#76EEC6", "#66CDAA", "#458B74"]
-}
+require './color.rb'
+
+# Examples of nice colors to try out to start with:
+# blue: 409ba5
+# red: 720100
+# green: 44a340
 
 get '/' do
     send_file File.join(settings.public_folder, 'index.html')
@@ -21,12 +22,13 @@
     end
 end
 
-get '/:scheme/:username' do
-	#example: ghchart.rshah.io/teal/2016rshah
-	#if scheme is not recognized, it will bounce back to the default color scheme
-	headers 'Content-Type' => "image/svg+xml"
+get '/:base/:username' do
+    headers 'Content-Type' => "image/svg+xml"
     username = params[:username].chomp('.svg')
-    scheme = COLOR_SCHEMES[params[:scheme].to_sym] #must correspond to the color schemes specified above
+
+    base_color = '#'+params[:base]
+    scheme = ['#EEEEEE', lighten_color(base_color, 0.3), lighten_color(base_color, 0.2), base_color, darken_color(base_color, 0.8)]
+    
     svg = GithubChart.new(user: username, colors: scheme).svg
     stream do |out|
       out << svg

color.rb

@@ -0,0 +1,28 @@
+#http://www.redguava.com.au/2011/10/lighten-or-darken-a-hexadecimal-color-in-ruby-on-rails/
+
+# COLOR_SCHEMES = {
+#     default: ['#eeeeee', '#d6e685', '#8cc665', '#44a340', '#1e6823'],
+#     halloween: ['#EEEEEE', '#FFEE4A', '#FFC501', '#FE9600', '#03001C'],
+#     teal: ['#EEEEEE', "#7FFFD4", "#76EEC6", "#66CDAA", "#458B74"],
+#     red: ['#EEEEEE', '#990100', '#8C0100', '#720100', '#4C0100']
+# }
+
+# Amount should be a decimal between 0 and 1. Lower means darker
+def darken_color(hex_color, amount=0.4)
+  hex_color = hex_color.gsub('#','')
+  rgb = hex_color.scan(/../).map {|color| color.hex}
+  rgb[0] = (rgb[0].to_i * amount).round
+  rgb[1] = (rgb[1].to_i * amount).round
+  rgb[2] = (rgb[2].to_i * amount).round
+  "#%02x%02x%02x" % rgb
+end
+  
+# Amount should be a decimal between 0 and 1. Higher means lighter
+def lighten_color(hex_color, amount=0.6)
+  hex_color = hex_color.gsub('#','')
+  rgb = hex_color.scan(/../).map {|color| color.hex}
+  rgb[0] = [(rgb[0].to_i + 255 * amount).round, 255].min
+  rgb[1] = [(rgb[1].to_i + 255 * amount).round, 255].min
+  rgb[2] = [(rgb[2].to_i + 255 * amount).round, 255].min
+  "#%02x%02x%02x" % rgb
+end