3 files changed, 45 insertions, 0 deletions
diff --git a/app.rb b/app.rb
index 3564523..4101e10 100644
--- a/app.rb
+++ b/app.rb
@@ -29,4 +29,11 @@ class App < Sinatra::Base
   get '/auth-callback' do
     erb :auth_callback
   end
+  
+  get '/initialise-labels' do
+    contextio = FancyBear::ContextIO::Label.new('sellevate.hack@gmail.com')
+    contextio.create_defaults
+    
+    redirect to('/')
+  end
 end
diff --git a/fancy_bear/contextio.rb b/fancy_bear/contextio.rb
index dc3f9ff..a7198e1 100644
--- a/fancy_bear/contextio.rb
+++ b/fancy_bear/contextio.rb
@@ -1,6 +1,7 @@
 require_relative 'contextio/config'
 require_relative 'contextio/base'
 require_relative 'contextio/auth'
+require_relative 'contextio/label'
 
 module FancyBear
   module ContextIO
diff --git a/fancy_bear/contextio/label.rb b/fancy_bear/contextio/label.rb
new file mode 100644
index 0000000..fb5a33a
--- /dev/null
+++ b/fancy_bear/contextio/label.rb
@@ -0,0 +1,37 @@
+require 'contextio'
+
+module FancyBear
+  module ContextIO
+    
+    class Label < Base
+      def initialize(email)
+        super()
+        
+        @account = @contextio.accounts.where(:email => email).first
+        @labels = []
+        @defaults = ['[FancyBear]/Low', '[FancyBear]/Medium', '[FancyBear]/High']
+      end
+      
+      def all
+        return if not @labels.empty?
+        
+        @account.sources.first.folders.each do |f|
+          @labels << f.name
+        end
+        @labels
+      end
+      
+      def create_defaults
+        all()
+        @labels.each do |l|
+          return if @defaults.include?(l)
+        end
+        
+        @defaults.each do |l|
+          system "contextio-put-folder.py '#{@account.id}' '#{l}'"
+        end
+      end
+    end
+    
+  end
+end