grafana/public/app/features/scenes/variables/SceneVariableSet.ts

import { variableRegex } from 'app/features/variables/utils';

import { SceneObjectBase } from '../core/SceneObjectBase';
import { SceneObject } from '../core/types';

import { SceneVariable, SceneVariables, SceneVariableSetState, SceneVariableState } from './types';

export class TextBoxSceneVariable extends SceneObjectBase<SceneVariableState> implements SceneVariable {}

export class SceneVariableSet extends SceneObjectBase<SceneVariableSetState> implements SceneVariables {
  getVariableByName(name: string): SceneVariable | undefined {
    // TODO: Replace with index
    return this.state.variables.find((x) => x.state.name === name);
  }
}

export function sceneTemplateInterpolator(target: string, sceneObject: SceneObject) {
  variableRegex.lastIndex = 0;

  return target.replace(variableRegex, (match, var1, var2, fmt2, var3, fieldPath, fmt3) => {
    const variableName = var1 || var2 || var3;
    const variable = lookupSceneVariable(variableName, sceneObject);

    if (!variable) {
      return match;
    }

    return variable.state.current.value;
  });
}

function lookupSceneVariable(name: string, sceneObject: SceneObject): SceneVariable | null | undefined {
  const variables = sceneObject.state.$variables;
  if (!variables) {
    if (sceneObject.parent) {
      return lookupSceneVariable(name, sceneObject.parent);
    } else {
      return null;
    }
  }

  const found = variables.getVariableByName(name);
  if (found) {
    return found;
  } else if (sceneObject.parent) {
    return lookupSceneVariable(name, sceneObject.parent);
  }

  return null;
}