import { AnnotationQuery, AnnotationSupport } from '@grafana/data'; import { InfluxQuery, InfluxQueryTag } from './types'; type LegacyAnnotation = { query?: string; queryType?: string; fromAnnotations?: boolean; tagsColumn?: string; textColumn?: string; timeEndColumn?: string; titleColumn?: string; name?: string; target?: { limit?: string | number | undefined; matchAny?: boolean; tags?: InfluxQueryTag[]; type?: string; }; }; // this becomes the target in the migrated annotations const migrateLegacyAnnotation = (json: LegacyAnnotation) => { // eslint-ignore-next-line const target: InfluxQuery = { refId: '', query: json.query ?? '', queryType: 'tags', fromAnnotations: true, tagsColumn: json.tagsColumn ?? '', textColumn: json.textColumn ?? '', timeEndColumn: json.timeEndColumn ?? '', titleColumn: json.titleColumn ?? '', name: json.name ?? '', }; // handle json target fields if (json.target && json.target.limit) { target.limit = json.target.limit; } if (json.target && json.target.matchAny) { target.matchAny = json.target.matchAny; } if (json.target && json.target.tags) { target.tags = json.target.tags; } if (json.target && json.target.type) { target.type = json.target.type; } return target; }; export const prepareAnnotation: AnnotationSupport<InfluxQuery, AnnotationQuery<InfluxQuery>>['prepareAnnotation'] = ( json ) => { // make sure that any additional target fields are migrated json.target = json.target && !json.target?.query ? migrateLegacyAnnotation(json) : json.target; return json; };