Module: Rodauth::OAuth

Defined in:
lib/rodauth/oauth.rb,
lib/rodauth/oauth/railtie.rb,
lib/rodauth/oauth/version.rb,
lib/rodauth/oauth/http_extensions.rb,
lib/rodauth/oauth/database_extensions.rb,
lib/generators/rodauth/oauth/views_generator.rb

Defined Under Namespace

Modules: ControllerMethods, FeatureExtensions, HTTPExtensions, Rails Classes: Railtie, TtlStore

Constant Summary collapse

VERSION = 
"1.3.0"

Class Method Summary collapse

Class Method Details

.ExtendDatabase(db) ⇒ Object

rubocop:disable Naming/MethodName



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
82
83
84
85
 
# File 'lib/rodauth/oauth/database_extensions.rb', line 6

def self.ExtendDatabase(db)
  Module.new do
    dataset = db.dataset

    if dataset.supports_returning?(:insert)
      def __insert_and_return__(dataset, _pkey, params)
        dataset.returning.insert(params).first
      end
    else
      def __insert_and_return__(dataset, pkey, params)
        id = dataset.insert(params)
        dataset.where(pkey => id).first
      end
    end

    if dataset.supports_returning?(:update)
      def __update_and_return__(dataset, params)
        dataset.returning.update(params).first
      end
    else
      def __update_and_return__(dataset, params)
        dataset.update(params)
        dataset.first
      end
    end

    if dataset.respond_to?(:supports_insert_conflict?) && dataset.supports_insert_conflict?
      def __insert_or_update_and_return__(dataset, pkey, unique_columns, params, conds = nil, to_update_extra = nil)
        to_update = Hash[(params.keys - unique_columns).map { |attribute| [attribute, Sequel[:excluded][attribute]] }]

        to_update.merge!(to_update_extra) if to_update_extra

        dataset = dataset.insert_conflict(
          target: unique_columns,
          update: to_update,
          update_where: conds
        )

        __insert_and_return__(dataset, pkey, params)
      end

      def __insert_or_do_nothing_and_return__(dataset, pkey, unique_columns, params)
        __insert_and_return__(
          dataset.insert_conflict(target: unique_columns),
          pkey,
          params
        ) || dataset.where(params).first
      end
    else
      def __insert_or_update_and_return__(dataset, pkey, unique_columns, params, conds = nil, to_update_extra = nil)
        find_params, update_params = params.partition { |key, _| unique_columns.include?(key) }.map { |h| Hash[h] }

        dataset_where = dataset.where(find_params)
        record = if conds
                   dataset_where_conds = dataset_where.where(conds)

                   # this means that there's still a valid entry there, so return early
                   return if dataset_where.count != dataset_where_conds.count

                   dataset_where_conds.first
                 else
                   dataset_where.first
                 end

        if record
          update_params.merge!(to_update_extra) if to_update_extra

          __update_and_return__(dataset_where, update_params)
        else
          __insert_and_return__(dataset, pkey, params)
        end
      end

      def __insert_or_do_nothing_and_return__(dataset, pkey, unique_columns, params)
        find_params = params.select { |key, _| unique_columns.include?(key) }
        dataset.where(find_params).first || __insert_and_return__(dataset, pkey, params)
      end
    end
  end
end