mirror of
https://github.com/ToolJet/ToolJet
synced 2026-05-12 19:48:36 +00:00
ccadcb2e9d
* Improves ruby code in app/models/app_user.rb based on suggestions from Rubocop * Rubocop fixes for more models * Rubocop fixes for controllers
133 lines
3.8 KiB
Ruby
133 lines
3.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class DataSourcesController < ApplicationController
|
|
def index
|
|
@data_sources = DataSource.where(app_id: params[:app_id])
|
|
end
|
|
|
|
def create
|
|
options = fetch_oauth_options(params[:options])
|
|
|
|
options_to_save = {}
|
|
options.each do |option|
|
|
if option["encrypted"]
|
|
credential = Credential.create(value: option["value"])
|
|
|
|
options_to_save[option["key"]] = {
|
|
credential_id: credential.id,
|
|
encrypted: option["encrypted"]
|
|
}
|
|
else
|
|
options_to_save[option["key"]] = {
|
|
value: option["value"],
|
|
encrypted: option["encrypted"]
|
|
}
|
|
end
|
|
end
|
|
|
|
@data_source = DataSource.create(
|
|
name: params[:name],
|
|
kind: params[:kind],
|
|
options: options_to_save,
|
|
app_id: params[:app_id]
|
|
)
|
|
end
|
|
|
|
def update
|
|
@data_source = DataSource.find params[:id]
|
|
|
|
options = fetch_oauth_options(params[:options])
|
|
|
|
options_to_save = {}
|
|
options.each do |option|
|
|
if option["encrypted"]
|
|
credential = Credential.create(value: option["value"])
|
|
|
|
options_to_save[option["key"]] = {
|
|
credential_id: credential.id,
|
|
encrypted: option["encrypted"]
|
|
}
|
|
else
|
|
options_to_save[option["key"]] = {
|
|
value: option["value"],
|
|
encrypted: option["encrypted"]
|
|
}
|
|
end
|
|
end
|
|
|
|
@data_source.update(
|
|
name: params[:name],
|
|
options: options_to_save
|
|
)
|
|
|
|
$connections.delete @data_source.id
|
|
end
|
|
|
|
def test_connection
|
|
service = DataSourceConnectionService.new params[:kind], params[:options]
|
|
service.process
|
|
render json: { status: 200 }
|
|
rescue StandardError => e
|
|
puts e
|
|
render json: { message: e }, status: :internal_server_error
|
|
end
|
|
|
|
def authorize_oauth2
|
|
data_source = DataSource.find params[:data_source_id]
|
|
options = CredentialService.new.decrypt_options(data_source.options)
|
|
access_token_url = options["access_token_url"]
|
|
|
|
custom_params = options["custom_auth_params"].to_h
|
|
|
|
response = HTTParty.post(access_token_url,
|
|
body: { code: params[:code],
|
|
client_id: options["client_id"],
|
|
client_secret: options["client_secret"],
|
|
grant_type: options["grant_type"],
|
|
redirect_uri: "#{ENV.fetch('TOOLJET_HOST')}/oauth2/authorize",
|
|
**custom_params }.to_json,
|
|
headers: { "Content-Type" => "application/json" })
|
|
|
|
result = JSON.parse(response.body)
|
|
access_token = result["access_token"]
|
|
|
|
options = { access_token: access_token }
|
|
|
|
DataSourceUserOauth2.create(
|
|
user: current_user,
|
|
data_source: data_source,
|
|
options: options.to_json.to_s
|
|
)
|
|
|
|
render json: { success: true }
|
|
end
|
|
|
|
def fetch_oauth2_base_url
|
|
provider = params[:provider]
|
|
service_class = "#{provider.capitalize}OauthService".constantize
|
|
url = service_class.generate_base_auth_url
|
|
|
|
render json: { url: url }
|
|
end
|
|
|
|
private
|
|
def fetch_oauth_options(options)
|
|
# Fetch necessary access token if OAuth2 based data source
|
|
if options.find { |option| option["key"] == "oauth2" }
|
|
provider = options.find { |option| option["key"] === "provider" } ["value"]
|
|
service_class = "#{provider.capitalize}OauthService".constantize
|
|
access_info = service_class.fetch_access_token(options.find { |option| option["key"] === "code" } ["value"])
|
|
options.reject! { |option| option["key"] == "code" }
|
|
|
|
access_info.each do |info|
|
|
option = {}
|
|
option["key"] = info[0]
|
|
option["value"] = info[1]
|
|
option["encrypted"] = true
|
|
options << option
|
|
end
|
|
end
|
|
|
|
options
|
|
end
|
|
end
|