forked from e621ng/e621ng
47 lines
1.4 KiB
Ruby
47 lines
1.4 KiB
Ruby
class StateChecker
|
|
include Singleton
|
|
|
|
def check!
|
|
ENV["SECRET_TOKEN"].present? || check_secret_token
|
|
ENV["SESSION_SECRET_KEY"].present? || check_session_secret_key
|
|
end
|
|
|
|
def secret_token
|
|
ENV["SECRET_TOKEN"] || File.read(secret_token_path)
|
|
end
|
|
|
|
def session_secret_key
|
|
ENV["SESSION_SECRET_KEY"] || File.read(session_secret_key_path)
|
|
end
|
|
|
|
private
|
|
|
|
def secret_token_path
|
|
File.expand_path("~/.danbooru/secret_token")
|
|
end
|
|
|
|
def check_secret_token
|
|
unless File.exist?(secret_token_path)
|
|
raise "You must create a file in #{secret_token_path} containing a secret key. It should be a string of at least 32 random characters."
|
|
end
|
|
|
|
if File.stat(secret_token_path).world_readable? || File.stat(secret_token_path).world_writable?
|
|
raise "#{secret_token_path} must not be world readable or writable"
|
|
end
|
|
end
|
|
|
|
def session_secret_key_path
|
|
File.expand_path("~/.danbooru/session_secret_key")
|
|
end
|
|
|
|
def check_session_secret_key
|
|
unless File.exist?(session_secret_key_path)
|
|
raise "You must create a file in #{session_secret_key_path} containing a secret key. It should be a string of at least 32 random characters."
|
|
end
|
|
|
|
if File.stat(session_secret_key_path).world_readable? || File.stat(session_secret_key_path).world_writable?
|
|
raise "#{session_secret_key_path} must not be world readable or writable"
|
|
end
|
|
end
|
|
end
|