1.2.2

p0358 · Feb 28, 2021 · a4921e8 · a4921e8
1 parent 2755655
commit a4921e8
Show file tree

Hide file tree

Showing 8 changed files with 199 additions and 47 deletions.
diff --git a/background.js b/background.js
@@ -12,7 +12,10 @@ function appendReferer(e) {
 let appendReferer_extraInfoSpec = ["blocking", "requestHeaders"];
 if (chrome.webRequest.OnBeforeSendHeadersOptions.hasOwnProperty('EXTRA_HEADERS')) appendReferer_extraInfoSpec.push('extraHeaders'); // Chrome needs this apparently...
 browser.webRequest.onBeforeSendHeaders.addListener(appendReferer,
-    {urls: ["https://portal.librus.pl/rodzina/synergia/loguj", "https://portal.librus.pl/szkola/synergia/loguj"]},
+    {urls: [
+        "https://portal.librus.pl/rodzina/synergia/loguj", "https://portal.librus.pl/szkola/synergia/loguj",
+        "https://portal.librus.pl/rodzina/synergia/loguj#*", "https://portal.librus.pl/szkola/synergia/loguj#*"
+    ]},
     appendReferer_extraInfoSpec
 );
 
@@ -27,7 +30,22 @@ browser.webRequest.onBeforeRequest.addListener(e => {
 
 // this login link would redirect user to the main page, so catch it and redirect directly to the actual login page instead
 browser.webRequest.onBeforeRequest.addListener(e => {
-    return {redirectUrl: 'https://portal.librus.pl/rodzina/synergia/loguj'};
+    let appendee = '';
+    let moveToUri = '';
+    if (e.url === 'https://synergia.librus.pl/loguj' && e.originUrl && e.originUrl.startsWith('https://synergia.librus.pl/')) {
+        // this will only work in Firefox
+        moveToUri = e.originUrl.replace('https://synergia.librus.pl', '');
+    }
+    else if (e.url && e.url.startsWith('https://synergia.librus.pl/loguj/przenies/')) {
+        moveToUri = e.url.replace('https://synergia.librus.pl/loguj/przenies', '');
+    }
+    moveToUri = moveToUri.replace(/\\\//g, '/');
+    if (moveToUri
+        && moveToUri !== '/rodzic/index' && moveToUri !== '/uczen/index' // we don't care about main page redirect, especially that mixing up "rodzic" and "uczen" would log us out again
+        && moveToUri !== '/loguj' && !moveToUri.startsWith('/loguj')
+    ) 
+        appendee = '#' + moveToUri;
+    return {redirectUrl: 'https://portal.librus.pl/rodzina/synergia/loguj' + appendee};
 },
     {urls: ["https://synergia.librus.pl/loguj", "https://synergia.librus.pl/loguj/przenies*"]},
     ["blocking"]

diff --git a/browser-polyfill.min.js b/browser-polyfill.min.js
diff --git a/browser-polyfill.min.js.map b/browser-polyfill.min.js.map
diff --git a/contentscript.js b/contentscript.js
@@ -2,7 +2,7 @@
 
 const isChrome = typeof browser === "undefined" || Object.getPrototypeOf(browser) !== Object.prototype;
 const isFirefox = !isChrome && !!browser;
-const debug = false;
+const debug = true;
 let url = document.location.toString().split('#')[0].split('?')[0];
 let didInjectCssAlready = false;
 
@@ -25,6 +25,7 @@ if (document.readyState === 'complete') onDocumentReady();
 
 setTimeout(() => modifyStuff(), 500); // just to make sure and catch the late loaded stuff
 setTimeout(() => modifyStuff(), 1500); // apparently the one above was not enough...
+setTimeout(() => modifyStuff(), 2500); // apparently the one above was not enough...
 
 // run on document ready once
 function onDocumentReady() {}
@@ -60,6 +61,25 @@ function modifyStuff() {
     if (url === 'https://portal.librus.pl/rodzina/synergia/loguj' || url === 'https://portal.librus.pl/szkola/synergia/loguj') {
         $("#synergiaLogin").next().remove(); // some ad
         $('section#app-download').remove(); // mobile app ad (only on mobile screens)
+
+        // they send: window.parent.postMessage({"command":"open_synergia_window","commandPayload":{"url":"https:\/\/synergia.librus.pl\/uczen\/index"}}, 'https://portal.librus.pl/rodzina');
+        let redirectUrl = window.location.hash && window.location.hash.substr(1);
+        if (redirectUrl && !window.didSetUpOpenSynergiaWindowListenerAlready) {
+            console.log("[Librus Enhancer] Setting up window message event listener to change redirect destination after login (as hash url is provided)");
+            window.addEventListener("message", (event) => {
+                if (debug) console.log("[Librus Enhancer] message event", event);
+                if (event.data && event.data.command && event.data.command === 'open_synergia_window') {
+                    console.log("[Librus Enhancer] intercepted open_synergia_window");
+                    event.stopImmediatePropagation();
+                    event.stopPropagation();
+                    //let url = event.data && event.data.commandPayload && event.data.commandPayload.url;
+                    let url = window.location.hash && window.location.hash.substr(1);
+                    if (!url) url = '/uczen/index';
+                    document.location.assign('https://synergia.librus.pl/' + url.replace(/^\/+/, ''));
+                }
+            }, true);
+            window.didSetUpOpenSynergiaWindowListenerAlready = true;
+        }
     }
 
     if (url === 'https://portal.librus.pl/rodzina' || url === 'https://portal.librus.pl/rodzina/login' || url === 'https://portal.librus.pl/szkola') {
@@ -128,61 +148,163 @@ function modifyStuff() {
 const PAGE_REFRESH_INTERVAL = 3 * 60 * 1000;
 if (url.startsWith('https://synergia.librus.pl/')) {
 
-    let isLoggedIn = !!$('#user-section').length;
+    let isLoggedIn = !!$('#user-section').length; // we are catching only full pages, not popups
     let div = $('#page .container .inside');
     let isLoggedOut = div && !!div.length && div.text() === 'Brak dostępu';
 
     if (isLoggedIn && !isLoggedOut) {
-        let regex = /jesteś zalogowany jako\:.*(uczeń|rodzic).*/gsi.exec($('#user-section').text());
-        let isParent = regex && regex[1] && regex[1] === 'rodzic';
+        let regex = /(jesteś zalogowan[ya\-\(\)\/]+ jako\:).*(uczeń|rodzic).*/gsi.exec($('#user-section').text());
+
+        let isPupil = regex && regex[1] && regex[2] && regex[2] === 'uczeń';
+        let isParent = regex && regex[1] && regex[2] && regex[2] === 'rodzic';
+        let isTeacher = regex && regex[1] && !isPupil && !isParent;
+
+        // fallback detections
+        //if (!isPupil && (!!$("a[href='/przegladaj_oceny/uczen']").length || !!$("a[href='/przegladaj_nb/uczen']").length)) isPupil = true;
+        //if (!isParent && (!!$("a[href='/przegladaj_oceny/rodzic']").length || !!$("a[href='/przegladaj_nb/rodzic']").length)) isParent = true; // both pupil and parent turn out to have "uczen" in these links!!!
+        if (!isTeacher && !!$("a[href='/interfejs_lekcyjny']").length) isTeacher = true;
+
+        let accountType = -1;
+        let accountTypeStr = "unknown";
+        switch (true) {
+            case isPupil && !isParent && !isTeacher: accountType = ACCOUNT_TYPE_PUPIL; accountTypeStr = "pupil"; break;
+            case isParent && !isPupil && !isTeacher: accountType = ACCOUNT_TYPE_PARENT; accountTypeStr = "parent"; break;
+            case isTeacher && !isPupil && !isParent: accountType = ACCOUNT_TYPE_TEACHER; accountTypeStr = "teacher"; break;
+        }
 
-        console.log("[Librus Enhancer] Detected that you are logged in!" + (isParent ? " (as parent)":""));
+        console.log("[Librus Enhancer] Detected that you are logged in! (as "+accountTypeStr+")");
+
+        $("a[href='/wyloguj']").on('click', () => storage.set({
+            //['lastLogin'+(isTeacher?'School':'')]: null,
+            lastLogin: null,
+            lastLoginSchool: null,
+            pupilNumber: -1
+        }));
+        revealLiblinks();
+
+        if (!regex || !regex[1] || accountType == -1)
+            console.warn("[Librus Enhancer] Could not detect account type!");
 
         // the difference between these two is documented above these functions' definitions
         if (isFirefox) {
-            setInterval(() => firefox_refreshPageInBackground(isParent), PAGE_REFRESH_INTERVAL);
-            //setTimeout(() => firefox_refreshPageInBackground(isParent), 3000); // testing
+            setInterval(() => firefox_refreshPageInBackground(accountType), PAGE_REFRESH_INTERVAL);
+            //setTimeout(() => firefox_refreshPageInBackground(accountType), 3000); // testing
         } else {
-            chrome_refreshPageInBackground_setupIntervalInPageContext(isParent);
+            chrome_refreshPageInBackground_setupIntervalInPageContext(accountType);
         }
 
-        $("a[href='/wyloguj']").on('click', () => storage.set({lastLogin: null}));
-        revealLiblinks();
+        if (accountType == ACCOUNT_TYPE_PUPIL || accountType == ACCOUNT_TYPE_PARENT)
+            grabPupilNumber();
 
-    } else {
-        console.log("[Librus Enhancer] Detected that you are logged out!");
-        if (isLoggedOut) {
-            // make the login button move us to the login page, and not main page 
-            let button = $("input[value='Loguj']");
-            if (button && !!button.length) {
-                button.attr('onclick', '');
-                button.on('click', ev => {
-                    ev.preventDefault();
-                    ev.stopImmediatePropagation();
-                    // below doesn't support the school version currently, as I have no account to test it with; I assume most users will be family users anyways
-                    window.location.replace('https://portal.librus.pl/rodzina/synergia/loguj'); // take us to the login page directly...
-                });
-            }
+    } else if (isLoggedOut) {
+        console.log("[Librus Enhancer] Detected that you were logged out!");
+        let isFamily = (document.body.innerHTML+'').includes('https:\\/\\/synergia.librus.pl\\/loguj\\/przenies\\/rodzic\\/index');
+        if (isFamily) {
+            let pathname = window.location.pathname;
+            if (pathname !== 'https://synergia.librus.pl/rodzic/index' && pathname !== 'https://synergia.librus.pl/uczen/index')
+                window.location.replace('https://portal.librus.pl/rodzina/synergia/loguj#' + pathname);
+        }
+
+        // make the login button move us to the login page, and not main page 
+        let button = $("input[value='Loguj']");
+        if (button && !!button.length) {
+            button.attr('onclick', '');
+            button.on('click', ev => {
+                ev.preventDefault();
+                ev.stopImmediatePropagation();
+                window.location.replace(`https://portal.librus.pl/${isFamily?'rodzina':'szkola'}/synergia/loguj`); // take us to the login page directly...
+            });
         }
+    } else {
+        console.log("[Librus Enhancer] Detected that you are neither logged in or out (popup or another page)");
     }
 
 }
 
+async function grabPupilNumber() {
+    let data = await storage.get(['pupilNumber']);
+    let pupilNumber = +data.pupilNumber;
+    if (!pupilNumber || pupilNumber == -1) {
+
+        if (debug) console.log('[Librus Enhancer] Pupil number was not in storage, gonna fetch it.');
+        // fetch it
+        let url = 'https://synergia.librus.pl/informacja';
+        let regex = /<th class="big">Nr w dzienniku\s*<\/th>\s*<td>\s*([0-9]+)\s*<\/td>/gs;
+
+        let code = `${debug ? "console.log('[Librus Enhancer] Fetching number in page context...');":""}
+        fetch('${url}', {
+            cache: 'no-cache',
+            credentials: 'include'
+        }).then(response => response.text()).then(text => {
+            let regex = /<th class="big">Nr w dzienniku\\s*<\\/th>\\s*<td>\\s*([0-9]+)\\s*<\\/td>/gs;
+            let data = regex.exec(text);
+            let number = data && data[1] && +data[1];
+            if (!number) return console.warn('[Librus Enhancer] Failed to fetch pupil number!', data);
+            ${debug ? "console.log('[Librus Enhancer] Fetched number in page context:', number);":""}
+            const event = new CustomEvent('PupilNumberEvent', {detail: {number}});
+            window.dispatchEvent(event);
+        });`;
+
+        window.addEventListener('PupilNumberEvent', e => {
+            let number = +(e.detail && e.detail.number);
+            if (!number) return console.warn('[Librus Enhancer] Received invalid pupil number in event listener!', e);
+            storage.set({pupilNumber: number});
+            processPupilNumber(number);
+        });
+
+        if (isFirefox) {
+            window.eval(code);
+        } else {
+            let script = document.createElement('script');
+            script.type = 'text/javascript';
+            script.async = true;
+            script.innerHTML = code;
+            document.head.appendChild(script);
+        }
+
+    } else processPupilNumber(pupilNumber);
+}
+
+function processPupilNumber(number) {
+    console.log("[Librus Enhancer] Determined your number:", number);
+    let currentLuckyNumber = +($('.luckyNumber').text()+'').trim();
+    if (currentLuckyNumber === number) {
+        console.log("[Librus Enhancer] Your number is lucky today!");
+        $('.luckyNumber').css('border', '15px solid red'); // dashed
+        $('.luckyNumber').css('font-size', '20px');
+        $('.luckyNumber').css('background-color', 'lime');
+    }
+}
+
+function getLinkToFetchForBackgroundSessionRefresh(accountType) {
+    let linkToFetch;
+    switch (accountType) {
+        case ACCOUNT_TYPE_PUPIL: linkToFetch = 'https://synergia.librus.pl/uczen/index'; break;
+        case ACCOUNT_TYPE_PARENT: linkToFetch = 'https://synergia.librus.pl/rodzic/index'; break;
+        case ACCOUNT_TYPE_TEACHER: linkToFetch = 'https://synergia.librus.pl/interfejs_lekcyjny'; break;
+        default: linkToFetch = 'https://synergia.librus.pl/ustawienia'; break;
+    }
+    return linkToFetch;
+}
+
 // the difference between below functions:
 // firefox one sets interval in content script (in code block above), and uses window.eval to then execute the request in page context
 // chrome one sets both the interval and request in page context (with script tag workaround, as window.eval is not available there)
 
 // prevent session expiration, so we don't get logged out...
-function firefox_refreshPageInBackground(isParent) { // this function only gets executed in Firefox
+function firefox_refreshPageInBackground(accountType) { // this function only gets executed in Firefox
     console.log("[Librus Enhancer] Running firefox_refreshPageInBackground in page context...");
+
+    let linkToFetch = getLinkToFetchForBackgroundSessionRefresh(accountType);
+
     /*fetch('https://synergia.librus.pl/uczen/index', { // content.fetch crashes tab lol
         //mode: 'same-origin',
         cache: 'no-cache',
         credentials: 'include'
     }).then(response => console.log("[Librus Enhancer] Refreshed page in background to preserve the session (response status: " + response.status + ", length: " + response.headers.get("content-length") + ")"));
     */
 
-    let code = `fetch('https://synergia.librus.pl/${isParent?'rodzic':'uczen'}/index', {
+    let code = `fetch('${linkToFetch}', {
         cache: 'no-cache',
         credentials: 'include'
     }).then(response => console.log("[Librus Enhancer] Refreshed page in background to preserve the session (response status: " + response.status + ", length: " + response.headers.get("content-length") + ")"));`;
@@ -193,8 +315,11 @@ function firefox_refreshPageInBackground(isParent) { // this function only gets
 
 // workaround for chrome
 // the modern-age Internet Explorer
-function chrome_refreshPageInBackground_setupIntervalInPageContext(isParent) { // this function only gets executed in Chrome, and not in Firefox
-    let fetchcode = `fetch('https://synergia.librus.pl/${isParent?'rodzic':'uczen'}/index', {
+function chrome_refreshPageInBackground_setupIntervalInPageContext(accountType) { // this function only gets executed in Chrome, and not in Firefox
+
+    let linkToFetch = getLinkToFetchForBackgroundSessionRefresh(accountType);
+
+    let fetchcode = `fetch('${linkToFetch}', {
         cache: 'no-cache',
         credentials: 'include'
     }).then(response => console.log("[Librus Enhancer] Refreshed page in background to preserve the session (response status: " + response.status + ", length: " + response.headers.get("content-length") + ")"));`;    
@@ -371,4 +496,4 @@ if (url === 'https://portal.librus.pl//vendor/widget-librus/index.html' || url =
 
     const observer = new MutationObserver(callback);
     observer.observe(document.body, {childList: true, subtree: true});
-}
+}
diff --git a/contentscript_common.js b/contentscript_common.js
@@ -1,3 +1,7 @@
+const ACCOUNT_TYPE_PUPIL = 0;
+const ACCOUNT_TYPE_PARENT = 1;
+const ACCOUNT_TYPE_TEACHER = 2;
+
 //#region ENCRYPTION
 
 // better to encrypt with a statically embedded key than to store in storage in plaintext 
@@ -46,5 +50,5 @@ async function decryptPassword(passEncrypted) {
 //#endregion ENCRYPTION
 
 /** @type {browser.storage.StorageArea} */
-//let storage = browser.storage.sync; // production?
-let storage = browser.storage.local; // development
+//let storage = browser.storage.sync;
+let storage = browser.storage.local;
diff --git a/contentscript_login_iframe.js b/contentscript_login_iframe.js
@@ -32,7 +32,7 @@ let loadingSavedLoginsPromise = (async () => {
 
         $("#Login").val(savedLogin.login);
         $("#Pass").val(await decryptPassword(savedLogin.passEncrypted));
-        $("#LoginBtn").click();
+        $("#LoginBtn").trigger('click');
     }
 })();*/
 
@@ -49,15 +49,15 @@ function insertSavedLoginsButtons() {
 
         let loginPart = $('<div class="loginPart">');
         loginPart.text(savedLogin.login);
-        loginPart.click(async () => {
+        loginPart.on('click', async () => {
             $("#Login").val(savedLogin.login);
             $("#Pass").val(await decryptPassword(savedLogin.passEncrypted));
-            $("#LoginBtn").click();
+            $("#LoginBtn").trigger('click');
         });
 
         let deletePart = $('<div class="deletePart">');
         deletePart.text('×');
-        deletePart.click(async () => {
+        deletePart.on('click', async () => {
             savedLogins = savedLogins.filter(sl => sl !== savedLogin);
             await storage.set({savedLogins});
             insertSavedLoginsButtons();
@@ -71,7 +71,7 @@ function insertSavedLoginsButtons() {
     $('.LoginBox').prepend(container);
 }
 
-$("#LoginBtn").click(async (ev) => {
+$("#LoginBtn").on('click', async (ev) => {
     let remember = $("input#remember").is(':checked'); // credentials will only be saved in storage if user has explicitly pressed the checkbox to "remember for quick login"
 
     let login = $("#Login").val();
@@ -86,7 +86,7 @@ $("#LoginBtn").click(async (ev) => {
         await loadingSavedLoginsPromise;
 
         // remove users that match login, but don't match the password, if user is updating one right now
-        savedLogins = savedLogins.filter(savedLogin => savedLogin.login === login && savedLogin.passEncrypted !== passEncrypted);
+        savedLogins = savedLogins.filter(savedLogin => !(savedLogin.login === login && savedLogin.passEncrypted !== passEncrypted));
 
         // if the entry doesn't exist on our list
         if (!savedLogins.some(savedLogin => savedLogin.login === login && savedLogin.passEncrypted === passEncrypted)) {
@@ -96,11 +96,14 @@ $("#LoginBtn").click(async (ev) => {
     }
 
     // if the user is saved, we will save their name for auto re-login, in case their session expired despite having this addon (interval was throttled or user had disconnected from the internet)
+    // currently not implemented
     // that feature will not affect manual sign-out
     if (login && pass && savedLogins.some(savedLogin => savedLogin.login === login && savedLogin.passEncrypted === passEncrypted)) {
-        storage.set({lastLogin: login});
+        storage.set({['lastLogin'+(isSchool?'School':'')]: login});
     }
+
 });
 
 $("input#remember").parent().remove(); // if we're hot-reloading the script
-$("#passwordRow").append(rememberLoginCode);
+$("#passwordRow").append(rememberLoginCode);
+storage.set({'pupilNumber': -1});
diff --git a/login_iframe.css b/login_iframe.css
@@ -9,8 +9,9 @@
     border-color: #975b83;
 
     box-sizing: border-box;
-    padding: 0;
+    padding: 0 !important;
     margin-top: .3em;
+    margin-right: .3em;
 }
 
 .btn-savedlogin > div:hover {