X-Git-Url: https://git.arvados.org/arvados-formula.git/blobdiff_plain/1c81e527e7a3b3a965958e33d7073135071b3ff3..b6fcdedd0d3233a0d1bc8b82be44090002b6e7df:/arvados/controller/resources/virtual_machines.sls

diff --git a/arvados/controller/resources/virtual_machines.sls b/arvados/controller/resources/virtual_machines.sls
index 7303dac..e0917c0 100644
--- a/arvados/controller/resources/virtual_machines.sls
+++ b/arvados/controller/resources/virtual_machines.sls
@@ -1,6 +1,10 @@
 # -*- coding: utf-8 -*-
 # vim: ft=sls
 
+# Copyright (C) The Arvados Authors. All rights reserved.
+#
+# SPDX-License-Identifier: Apache-2.0
+
 {#- Get the `tplroot` from `tpldir` #}
 {%- set tplroot = tpldir.split('/')[0] %}
 {%- set sls_config_file = tplroot ~ '.config.file' %}
@@ -11,6 +15,9 @@
 {%- set api_token = arvados.cluster.tokens.system_root | yaml_encode %}
 {%- set api_host = arvados.cluster.Services.Controller.ExternalURL | regex_replace('^http(s?)://', '', ignorecase=true) %}
 
+{%- set arv_command = '/usr/local/rvm/bin/rvm-exec default arv'
+                      if arvados.ruby.manage_ruby and arvados.ruby.use_rvm
+                      else 'arv' %}
 include:
   - ..package
   - {{ sls_config_file }}
@@ -24,8 +31,8 @@ arvados-controller-resources-virtual-machines-jq-pkg-installed:
   {%- set vm_name = vm_params.name | default(vm) %}
 
   {%- set cmd_query_vm_uuid = 'ARVADOS_API_TOKEN=' ~ api_token ~
-                              ' ARVADOS_API_HOST=' ~ api_host ~
-                              ' arv --short virtual_machine list' ~
+                              ' ARVADOS_API_HOST=' ~ api_host ~ ' ' ~
+                              arv_command ~ ' --short virtual_machine list' ~
                               ' --filters \'[["hostname", "=", "' ~ vm_name ~ '"]]\''
   %}
 
@@ -36,16 +43,17 @@ arvados-controller-resources-virtual-machines-{{ vm }}-record-cmd-run:
       - ARVADOS_API_TOKEN: {{ api_token }}
       - ARVADOS_API_HOST: {{ api_host }}
     - name: |
-        arv --format=uuid \
+        {{ arv_command }} --format=uuid \
           virtual_machine \
           create \
           --virtual-machine '{"hostname":"{{ vm_name }}" }'
     - unless: |
           {{ cmd_query_vm_uuid }} | \
-          /bin/grep -qE "fixme-2x53u-[a-z0-9]{15}"
+          /bin/grep -qE "[a-z0-9]{5}-2x53u-[a-z0-9]{15}"
     - require:
       - pkg: arvados-controller-package-install-pkg-installed
       - cmd: arvados-controller-service-running-service-ready-cmd-run
+      - gem: arvados-controller-package-install-gem-arvados-cli-installed
 
 # We need to use the UUID generated in the previous command to see if there's a
 # scoped token for it. There's no easy way to pass the value from a shellout
@@ -54,18 +62,22 @@ arvados-controller-resources-virtual-machines-{{ vm }}-record-cmd-run:
 # the file content is a token uuid :|
 arvados-controller-resources-virtual-machines-{{ vm }}-get-vm_uuid-cmd-run:
   cmd.run:
+    {%- if arvados.ruby.manage_ruby and arvados.ruby.use_rvm %}
+    - prepend_path: /usr/local/rvm/gems/{{ arvados.ruby.pkg }}/bin
+    {%- endif %}
     - name: {{ cmd_query_vm_uuid }} | head -1 | tee /tmp/{{ vm }}
     - require:
       - cmd: arvados-controller-resources-virtual-machines-{{ vm }}-record-cmd-run
+      - gem: arvados-controller-package-install-gem-arvados-cli-installed
     - unless:
-      - /bin/grep -qE "fixme-2x53u-[a-z0-9]{15}" /tmp/{{ vm }}
+      - /bin/grep -qE "[a-z0-9]{5}-2x53u-[a-z0-9]{15}" /tmp/{{ vm }}
 
   # There's no direct way to query the scoped_token for a given virtual_machine
   # so we need to parse the api_client_authorization list through some jq
   {%- set cmd_query_scoped_token_url = 'VM_UUID=$(cat /tmp/' ~ vm ~ ') && ' ~
                                        ' ARVADOS_API_TOKEN=' ~ api_token ~
-                                       ' ARVADOS_API_HOST=' ~ api_host ~
-                                       ' arv api_client_authorization list |' ~
+                                       ' ARVADOS_API_HOST=' ~ api_host ~ ' ' ~
+                                       arv_command ~ ' api_client_authorization list |' ~
                                        ' /usr/bin/jq -e \'.items[].scopes[] | select(. == "GET ' ~
                                        '/arvados/v1/virtual_machines/\'${VM_UUID}\'/logins")\' && ' ~
                                        'unset VM_UUID'
@@ -79,7 +91,7 @@ arvados-controller-resources-virtual-machines-{{ vm }}-scoped-token-cmd-run:
       - ARVADOS_API_HOST: {{ api_host }}
     - name: |
         VM_UUID=$(cat /tmp/{{ vm }}) &&
-        arv --format=uuid \
+        {{ arv_command }} --format=uuid \
           api_client_authorization \
           create \
           --api-client-authorization '{"scopes":["GET /arvados/v1/virtual_machines/'${VM_UUID}'/logins"]}'
@@ -88,5 +100,6 @@ arvados-controller-resources-virtual-machines-{{ vm }}-scoped-token-cmd-run:
       - pkg: arvados-controller-package-install-pkg-installed
       - pkg: arvados-controller-resources-virtual-machines-jq-pkg-installed
       - cmd: arvados-controller-resources-virtual-machines-{{ vm }}-get-vm_uuid-cmd-run
+      - gem: arvados-controller-package-install-gem-arvados-cli-installed
 
 {%- endfor %}