File: environment.rb

package info (click to toggle)
ruby-nenv 0.3.0-5
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 184 kB
  • sloc: ruby: 517; sh: 4; makefile: 2
file content (81 lines) | stat: -rw-r--r-- 1,930 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
require 'nenv/environment/dumper'
require 'nenv/environment/loader'

module Nenv
  class Environment
    class Error < ArgumentError
    end

    class MethodError < Error
      def initialize(meth)
        @meth = meth
      end
    end

    class AlreadyExistsError < MethodError
      def message
        format('Method %s already exists', @meth.inspect)
      end
    end

    def initialize(namespace = nil)
      @namespace = (namespace ? namespace.upcase : nil)
    end

    def create_method(meth, &block)
      self.class._create_env_accessor(singleton_class, meth, &block)
    end

    private

    def _sanitize(meth)
      meth.to_s[/^([^=?]*)[=?]?$/, 1].upcase
    end

    def _namespaced_sanitize(meth)
      [@namespace, _sanitize(meth)].compact.join('_')
    end

    class << self
      def create_method(meth, &block)
        _create_env_accessor(self, meth, &block)
      end

      def _create_env_accessor(klass, meth, &block)
        _fail_if_accessor_exists(klass, meth)

        if meth.to_s.end_with? '='
          _create_env_writer(klass, meth, &block)
        else
          _create_env_reader(klass, meth, &block)
        end
      end

      private

      def _create_env_writer(klass, meth, &block)
        env_name = nil
        dumper = nil
        klass.send(:define_method, meth) do |raw_value|
          env_name ||= _namespaced_sanitize(meth)
          dumper ||= Dumper.setup(&block)
          ENV[env_name] = dumper.(raw_value)
        end
      end

      def _create_env_reader(klass, meth, &block)
        env_name = nil
        loader = nil
        klass.send(:define_method, meth) do
          env_name ||= _namespaced_sanitize(meth)
          loader ||= Loader.setup(meth, &block)
          loader.(ENV[env_name])
        end
      end

      def _fail_if_accessor_exists(klass, meth)
        fail(AlreadyExistsError, meth) if klass.method_defined?(meth)
      end
    end
  end
end