adds a sessions storage system

app/common.rb

@@ -15,7 +15,6 @@ class Common < Sinatra::Base
       set :config, Hieracles::Config.new({ config: configfile })
       enable :session
       enable :logging
-      set :store, Hieraviz::Store.new
     end
 
   end

app/config/hieraviz.default.yml

@@ -3,6 +3,7 @@ app_name: HieraViz
 basepath: "../puppet"
 classpath: "farm_modules/%s/manifests/init.pp"
 hierafile: "dev/hiera-local.yaml"
+tmpdir: "/tmp"
 session_seed: "xxx"
 usedb: true
 puppetdb:

app/views/data.erb

@@ -0,0 +1,4 @@
+<div class="meat text">
+<h1>Raw Data</h1>
+<pre><%= @data %></pre>
+</div>

app/views/home.erb

@@ -1,5 +1,5 @@
 <div class="meat text">
 Welcome to hieraviz<br>
 <%= session['access_token'] %><br>
-<pre><%= settings.store.get session['access_token'] if session['access_token'] %></pre>
+<pre><%= Hieraviz::Store.get session['access_token'] if session['access_token'] %></pre>
 </div>

app/views/store.erb

@@ -1,5 +1,6 @@
 <div class="meat text">
 <h1>Store</h1>
 <%= session['access_token'] %><br>
-<pre><%= settings.store.dump %></pre>
+<pre><%= Hieraviz::Store.dump %></pre>
+<pre><%= Hieraviz::Store.tmpdir %></pre>
 </div>

app/web.rb

@@ -2,7 +2,6 @@
 require 'sinatra/flash'
 
 require 'better_errors'
-require 'digest/sha1'
 require 'dotenv'
 require 'oauth2'
 
@@ -20,7 +19,6 @@ class Web < Common
       set :session_secret, settings.configdata['session_seed']
       set :public_folder, Proc.new { File.join(root, "public") }
       set :views_folder, Proc.new { File.join(root, "views") }
-      set :store, Hieraviz::Store.new
       set :erb, layout: :_layout
       enable :sessions
     end
@@ -70,7 +68,7 @@ def check_authorization
       get '/logged-in' do
         access_token = settings.oauth.access_token(request, params[:code])
         session[:access_token] = access_token.token
-        settings.store.set access_token.token, settings.oauth.user_info(access_token.token)
+        Hieraviz::Store.set access_token.token, settings.oauth.user_info(access_token.token)
         flash['info'] = "Successfully authenticated with the server"
         redirect '/'
       end
@@ -111,9 +109,19 @@ def check_authorization
     end
 
     get '/store' do
+      # Hieraviz::Store.set 'woot', 'nada'
       erb :store
     end
 
+    get '/user' do
+      if session[:access_token]
+        @data = settings.oauth.user_info(session[:access_token])
+      else
+        @data = 'nada'
+      end
+      erb :data
+    end
+
     not_found do
       erb :not_found, layout: :_layout
     end

lib/hieraviz/store.rb

@@ -1,20 +1,45 @@
 module Hieraviz
-  class Store
+  module Store
+    extend self
 
-    def initialize
-      @data = {}
+    def data
+      @_data ||= {}
     end
 
     def set(key, value)
-      @data[key.to_sym] = value
+      File.open(tmpfile(key), 'w') do |f|
+        f.print Marshal::dump(value)
+      end
+      data[key] = value
     end
 
     def get(key)
-      @data[key.to_sym]
+      data[key] ||= Marshal::load(File.read(tmpfile(key)).chomp)
     end
 
     def dump
-      @data
+      data
+    end
+
+    def tmpfile(name)
+      File.join tmpdir, name.gsub(/[^a-z0-9]/,'')
+    end
+
+    def tmpdir
+      @_tmpdir ||= init_tmpdir 
+    end
+
+    def init_tmpdir
+      configfile = ENV['HIERAVIZ_CONFIG_FILE'] || File.join("config", "hieraviz.yml")
+      configfile = File.expand_path(File.join('../../../app', configfile), __FILE__) unless configfile[0] == '/'
+      config = YAML.load_file(configfile)
+      tmp = config['tmpdir'] || '/tmp'
+      begin
+        FileUtils.mkdir_p(tmp) unless Dir.exist?(tmp)
+      rescue Exception => e
+        tmp = '/tmp'
+      end
+      tmp
     end
 
   end