HR-Collector/Extension/content.js
2024-12-08 22:52:38 +01:00

326 lines
No EOL
17 KiB
JavaScript

chrome.runtime.onMessage.addListener(async (request, sender, sendResponse) => {
try {
if (request.action === "getHorseBasicData") {
console.log("Getting horse data");
const idEl = document.querySelector('.right:nth-child(2)');
const ageEl = document.querySelector('.right:nth-child(6)');
const genderEl = document.querySelector('img.icon16');
const breedEl = document.querySelector('.right:nth-child(4)');
if (!idEl || !ageEl || !genderEl || !breedEl) {
console.error("Some required elements for basic horse data not found on the page.");
sendResponse({ success: false, message: "Required elements not found." });
return true;
}
const idString = idEl.innerText.replace('#', '');
const id = BigInt(idString).toString();
const horseData = {
id: parseFloat(id),
age: parseInt(ageEl.innerText, 10) || 0,
name: (document.title.replace(/ - Horse Reality.*$/, '') || "Unknown").trim(),
gender: genderEl.alt || "Unknown",
breed: breedEl.innerText || "Unknown",
link: window.location.href || "",
lastDrawnDate: new Date(Date.now()).toISOString()
};
updateSingleLoadStateUI("Basic", true, false);
console.log("Horse data gathered:", horseData);
sendResponse({ success: true, data: horseData });
} else if (request.action === "getHorseCurrentData") {
console.log("Getting current horse selected tab data");
const selectedTab = getTabselText();
console.log("Selected tab:", selectedTab);
if (selectedTab === "Summary") {
const pedigreeLinks = document.querySelectorAll('.pedigree a');
if (!pedigreeLinks) {
console.warn("No pedigree links found for Summary tab.");
sendResponse({ success: false, message: "No pedigree links found." });
return true;
}
const relatedHorses = Array.from(pedigreeLinks).map(link => link.href);
console.log("Related horses:", relatedHorses);
try {
sendResponse({ success: true, data: "Processing..." });
const response = await setHorsePedigreeAPIAsync(request.data.id, relatedHorses);
if (!response.ok) {
console.error("API returned an error:", response.statusText);
sendResponse({ success: false, message: "API error: " + response.statusText });
return true;
}
const jsonData = await response.json();
console.log("API Response:", jsonData);
sendResponse({ success: true, data: jsonData });
} catch (error) {
console.error("Error in Summary tab:", error);
sendResponse({ success: false, message: error.message });
}
updateSingleLoadStateUI("Summary", true, false);
} else if (selectedTab === "Training") {
const trainingEl8 = document.querySelector('.top:nth-child(8)');
const trainingEl6 = document.querySelector('.top:nth-child(6)');
const training = (trainingEl8 && trainingEl8.innerText)
|| (trainingEl6 && trainingEl6.innerText)
|| "Unknown";
try {
sendResponse({ success: true, data: "Processing..." });
const response = await setHorseTrainingAPIAsync(request.data.id, training);
if (!response.ok) {
console.error("API returned an error:", response.statusText);
sendResponse({ success: false, message: "API error: " + response.statusText });
return true;
}
const jsonData = await response.json();
console.log("API Response:", jsonData);
sendResponse({ success: true, data: jsonData });
} catch (error) {
console.error("Error in Training tab:", error);
sendResponse({ success: false, message: error.message });
}
updateSingleLoadStateUI("Training", true);
} else if (selectedTab === "Genetics") {
try {
const gpDiv = Array.from(document.querySelectorAll('.top div')).find(el => el.innerText.includes('GP'));
if (!gpDiv) {
console.error("No GP element found.");
sendResponse({ success: false, message: "No GP element found." });
return true;
}
const totalGeneticPotentialStr = gpDiv.innerText.replace("GP total: ", "").trim();
const totalGeneticPotential = parseInt(totalGeneticPotentialStr, 10) || 0;
const geneticElems = Array.from(document.querySelectorAll('.genetic_name , .genetic_result'));
if (geneticElems.length === 0) {
console.error("No genetic elements found.");
sendResponse({ success: false, message: "No genetic elements found." });
return true;
}
const geneticDictionary = Object.fromEntries(
geneticElems.map(x => x.innerText)
.reduce((acc, cur, i, arr) => (i % 2 === 0 ? acc.push([cur, arr[i + 1]]) : null, acc), [])
);
const geneticsStats = Array.from(document.querySelectorAll('.genetic_stats'));
if (geneticsStats.length < 10) {
console.error("Not enough genetic stats found.");
sendResponse({ success: false, message: "Not enough genetic stats found for genetics calculation." });
return true;
}
const geneticPotentialList = geneticsStats
.map(x => x.innerText.trim())
.filter(value => !isNaN(value) && value !== "")
.map(value => parseFloat(value));
const GeneticPotential = {
GP: totalGeneticPotential,
GeneticPotential: {
"Acceleration": geneticPotentialList[0],
"Agility": geneticPotentialList[1],
"Balance": geneticPotentialList[2],
"Bascule": geneticPotentialList[3],
"Pulling power": geneticPotentialList[4],
"Speed": geneticPotentialList[5],
"Sprint": geneticPotentialList[6],
"Stamina": geneticPotentialList[7],
"Strength": geneticPotentialList[8],
"Surefootedness": geneticPotentialList[9]
},
Disciplines: {
"Dressage": geneticPotentialList[1] + geneticPotentialList[2] + geneticPotentialList[8],
"Driving": geneticPotentialList[1] + geneticPotentialList[4] + geneticPotentialList[5] + geneticPotentialList[7] + geneticPotentialList[8],
"Endurance": geneticPotentialList[5] + geneticPotentialList[7] + geneticPotentialList[8] + geneticPotentialList[9],
"Eventing": geneticPotentialList[2] + geneticPotentialList[3] + geneticPotentialList[5] + geneticPotentialList[8] + geneticPotentialList[9],
"Flat Racing": geneticPotentialList[5] + geneticPotentialList[0] + geneticPotentialList[7] + geneticPotentialList[6],
"Show Jumping": geneticPotentialList[0] + geneticPotentialList[1] + geneticPotentialList[3] + geneticPotentialList[6] + geneticPotentialList[8],
"Western Reining": geneticPotentialList[0] + geneticPotentialList[1] + geneticPotentialList[2] + geneticPotentialList[9]
},
Colors: geneticDictionary
};
console.log("Genetic Potential:", GeneticPotential);
sendResponse({ success: true, data: "Processing..." });
const response = await setHorseGeneticsAPIAsync(request.data.id, GeneticPotential);
if (!response.ok) {
console.error("API returned an error:", response.statusText);
sendResponse({ success: false, message: "API error: " + response.statusText });
return true;
}
const jsonData = await response.json();
console.log("API Response:", jsonData);
sendResponse({ success: true, data: jsonData });
} catch (error) {
console.error("Error while processing Genetics tab:", error);
sendResponse({ success: false, message: error.message });
}
updateSingleLoadStateUI("Genetics", true, false);
} else if (selectedTab === "Achievements") {
const geneticStatsEls = document.querySelectorAll('.genetic_stats');
if (!geneticStatsEls || geneticStatsEls.length === 0) {
console.error("No genetic_stats elements found for Achievements.");
sendResponse({ success: false, message: "No genetic_stats found for Achievements." });
return true;
}
const competitionResults = Array.from(document.querySelectorAll('.row_460 .col_120+ .col_70'))
.map(x => parseFloat(x.innerText.trim()))
.sort((a, b) => a - b)
.filter(x => !isNaN(x))
.length > 0
? Array.from(document.querySelectorAll('.row_460 .col_120+ .col_70'))
.map(x => parseFloat(x.innerText.trim()))
.sort((a, b) => a - b)
.filter(x => !isNaN(x))
: [document.querySelector('.even > p') ? document.querySelector('.even > p').innerText : ''];
const showResults = Array.from(document.querySelectorAll('.row_460 .col_120 + .col_90'))
.map(x => parseFloat(x.innerText.trim()))
.sort((a, b) => a - b)
.filter(x => !isNaN(x))
.length > 0
? Array.from(document.querySelectorAll('.row_460 .col_120 + .col_90'))
.map(x => parseFloat(x.innerText.trim()))
.sort((a, b) => a - b)
.filter(x => !isNaN(x))
: [document.querySelector('.even > p') ? document.querySelector('.even > p').innerText : ''];
const conformation = Object.fromEntries(
Array.from(document.querySelectorAll('.genetic_table_row')).flatMap(row => {
const keys = Array.from(row.querySelectorAll('.genetic_potential')).map(key => key.innerText.trim()); // Alle Schlüssel holen
const values = Array.from(row.querySelectorAll('.genetic_stats'))
.map(value => value.innerText.trim())
.filter(value => /poor|below average|average|good|very good/i.test(value)); // Nur relevante Werte behalten
return keys.map((key, index) => [key, values[index]]).filter(pair => pair[1]); // Paare mit gültigen Werten
})
);
const conformationMapping = {
"Good": "G",
"Very good": "VG",
"Average": "A",
"Below average": "BA",
"Poor": "P"
};
// Alle Werte aus dem bestehenden `conformation`-Objekt durchlaufen
let shortConformation = "";
const counts = Object.values(conformation)
.filter(value => conformationMapping[value]) // Nur Werte berücksichtigen, die in der Mapping-Tabelle existieren
.reduce((counts, value) => {
counts[value] = (counts[value] || 0) + 1; // Zähle die Vorkommen jedes Werts
return counts;
}, {});
// Gehe durch die gezählten Werte und erstelle die Kurzform
Object.entries(counts).forEach(([value, count]) => {
shortConformation += `${count}${conformationMapping[value]} `;
});
// Entferne Leerzeichen am Ende
shortConformation = shortConformation.trim();
console.log(shortConformation);
try {
const achievementTab =
{
ShowResults: cleanShowResults(showResults),
Conformation: conformation,
ShortConformation: shortConformation,
MaxShowResult: cleanResult(showResults[showResults.length - 1]),
MinShowResult: cleanResult(showResults[0]),
MaxCompetitionResult: cleanResult(competitionResults[competitionResults.length - 1]),
MinCompetitionResult: cleanResult(competitionResults[0])
};
sendResponse({ success: true, data: "Processing..." });
const response = await setHorseAchievementsAPIAsync(request.data.id, achievementTab);
if (!response.ok) {
console.error("API returned an error:", response.statusText);
sendResponse({ success: false, message: "API error: " + response.statusText });
return true;
}
const jsonData = await response.json();
console.log("API Response:", jsonData);
sendResponse({ success: true, data: jsonData });
} catch (error) {
console.error("Error in Achievements tab:", error);
sendResponse({ success: false, message: error.message });
}
updateSingleLoadStateUI("Achievements", true, true);
} else if (selectedTab === "Health") {
const healthEl = document.querySelector("#tab_health2 p");
if (!healthEl) {
console.error("Health element not found.");
sendResponse({ success: false, message: "Health element not found." });
return true;
}
const matches = [...healthEl.innerText.matchAll(/:\s*(\w+)/g)];
const health = Object.fromEntries(document.querySelector("#tab_health2 p").innerText.split('\n').map(l=>l.trim()).filter(l=>l&&l.includes(':')).map(l=>l.split(':').map(x=>x.trim())));
console.log("Health data:", health);
try {
sendResponse({ success: true, data: "Processing..." });
const response = await setHorseHealthAPIAsync(request.data.id, health);
if (!response.ok) {
console.error("API returned an error:", response.statusText);
sendResponse({ success: false, message: "API error: " + response.statusText });
return true;
}
const jsonData = await response.json();
console.log("API Response:", jsonData);
sendResponse({ success: true, data: jsonData });
} catch (error) {
console.error("Error in Health tab:", error);
sendResponse({ success: false, message: error.message });
}
updateSingleLoadStateUI("Health", true, false);
} else {
console.warn("Unknown or no tab selected.");
sendResponse({ success: false, message: "Unknown or no tab selected." });
}
} else {
console.error("Unsupported action:", request.action);
sendResponse({ success: false, message: "Unsupported action." });
}
} catch (error) {
console.error("Unexpected error occurred:", error);
sendResponse({ success: false, message: error.message });
}
return true; // Wichtig, um asynchrone Antworten zu ermöglichen
});
function getTabselText() {
return document.querySelector('div.tabsel')?.textContent?.trim() || "Unknown";
}
function cleanResult(value) {
// Check if the value is not a number or cannot be converted to a number
if (isNaN(value) || value === null || value === undefined || value === "") {
return -1;
}
return parseFloat(value); // Convert to a number if it is valid
}
function cleanShowResults(results) {
// Überprüfen, ob das Array gültig ist
if (!Array.isArray(results)) {
console.error("Invalid ShowResults data: not an array.");
return [];
}
// Bereinigen der Ergebnisse
return [];
}