sql.rb
No OneTemporary
Actions

Size

818 B

Subscribers

None

sql.rb
View Options

	#!/opt/puppetlabs/puppet/bin/ruby
	# frozen_string_literal: true

	require 'json'
	require 'open3'
	require 'puppet'

	def get(sql, database, user, password)
	cmd = ['mysql', '-e', "#{sql} "]
	cmd << "--database=#{database}" unless database.nil?
	cmd << "--user=#{user}" unless user.nil?
	cmd << "--password=#{password}" unless password.nil?
	stdout, stderr, status = Open3.capture3(*cmd)
	raise Puppet::Error, _("stderr: '%{stderr}'" % { stderr: stderr }) if status != 0
	{ status: stdout.strip }
	end

	params = JSON.parse(STDIN.read)
	database = params['database']
	user = params['user']
	password = params['password']
	sql = params['sql']

	begin
	result = get(sql, database, user, password)
	puts result.to_json
	exit 0
	rescue Puppet::Error => e
	puts({ status: 'failure', error: e.message }.to_json)
	exit 1
	end