zhandao/zero-rails_openapi

# frozen_string_literal: true

require 'active_support/concern'

module OpenApi
  module ConfigDSL
    extend ActiveSupport::Concern

    class_methods do
      def open_api name, base_doc_classes:
        @api = name
        open_api_docs[name] = { base_doc_classes: base_doc_classes }
      end

      def info version:, title:, desc: '', **addition
        open_api_docs[@api][:info] = { version: version, title: title, description: desc, **addition }
      end

      def server url, desc: ''
        (open_api_docs[@api][:servers] ||= [ ]) << { url: url, description: desc }
      end

      def security_scheme scheme_name, other_info# = { }
        other_info[:description] = other_info.delete(:desc) if other_info.key?(:desc)
        (open_api_docs[@api][:securitySchemes] ||= { })[scheme_name] = other_info
      end

      def base_auth scheme_name, other_info = { }
        security_scheme scheme_name, { type: 'http', scheme: 'basic', **other_info }
      end

      def bearer_auth scheme_name, format = 'JWT', other_info = { }
        security_scheme scheme_name, { type: 'http', scheme: 'bearer', bearerFormat: format, **other_info }
      end

      def api_key scheme_name, field:, in: 'header', **other_info
        _in = binding.local_variable_get(:in)
        security_scheme scheme_name, { type: 'apiKey', name: field, in: _in, **other_info }
      end

      def global_security_require scheme_name, scopes: [ ]
        (open_api_docs[@api][:global_security] ||= [ ]) << { scheme_name => scopes }
      end

      alias global_security global_security_require
      alias global_auth     global_security_require
      alias auth_scheme     security_scheme
    end
  end
end