From 8a229d56d16d5d5cdbfa646b8f6728edff3b458b Mon Sep 17 00:00:00 2001
From: Francesco Torchia <francesco.torchia@suse.com>
Date: Sun, 8 Sep 2024 18:05:03 +0200
Subject: [PATCH] Split passthrough to usb and pci components

Signed-off-by: Francesco Torchia <francesco.torchia@suse.com>
---
 ...ssthrough.vue => EnablePciPassthrough.vue} |   2 +-
 pkg/harvester/dialog/EnableUSBPassthrough.vue | 114 ++++++++++++++++++
 .../devices.harvesterhci.io.pcidevice.js      |   2 +-
 .../devices.harvesterhci.io.usbdevice.js      |   2 +-
 4 files changed, 117 insertions(+), 3 deletions(-)
 rename pkg/harvester/dialog/{EnablePassthrough.vue => EnablePciPassthrough.vue} (98%)
 create mode 100644 pkg/harvester/dialog/EnableUSBPassthrough.vue

diff --git a/pkg/harvester/dialog/EnablePassthrough.vue b/pkg/harvester/dialog/EnablePciPassthrough.vue
similarity index 98%
rename from pkg/harvester/dialog/EnablePassthrough.vue
rename to pkg/harvester/dialog/EnablePciPassthrough.vue
index 35516dee..b57fee25 100644
--- a/pkg/harvester/dialog/EnablePassthrough.vue
+++ b/pkg/harvester/dialog/EnablePciPassthrough.vue
@@ -6,7 +6,7 @@ import { escapeHtml } from '@shell/utils/string';
 import { HCI } from '../types';
 
 export default {
-  name: 'HarvesterEnablePassthrough',
+  name: 'HarvesterEnablePciPassthrough',
 
   emits: ['close'],
 
diff --git a/pkg/harvester/dialog/EnableUSBPassthrough.vue b/pkg/harvester/dialog/EnableUSBPassthrough.vue
new file mode 100644
index 00000000..793e09c1
--- /dev/null
+++ b/pkg/harvester/dialog/EnableUSBPassthrough.vue
@@ -0,0 +1,114 @@
+<script>
+import { HCI } from '../types';
+import { mapGetters } from 'vuex';
+import { Card } from '@components/Card';
+import AsyncButton from '@shell/components/AsyncButton';
+import { escapeHtml } from '@shell/utils/string';
+
+export default {
+  name: 'HarvesterEnableUSBPassthrough',
+
+  components: {
+    AsyncButton,
+    Card,
+  },
+
+  props: {
+    resources: {
+      type:     Array,
+      required: true
+    }
+  },
+
+  data() {
+    return {};
+  },
+
+  computed: { ...mapGetters({ t: 'i18n/t' }) },
+
+  methods: {
+    close() {
+      this.$emit('close');
+    },
+
+    async save(buttonCb) {
+      // isSingleProduct == this is a standalone Harvester cluster
+      const isSingleProduct = this.$store.getters['isSingleProduct'];
+      let userName = 'admin';
+
+      // if this is imported Harvester, there may be users other than 'admin
+      if (!isSingleProduct) {
+        const user = this.$store.getters['auth/v3User'];
+
+        userName = user?.username || user?.id;
+      }
+
+      for (let i = 0; i < this.resources.length; i++) {
+        const actionResource = this.resources[i];
+        const inStore = this.$store.getters['currentProduct'].inStore;
+        const pt = await this.$store.dispatch(`${ inStore }/create`, {
+          type:     HCI.USB_CLAIM,
+          metadata: {
+            name:            actionResource.metadata.name,
+            ownerReferences: [{
+              apiVersion: 'devices.harvesterhci.io/v1beta1',
+              kind:       'USBDevice',
+              name:       actionResource.metadata.name,
+              uid:        actionResource.metadata.uid,
+            }]
+          },
+          spec: {
+            pciAddress:  actionResource.status.pciAddress,
+            nodeName:    actionResource.status.nodeName,
+            userName
+          }
+        } );
+
+        try {
+          await pt.save();
+          buttonCb(true);
+          this.close();
+        } catch (err) {
+          this.$store.dispatch('growl/fromError', {
+            title: this.t('harvester.usb.claimError', { name: escapeHtml(actionResource.metadata.name) }),
+            err,
+          }, { root: true });
+          buttonCb(false);
+        }
+      }
+    }
+  }
+};
+</script>
+
+<template>
+  <Card :show-highlight-border="false">
+    <h4
+      slot="title"
+      v-clean-html="t('promptRemove.title')"
+      class="text-default-text"
+    />
+
+    <div slot="actions" class="actions">
+      <div class="buttons">
+        <button class="btn role-secondary mr-10" @click="close">
+          {{ t('generic.cancel') }}
+        </button>
+
+        <AsyncButton mode="enable" @click="save" />
+      </div>
+    </div>
+  </Card>
+</template>
+
+<style lang="scss" scoped>
+.actions {
+  width: 100%;
+}
+
+.buttons {
+  display: flex;
+  justify-content: flex-end;
+  width: 100%;
+}
+</style>
diff --git a/pkg/harvester/models/devices.harvesterhci.io.pcidevice.js b/pkg/harvester/models/devices.harvesterhci.io.pcidevice.js
index 2cbbac84..e337b92e 100644
--- a/pkg/harvester/models/devices.harvesterhci.io.pcidevice.js
+++ b/pkg/harvester/models/devices.harvesterhci.io.pcidevice.js
@@ -135,7 +135,7 @@ export default class PCIDevice extends SteveModel {
   enablePassthroughBulk(resources = this) {
     this.$dispatch('promptModal', {
       resources,
-      component: 'EnablePassthrough'
+      component: 'EnablePciPassthrough'
     });
   }
 
diff --git a/pkg/harvester/models/devices.harvesterhci.io.usbdevice.js b/pkg/harvester/models/devices.harvesterhci.io.usbdevice.js
index 38d18164..88b7c054 100644
--- a/pkg/harvester/models/devices.harvesterhci.io.usbdevice.js
+++ b/pkg/harvester/models/devices.harvesterhci.io.usbdevice.js
@@ -135,7 +135,7 @@ export default class USBDevice extends SteveModel {
   enablePassthroughBulk(resources = this) {
     this.$dispatch('promptModal', {
       resources,
-      component: 'EnablePassthrough'
+      component: 'EnableUSBPassthrough'
     });
   }