Contents

require 'jwt'
require 'pathname'
require 'forwardable'

module Terminalwire::Rails
  class Session
    # JWT file name for the session file.
    FILENAME = "session.jwt"

    # Empty dictionary the user can stash all their session data into.
    EMPTY_SESSION = {}.freeze

    extend Forwardable

    # Delegate `dig` and `fetch` to the `read` method
    def_delegators :read,
      :dig, :fetch, :[]

    def initialize(context:, path: nil, secret_key: self.class.secret_key)
      @context = context
      @path = path || context.storage_path
      @config_file_path = @path.join(FILENAME)
      @secret_key = secret_key

      ensure_file
    end

    def read
      jwt_token = @context.file.read(@config_file_path)
      decoded_data = JWT.decode(jwt_token, @secret_key, true, algorithm: 'HS256')
      decoded_data[0]  # JWT payload is the first element in the array
    rescue JWT::DecodeError => e
      raise "Invalid or tampered file: #{e.message}"
    end

    def reset
      @context.file.delete @config_file_path
    end

    def edit
      config = read
      yield config
      write(config)
    end

    def []=(key, value)
      edit { |config| config[key] = value }
    end

    def write(config)
      token = JWT.encode(config, @secret_key, 'HS256')
      @context.file.write(@config_file_path, token)
    end

    private

    def ensure_file
      return true if @context.file.exist? @config_file_path
      # Create the path if it doesn't exist on the client.
      @context.file.mkdir(@path) unless @context.file.exist?(@path)
      # Write an empty configuration on initialization
      write(EMPTY_SESSION)
    end

    def self.secret_key
      Rails.application.secret_key_base
    end
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
terminalwire-0.1.8	lib/terminalwire/rails.rb
terminalwire-0.1.7	lib/terminalwire/rails.rb
terminalwire-0.1.6	lib/terminalwire/rails.rb
terminalwire-0.1.5	lib/terminalwire/rails.rb
terminalwire-0.1.4	lib/terminalwire/rails.rb
terminalwire-0.1.3	lib/terminalwire/rails.rb
terminalwire-0.1.2	lib/terminalwire/rails.rb