1+ import { Configuration , WatchApi , EVENT_DATA } from '@gitlab/cluster-client' ;
12import axios from '~/lib/utils/axios_utils' ;
23import {
34 HELM_RELEASES_RESOURCE_TYPE ,
45 KUSTOMIZATIONS_RESOURCE_TYPE ,
56} from '~/environments/constants' ;
7+ import fluxKustomizationStatusQuery from '../queries/flux_kustomization_status.query.graphql' ;
8+ import fluxHelmReleaseStatusQuery from '../queries/flux_helm_release_status.query.graphql' ;
69
710const helmReleasesApiVersion = 'helm.toolkit.fluxcd.io/v2beta1' ;
811const kustomizationsApiVersion = 'kustomize.toolkit.fluxcd.io/v1beta1' ;
912
13+ const helmReleaseField = 'fluxHelmReleaseStatus' ;
14+ const kustomizationField = 'fluxKustomizationStatus' ;
15+
1016const handleClusterError = ( err ) => {
1117 const error = err ?. response ?. data ?. message ? new Error ( err . response . data . message ) : err ;
1218 throw error ;
@@ -22,14 +28,57 @@ const buildFluxResourceUrl = ({
2228 return `${ basePath } ${ apiVersion } ${ namespace } ${ resourceType } ${ environmentName } ;
2329} ;
2430
25- const getFluxResourceStatus = ( configuration , url ) => {
26- const { headers } = configuration ;
31+ const buildFluxResourceWatchPath = ( { namespace, apiVersion, resourceType } ) => {
32+ return `/apis/${ apiVersion } ${ namespace } ${ resourceType } ;
33+ } ;
34+
35+ const watchFluxResource = ( { watchPath, resourceName, query, variables, field, client } ) => {
36+ const config = new Configuration ( variables . configuration ) ;
37+ const watcherApi = new WatchApi ( config ) ;
38+ const fieldSelector = `metadata.name=${ decodeURIComponent ( resourceName ) } ;
39+
40+ watcherApi
41+ . subscribeToStream ( watchPath , { watch : true , fieldSelector } )
42+ . then ( ( watcher ) => {
43+ let result = [ ] ;
44+
45+ watcher . on ( EVENT_DATA , ( data ) => {
46+ result = data [ 0 ] ?. status ?. conditions ;
47+
48+ client . writeQuery ( {
49+ query,
50+ variables,
51+ data : { [ field ] : result } ,
52+ } ) ;
53+ } ) ;
54+ } )
55+ . catch ( ( err ) => {
56+ handleClusterError ( err ) ;
57+ } ) ;
58+ } ;
59+
60+ const getFluxResourceStatus = ( { url, watchPath, query, variables, field, client } ) => {
61+ const { headers } = variables . configuration ;
2762 const withCredentials = true ;
2863
2964 return axios
3065 . get ( url , { withCredentials, headers } )
3166 . then ( ( res ) => {
32- return res ?. data ?. status ?. conditions || [ ] ;
67+ const fluxData = res ?. data ;
68+ const resourceName = fluxData ?. metadata ?. name ;
69+
70+ if ( gon . features ?. k8sWatchApi && resourceName ) {
71+ watchFluxResource ( {
72+ watchPath,
73+ resourceName,
74+ query,
75+ variables,
76+ field,
77+ client,
78+ } ) ;
79+ }
80+
81+ return fluxData ?. status ?. conditions || [ ] ;
3382 } )
3483 . catch ( ( err ) => {
3584 handleClusterError ( err ) ;
@@ -62,7 +111,16 @@ const getFluxResources = (configuration, url) => {
62111} ;
63112
64113export default {
65- fluxKustomizationStatus ( _ , { configuration, namespace, environmentName, fluxResourcePath = '' } ) {
114+ fluxKustomizationStatus (
115+ _ ,
116+ { configuration, namespace, environmentName, fluxResourcePath = '' } ,
117+ { client } ,
118+ ) {
119+ const watchPath = buildFluxResourceWatchPath ( {
120+ namespace,
121+ apiVersion : kustomizationsApiVersion ,
122+ resourceType : KUSTOMIZATIONS_RESOURCE_TYPE ,
123+ } ) ;
66124 let url ;
67125
68126 if ( fluxResourcePath ) {
@@ -76,9 +134,25 @@ export default {
76134 environmentName,
77135 } ) ;
78136 }
79- return getFluxResourceStatus ( configuration , url ) ;
137+ return getFluxResourceStatus ( {
138+ url,
139+ watchPath,
140+ query : fluxKustomizationStatusQuery ,
141+ variables : { configuration, namespace, environmentName, fluxResourcePath } ,
142+ field : kustomizationField ,
143+ client,
144+ } ) ;
80145 } ,
81- fluxHelmReleaseStatus ( _ , { configuration, namespace, environmentName, fluxResourcePath } ) {
146+ fluxHelmReleaseStatus (
147+ _ ,
148+ { configuration, namespace, environmentName, fluxResourcePath } ,
149+ { client } ,
150+ ) {
151+ const watchPath = buildFluxResourceWatchPath ( {
152+ namespace,
153+ apiVersion : helmReleasesApiVersion ,
154+ resourceType : HELM_RELEASES_RESOURCE_TYPE ,
155+ } ) ;
82156 let url ;
83157
84158 if ( fluxResourcePath ) {
@@ -92,7 +166,14 @@ export default {
92166 environmentName,
93167 } ) ;
94168 }
95- return getFluxResourceStatus ( configuration , url ) ;
169+ return getFluxResourceStatus ( {
170+ url,
171+ watchPath,
172+ query : fluxHelmReleaseStatusQuery ,
173+ variables : { configuration, namespace, environmentName, fluxResourcePath } ,
174+ field : helmReleaseField ,
175+ client,
176+ } ) ;
96177 } ,
97178 fluxKustomizations ( _ , { configuration, namespace } ) {
98179 const url = buildFluxResourceUrl ( {
0 commit comments