# -*- coding: utf-8 -*- require "magellan/cli/resources" module Magellan module Cli module Resources class Worker < Base self.resource_name = "functions~worker" self.resource_dependency = {"stage" => "stage-version"} self.hidden_fields = %w[created_at updated_at].map(&:freeze).freeze self.multiline_fields = %w[migration_command_1 migration_command_2 run_command environment_vars_yaml].map(&:freeze).freeze self.field_associations = {"stage_version_id" => {name: "stage", resource: "stage~version"} } desc "create NAME, IMAGE", "create WorkerVersion with NAME, IMAGE" method_option :attributes_yaml, aliases: "-A", desc: "path to YAML file which defines attributes" def create(name, image_name) s = load_selection("stage-version") attrs = if path = options[:attributes_yaml] YAML.load_file(path) else {} end params = { "functions_worker" => { "stage_version_id" => s["id"], "name" => name, "image_name" => image_name, # "instance_base_name" => name, }.update(attrs) } post_json("/admin/#{self.resource_name}/new.json", params) end desc "update ATTRIBUTE", "update selected worker with ATTRIBUTE(filename or JSON)" def update(attrs) if File.readable?(attrs) attrs = YAML.load_file(attrs) else attrs = JSON.parse(attrs) end w = load_selection(parameter_name) self.class.hidden_fields.each do |f| attrs.delete(f) end self.class.field_associations.keys.each do |f| attrs.delete(f) end params = { parameter_name => attrs } put_json("/admin/#{resource_name}/#{w["id"]}/edit.js", params) end end end end end