forked from dixonwille/aws-account-header
-
Notifications
You must be signed in to change notification settings - Fork 0
/
aws.js
68 lines (60 loc) · 2.09 KB
/
aws.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
function onError(error) {
console.log(`Error: ${error}`);
}
function normaliseAccountID(acct) {
return acct.replace(/-/ig, '');
}
function onGot(item, acct) {
if (item.hasOwnProperty(acct)) {
document
.querySelectorAll(
"body #awsgnav #nav-menubar, body #awsgnav #nav-menubar .nav-menu, #nav-menu-right, #awsc-nav-header"
)
.forEach(function(ele) {
ele.style.backgroundColor = item[acct];
});
}
}
/**
* injectScript - Inject internal script to available access to the `window`
*
* @param {type} file_path Local path of the internal script.
* @param {type} tag The tag as string, where the script will be append (default: 'body').
* @see {@link http://stackoverflow.com/questions/20499994/access-window-variable-from-content-script}
*/
function injectScript(file_path, tag) {
var node = document.getElementsByTagName(tag)[0];
var script = document.createElement('script');
script.setAttribute('type', 'text/javascript');
script.setAttribute('src', file_path);
node.appendChild(script);
}
function requestCurrentAccountInfo( callback ) {
if (chrome) {
window.addEventListener("message", function(message) {
console.log(message);
if (message.data && message.data.from && ('aws-account-header' === message.data.from ) ) {
callback (message.data.AccountInfo)
}
});
injectScript(chrome.extension.getURL('inject.js'), 'body');
} else {
if ( window.wrappedJSObject && window.wrappedJSObject.ConsoleNavService && window.wrappedJSObject.ConsoleNavService.AccountInfo) {
callback ( window.wrappedJSObject.ConsoleNavService.AccountInfo );
} else {
console.log ('no ConsoleNavService found');
}
}
}
requestCurrentAccountInfo ( function (accountInfo) {
var normalisedAccId = normaliseAccountID(accountInfo.userAccountNumber);
if (chrome) {
chrome.storage.sync.get(function (item) {
return onGot( item, normalisedAccId)
});
} else {
browser.storage.sync.get().then(function (item) {
return onGot( item, normalisedAccId)
}, onError);
}
});