import $ from 'jquery'; import { PortainerEndpointTypes } from 'Portainer/models/endpoint/models'; /* @ngInject */ export function onStartupAngular($rootScope, $state, $interval, LocalStorage, EndpointProvider, SystemService, cfpLoadingBar, $transitions, HttpRequestHelper) { EndpointProvider.initialize(); $rootScope.$state = $state; const defaultTitle = document.title; $transitions.onEnter({}, () => { const endpoint = EndpointProvider.currentEndpoint(); if (endpoint) { document.title = `${defaultTitle} | ${endpoint.Name}`; } }); // Workaround to prevent the loading bar from going backward // https://github.com/chieffancypants/angular-loading-bar/issues/273 const originalSet = cfpLoadingBar.set; cfpLoadingBar.set = function overrideSet(n) { if (n > cfpLoadingBar.status()) { originalSet.apply(cfpLoadingBar, arguments); } }; $transitions.onBefore({}, () => { HttpRequestHelper.resetAgentHeaders(); }); // Keep-alive Edge endpoints by sending a ping request every minute $interval(() => { ping(EndpointProvider, SystemService); }, 60 * 1000); $(document).ajaxSend((event, jqXhr, jqOpts) => { const type = jqOpts.type === 'POST' || jqOpts.type === 'PUT' || jqOpts.type === 'PATCH'; const hasNoContentType = jqOpts.contentType !== 'application/json' && jqOpts.headers && !jqOpts.headers['Content-Type']; if (type && hasNoContentType) { jqXhr.setRequestHeader('Content-Type', 'application/json'); } jqXhr.setRequestHeader('Authorization', 'Bearer ' + LocalStorage.getJWT()); }); } function ping(EndpointProvider, SystemService) { const endpoint = EndpointProvider.currentEndpoint(); if (endpoint && endpoint.Type == PortainerEndpointTypes.EdgeAgentOnDockerEnvironment) { SystemService.ping(endpoint.Id); } }