mirror of
https://github.com/ToolJet/ToolJet
synced 2026-05-22 16:38:21 +00:00
b10d777f63
* Updated project set up guide for Mac, added node version and Webpack install steps. * Worked on PR comment i.e Can we change this line to Install Node.js ( version: v14.9.0 ) * Fixed "Style/StringLiterals: Prefer double-quoted strings unless you need single quotes to avoid extra backslashes for escaping." * Resolved rubocop comments i.e "Style/FrozenStringLiteralComment: Missing frozen string literal comment." in test folder Co-authored-by: Deepti Kakade <deepti@saeloun.com> Co-authored-by: Deepti Kakade <“deepti@saeloun.com”>
130 lines
3.2 KiB
Ruby
130 lines
3.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class FirestoreQueryService
|
|
require "google/cloud/firestore"
|
|
include DatasourceUtils
|
|
|
|
attr_accessor :data_query, :options, :source_options, :current_user, :data_source
|
|
|
|
def initialize(data_query, data_source, options, source_options, current_user)
|
|
@data_query = data_query
|
|
@data_source = data_source
|
|
@options = options
|
|
@source_options = source_options
|
|
@current_user = current_user
|
|
end
|
|
|
|
def self.connection(options)
|
|
gcp_key = JSON.parse(options.dig("gcp_key", "value"))
|
|
|
|
Google::Cloud::Firestore.configure do |config|
|
|
config.credentials = gcp_key
|
|
end
|
|
|
|
firestore = Google::Cloud::Firestore.new
|
|
|
|
firestore.cols # Try to fetch collections
|
|
end
|
|
|
|
def process
|
|
data = {}
|
|
error = nil
|
|
|
|
begin
|
|
|
|
firestore = get_cached_connection(data_source)
|
|
firestore = create_connection unless firestore
|
|
|
|
operation = options["operation"]
|
|
|
|
update_document(options["path"], options["body"].as_json, firestore) if operation == "update_document"
|
|
|
|
if operation == "bulk_update"
|
|
records = options["records"]
|
|
collection = options["collection"]
|
|
doc_key_id = options["document_id_key"]
|
|
|
|
records.each do |record|
|
|
path = "#{collection}/#{record[doc_key_id]}"
|
|
record.delete(doc_key_id)
|
|
update_document(path, record.as_json, firestore)
|
|
end
|
|
end
|
|
|
|
if operation == "get_document"
|
|
path = options["path"]
|
|
doc_ref = firestore.doc path
|
|
snapshot = doc_ref.get
|
|
data = snapshot.data
|
|
end
|
|
|
|
if operation == "set_document"
|
|
path = options["path"]
|
|
body = JSON.parse(options["body"])
|
|
doc_ref = firestore.doc path
|
|
doc_ref.set body
|
|
end
|
|
|
|
if operation == "add_document"
|
|
path = options["path"]
|
|
body = JSON.parse(options["body"])
|
|
col_ref = firestore.col path
|
|
col_ref.add body
|
|
end
|
|
|
|
if operation == "delete_document"
|
|
path = options["path"]
|
|
body = JSON.parse(options["body"])
|
|
doc_ref = firestore.doc path
|
|
doc_ref.delete
|
|
end
|
|
|
|
if operation == "query_collection"
|
|
path = options["path"]
|
|
doc_ref = firestore.col path
|
|
|
|
# execute where condition
|
|
|
|
if options["where_field"]
|
|
doc_ref = doc_ref.where options["where_field"], options["where_operation"], options["where_value"]
|
|
end
|
|
|
|
if options["order"]
|
|
doc_ref = doc_ref.order(options["order"], "desc")
|
|
end
|
|
|
|
if options["limit"]
|
|
doc_ref = doc_ref.limit(options["limit"].to_i)
|
|
end
|
|
|
|
data = []
|
|
doc_ref.get do |doc|
|
|
data << { data: doc.data, document_id: doc.document_id }
|
|
end
|
|
end
|
|
rescue StandardError => e
|
|
puts e
|
|
error = e.message
|
|
end
|
|
|
|
{ data: data, error: error }
|
|
end
|
|
|
|
private
|
|
|
|
def update_document(path, body, firestore)
|
|
doc_ref = firestore.doc path
|
|
doc_ref.update body
|
|
end
|
|
|
|
def create_connection
|
|
credential_json = JSON.parse(source_options["gcp_key"])
|
|
Google::Cloud::Firestore.configure do |config|
|
|
config.credentials = credential_json
|
|
end
|
|
firestore = Google::Cloud::Firestore.new
|
|
|
|
cache_connection(data_source, firestore)
|
|
firestore
|
|
end
|
|
end
|