add tenant information as a query parameter to url in Run in vmui button (#485)

Author: arturminchukov

CHANGELOG.md

@@ -4,6 +4,7 @@
 
 * BUGFIX: fix an issue where the `Internal link` from derived fields config was disabled. See [#479](https://github.com/VictoriaMetrics/victorialogs-datasource/pull/479).
 * BUGFIX: fix an issue where the `show context` failed if disable labels with transformations. See [#480](https://github.com/VictoriaMetrics/victorialogs-datasource/pull/480).
+* BUGFIX: add tenant information as a query parameter to `url` in `Run in vmui` button. See [#480](https://github.com/VictoriaMetrics/victorialogs-datasource/pull/480).
 
 ## v0.22.3
 

pkg/plugin/datasource.go

@@ -618,10 +618,21 @@ func (d *Datasource) VMUIQuery(rw http.ResponseWriter, req *http.Request) {
 		return
 	}
 
+	accountID := di.grafanaSettings.MultitenancyHeaders["AccountID"]
+	if accountID == "" {
+		accountID = "0"
+	}
+
+	projectID := di.grafanaSettings.MultitenancyHeaders["ProjectID"]
+	if projectID == "" {
+		projectID = "0"
+	}
+
 	rw.Header().Add("Content-Type", "application/json")
 	rw.WriteHeader(http.StatusOK)
 
-	if _, err := fmt.Fprintf(rw, `{"vmuiURL": %q}`, vmuiUrl); err != nil {
+	if _, err := fmt.Fprintf(rw, `{"vmuiURL": %q, "accountID": %q, "projectID": %q}`,
+		vmuiUrl, accountID, projectID); err != nil {
 		d.logger.Warn("Error writing response", "error", err)
 	}
 }

src/components/QueryEditor/VmuiLink.tsx

@@ -22,7 +22,7 @@ import { VictoriaLogsDatasource } from "../../datasource";
 import { Query } from "../../types";
 import { getDurationFromMilliseconds } from "../../utils/timeUtils";
 
-export const getTimeUrlParams = (panelData?: PanelData) => {
+const getTimeUrlParams = (panelData?: PanelData) => {
   const timeRange = panelData?.timeRange || getDefaultTimeRange();
   const rangeRaw = timeRange.raw;
   let relativeTimeId = 'none';
@@ -84,12 +84,23 @@ export const relativeTimeOptionsVMUI = [
   ...o
 }))
 
+type Tenant = {
+  projectID: string;
+  accountID: string;
+}
+
+const DEFAULT_TENANT: Tenant = {
+  projectID: '0',
+  accountID: '0',
+};
+
 const VmuiLink: FC<Props> = ({
   panelData,
   query,
   datasource,
 }) => {
   const [baseVmuiUrl, setBaseVmuiUrl] = useState('');
+  const [tenant, setTenant] = useState<Tenant>(DEFAULT_TENANT);
 
   useEffect(() => {
     if (!datasource) {
@@ -98,8 +109,9 @@ const VmuiLink: FC<Props> = ({
 
     const fetchVmuiUrl = async () => {
       try {
-        const resp = await datasource.getResource<{ vmuiURL: string }>('vmui');
+        const resp = await datasource.getResource<Tenant & { vmuiURL: string }>('vmui');
         setBaseVmuiUrl(resp.vmuiURL.includes('#') ? resp.vmuiURL.split('/#')[0] : resp.vmuiURL);
+        setTenant({ projectID: resp.projectID, accountID: resp.accountID });
       } catch (error) {
         console.error('Error fetching VMUI URL:', error);
       }
@@ -114,10 +126,11 @@ const VmuiLink: FC<Props> = ({
 
     return `${baseVmuiUrl}/#/?` + new URLSearchParams({
       ...timeParams,
+      ...tenant,
       query: queryExpr,
       tab: '0',
     }).toString();
-  }, [baseVmuiUrl, datasource, panelData, query.expr]);
+  }, [baseVmuiUrl, datasource, panelData, query.expr, tenant]);
 
   return (
     <a href={textUtil.sanitizeUrl(href)} target="_blank" rel="noopener noreferrer"