From 03e8906b09ff1085d9b8d4bf7e96cbb7ad2ba1c6 Mon Sep 17 00:00:00 2001
From: r888888888 <r888888888@gmail.com>
Date: Thu, 6 Mar 2014 18:18:18 -0800
Subject: [PATCH] fixes #1366

---
 config/deploy.rb | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/config/deploy.rb b/config/deploy.rb
index 8be25577b..cb37b7251 100644
--- a/config/deploy.rb
+++ b/config/deploy.rb
@@ -14,6 +14,10 @@ set :whenever_command, "bundle exec whenever"
 set :whenever_environment, defer {stage}
 require 'whenever/capistrano'
 
+require 'securerandom'
+set :secret_1, SecureRandom.base64(32)
+set :secret_2, SecureRandom.base64(32)
+
 set :application, "danbooru"
 set :repository,  "git://github.com/r888888888/danbooru.git"
 set :scm, :git
@@ -30,6 +34,15 @@ namespace :local_config do
     run "mkdir -p #{deploy_to}/shared/config"
   end
 
+  desc "Initialize the secrets"
+  task :setup_secrets do
+    run "mkdir -p ~/.danbooru"
+    run "if [[ ! -e ~/.danbooru/session_secret_key ]] ; then echo '#{secret_1}' > ~/.danbooru/session_secret_key ; fi"
+    run "if [[ ! -e ~/.danbooru/secret_token ]] ; then echo '#{secret_2}' > ~/.danbooru/secret_token ; fi"
+    run "chmod -R 600 ~/.danbooru"
+    run "chown -R #{user}:#{user} ~/.danbooru"
+  end
+
   desc "Initialize local config files"
   task :setup_local_files do
     run "curl -s https://raw.github.com/r888888888/danbooru/master/script/install/danbooru_local_config.rb.templ > #{deploy_to}/shared/config/danbooru_local_config.rb"
@@ -138,6 +151,7 @@ after "deploy:setup", "reset_ownership_of_common_directory"
 after "deploy:setup", "local_config:setup_shared_directory"
 after "deploy:setup", "local_config:setup_local_files"
 after "deploy:setup", "data:setup_directories"
+after "deploy:setup", "local_config:setup_secrets"
 after "deploy:create_symlink", "local_config:link_local_files"
 after "deploy:create_symlink", "data:link_directories"
 after "deploy:start", "delayed_job:start"