tiktok.js

(function () {
	function pushMessage(data){	  
		try {
			chrome.runtime.sendMessage(chrome.runtime.id, { "message": data }, function(e){});
		} catch(e){}
	}

	function getTranslation(key, value=false){
		if (settings.translation && settings.translation.innerHTML && (key in settings.translation.innerHTML)){ // these are the proper translations
			return settings.translation.innerHTML[key];
		} else if (settings.translation && settings.translation.miscellaneous && settings.translation.miscellaneous && (key in settings.translation.miscellaneous)){ 
			return settings.translation.miscellaneous[key];
		} else if (value!==false){
			return value;
		} else {
			return key.replaceAll("-", " "); //
		}
	}

	function toDataURL(url, callback) {
	  var xhr = new XMLHttpRequest();
	  xhr.onload = function() {
		  
		var blob = xhr.response;
    
		if (blob.size > (55 * 1024)) {
		  callback(url); // Image size is larger than 25kb.
		  return;
		}

		var reader = new FileReader();
		
		
		reader.onloadend = function() {
		  callback(reader.result);
		}
		reader.readAsDataURL(xhr.response);
	  };
	  xhr.open('GET', url);
	  xhr.responseType = 'blob';
	  xhr.send();
	}
	
	function escapeHtml(unsafe){
		try {
			if (settings.textonlymode){ // we can escape things later, as needed instead I guess.
				return unsafe;
			}
			return unsafe
			 .replace(/&/g, "&")
			 .replace(/</g, "&lt;")
			 .replace(/>/g, "&gt;")
			 .replace(/"/g, "&quot;")
			 .replace(/'/g, "&#039;") || "";
		} catch(e){
			return "";
		}
	}

	function getAllContentNodes(element) { // takes an element.
		var resp = "";
		if (!element){return resp;}
		if (!element.childNodes || !element.childNodes.length){
			if (element.textContent){
				return escapeHtml(element.textContent) || "";
			} else {
				return "";
			}
		}
		element.childNodes.forEach(node=>{
			if (node.childNodes.length){
				resp += getAllContentNodes(node)
			} else if ((node.nodeType === 3) && node.textContent){
				resp += escapeHtml(node.textContent);
			} else if (node.nodeType === 1){
				if (!settings.textonlymode){
					if ((node.nodeName == "IMG") && node.src){
						node.src = node.src+"";
						resp += "<img src='"+node.src+"' />";
					} else {
						resp += node.outerHTML;
					}
				}
			}
		});
		return resp;
	}
	
	function rankToColor(rank, maxRank = 40) {
	  // Start and end colors in RGB
	  const startColor = { r: 197, g: 204, b: 218 }; // #4F6692
	  const midColor = { r: 100, g: 115, b: 225 };    // #2026B0
	  const endColor = { r: 81, g: 85, b: 255 };      // #0000FF

	  // Determine color stops based on rank
	  const midRank = parseInt(maxRank/2);
	  let colorStop;
	  

	  if (rank <= midRank) {
		// Calculate how far the rank is between 1 and midRank
		const ratio = (rank - 1) / (midRank - 1);
		colorStop = {
		  r: startColor.r + ratio * (midColor.r - startColor.r),
		  g: startColor.g + ratio * (midColor.g - startColor.g),
		  b: startColor.b + ratio * (midColor.b - startColor.b),
		};
	  } else {
		// Calculate how far the rank is between midRank and maxRank
		const ratio = (rank - midRank) / (maxRank - midRank);
		colorStop = {
		  r: midColor.r + ratio * (endColor.r - midColor.r),
		  g: midColor.g + ratio * (endColor.g - midColor.g),
		  b: midColor.b + ratio * (endColor.b - midColor.b),
		};
	  }

	  // Convert the RGB color stop to a hex color code
	  const hexColor = `#${Math.round(colorStop.r).toString(16).padStart(2, '0')}` +
					   `${Math.round(colorStop.g).toString(16).padStart(2, '0')}` +
					   `${Math.round(colorStop.b).toString(16).padStart(2, '0')}`;
	  return hexColor;
	}
	var lut = [];
	for (var i =1;i<=40;i++){
		lut.push(rankToColor(i,40));
	}

	var savedavatars = {};
	var channelName = false;
	var msgCount = 0;
	
	const messageLog = {};
	function isDuplicateMessage(name, message, ele) {
		const currentTime = Date.now();
		const messageKey = `${name}:${message}`;
		
		if (messageLog[messageKey] && (currentTime - messageLog[messageKey].currentTime) <= 3000) {
			if (messageLog[messageKey].ele && !messageLog[messageKey].ele.isConnected){
				return true; // exact same message already posted.
			}
		}
		messageLog[messageKey] = {currentTime:currentTime, ele:ele};
		
		for (const key in messageLog) {
			if ((currentTime - messageLog[key].currentTime) > 3000) {
				delete messageLog[key];
			}
		}
		return false;
	}

	function processMessage(ele, ital=false){
		
		if (ele && ele.dataset.skip){
			return;
		}
		ele.dataset.skip = ++msgCount;
		
		var chatimg="";
		try{
			chatimg = ele.children[0].querySelector("img");
			if (!chatimg){
				chatimg = "";
			} else {
				chatimg = chatimg.src;
			}
		} catch(e){}
		
		
		if (ele.querySelector("[class*='DivTopGiverContainer']")){return;}
		
		var membership = "";
		var chatbadges = "";
		var rank = 0;
		
		var nameColor = "";
		
		try{
			var cb = ele.children[1].querySelectorAll("img[class*='ImgBadgeChatMessage'], img[class*='ImgCombineBadgeIcon']");
			if (cb.length){
				chatbadges = [];
				cb.forEach(cbimg =>{
					try {
						if (cbimg.src){
							chatbadges.push(cbimg.src);
							if (cbimg.src.includes("/moderator_")){
								if (!settings.nosubcolor) {
									nameColor = "#F5D5D1";
								}
							} else if (cbimg.src.includes("/moderater_")){
								if (!settings.nosubcolor) {
									nameColor = "#F5D5D1";
								}
							} else if (cbimg.src.includes("/sub_")){
								membership = getTranslation("subscriber", "SUBSCRIBER");
								if (!settings.nosubcolor) {
									nameColor = "#139F1D";
								}
							} else if (cbimg.src.includes("/subs_")){
								membership = getTranslation("subscriber", "SUBSCRIBER");
								if (!settings.nosubcolor) {
									nameColor = "#139F1D";
								}
							} else if (!rank && !nameColor && cbimg.src.includes("/grade_")){
								try {
									rank = parseInt(cbimg.nextElementSibling.innerText) || 1;
									if (!settings.nosubcolor) {
										if (rank>40){rank=40;}
										nameColor = lut[rank];
									}
								} catch(e){
								}
							}
						}
					} catch(e){}
				});
			}
		} catch(e){}
		
		var chatname = "";
		
		
		try {
			chatname = ele.querySelector("span[data-e2e='message-owner-name']").textContent;
			chatname = escapeHtml(chatname);
		} catch(e){}
		
		try{
			if (!chatname){
				if (ele.childNodes[1].childNodes[0].children.length){
					chatname = escapeHtml(ele.childNodes[1].childNodes[0].childNodes[0].innerText);
				} else {
					chatname = escapeHtml(ele.childNodes[1].childNodes[0].innerText);
				}
			}
			
		} catch(e){}
		
		var chatmessage = "";
		try {
			chatmessage = getAllContentNodes(ele.querySelector("div[class*='-DivUserInfo'], span[data-e2e='message-owner-name']").nextSibling);
			//live-shared-ui-chat-list-chat-message-comment
			if (!chatmessage){
				try {
					chatmessage = getAllContentNodes(ele.querySelector(".live-shared-ui-chat-list-chat-message-comment"));
				} catch(e){
					chatmessage = "";
				}
			}
		} catch(e){
			try {
				chatmessage = getAllContentNodes(ele.querySelector(".live-shared-ui-chat-list-chat-message-comment"));
			} catch(e){
				chatmessage = "";
			}
		}
		
		if (chatmessage == "Moderator"){
			chatmessage = "";
		}
		
		try {
			if (!chatmessage){
				var eles = ele.childNodes[1].childNodes;
				if (eles.length>1){
					for (var i  = 1; i<eles.length;i++){
						if (eles[i].nodeName === "#text"){
							chatmessage = escapeHtml(eles[i].textContent);
						} else if (settings.textonlymode){
							chatmessage = escapeHtml(eles[i].textContent);
						} else {
							chatmessage = eles[i].innerHTML;
						}
					}
				} else if (eles.length==1){
					for (var i  = 1; i<eles[0].childNodes.length;i++){
						if (settings.textonlymode){
							chatmessage = escapeHtml(eles[0].childNodes[i].textContent);
						} else {
							chatmessage = eles[0].childNodes[i].innerHTML;
						}
					}
				}
			}
		} catch(e){}
		
		if (!chatmessage && !chatbadges){
			return;
		} else if (chatmessage){
			chatmessage = chatmessage.trim();
		}
		
		if (chatmessage == "Moderator"){
			//console.log(ele);
			return;
			//alert("!!");
		}
		
		if (chatname && chatimg){
			savedavatars[chatname] = chatimg;
			if (savedavatars.length >300){
				var keys = Object.keys(savedavatars);
				for (var i=0;i<50;i++){ // delete 50 keys at a time, so I dont have to do this all the time.
					delete savedavatars[keys[0]];
				}
			}
		} else if (chatname){
			chatimg = savedavatars[chatname];
		} else if (chatmessage && (chatmessage==="----")){ // no chat name
			return;
		} 
		
		if (ital && chatmessage && (chatmessage==="joined")){ // no chat name
			if (!settings.capturejoinedevent){
				return;
			}
		}
		
		if (settings.customtiktokstate) {
			var channel = window.location.pathname.split("/@");
			if (channel.length>1){
				channel = channel[1].split("/")[0].trim();
			}
			if (!channel){return;}
			if (settings.customtiktokaccount && settings.customtiktokaccount.textsetting && ((settings.customtiktokaccount.textsetting.toLowerCase() !== channel.toLowerCase()) && (settings.customtiktokaccount.textsetting.toLowerCase() !== "@"+channel.toLowerCase()))) {
				return;
			} else if (!settings.customtiktokaccount) {
				return;
			}
		}
		
		if (isDuplicateMessage(chatname,chatmessage,ele)){
			//console.log("duplicate message; skipping");
			return;
		}
	  
		var data = {};
		data.chatname = chatname;
		data.chatbadges = chatbadges;
		data.backgroundColor = "";
		data.nameColor = nameColor;
		data.textColor = "";
		data.chatmessage = chatmessage;
		data.chatimg = chatimg;
		data.hasDonation = "";
		data.membership = membership;
		data.contentimg = "";
		// data.metaClass = "";
		data.textonly = settings.textonlymode || false;
		data.type = "tiktok";
		data.event = ital; // if an event or actual message
		
		//console.log(data);
		
		pushMessage(data);
	}
	
	
	function start() {
		
		var target = document.querySelector('[class*="DivChatMessageList"], .live-shared-ui-chat-list-scrolling-list'); //  [class^="LiveChat-"]
		if (!target){
			return;
		}
		
		if (window.location.href.includes("livecenter")){
			//
		} else if ((!(window.location.pathname.includes("@") && window.location.pathname.includes("live")))){
			return;
		}
		
		if (target.set123){
			return;
		} else {
			target.set123 = true;
		}
		
		console.log("STARTED SOCIAL STREAM");
		
		var MutationObserver = window.MutationObserver || window.WebKitMutationObserver;
		
		
		var onMutationsObserved = function(mutations) {
			mutations.forEach(function(mutation) {
				if (mutation.addedNodes.length) {
					for (var i = 0, len = mutation.addedNodes.length; i < len; i++) {
						try {
							if (mutation.addedNodes[i].dataset && (mutation.addedNodes[i].dataset.e2e == "chat-message")){
								setTimeout(function(ele2){
									if (ele2 && ele2.isConnected) {
										processMessage(ele2);
									}
								},300, mutation.addedNodes[i]);
							} else if (mutation.addedNodes[i].dataset.e2e){
								setTimeout(function(ele2){
									processMessage(ele2, true); // event
								},300, mutation.addedNodes[i]);
							} else if (settings.captureevents){
								setTimeout(function(ele2){
									processMessage(ele2); // donation?
								},300, mutation.addedNodes[i]);
							}
						} catch(e){}
					}
				}
			});
		};
		var config = { childList: true, subtree: false };
		
		var observer = new MutationObserver(onMutationsObserved);
		observer.observe(target, config);
		
		///////
	}	
	function start2() {
		
		
		if (isExtensionOn && document.querySelector("[class*='DivUnreadTipsContent']")){
			document.querySelector("[class*='DivScrollingListIndicatorWrap']").parentNode.scrollTop = document.querySelector("[class*='DivScrollingListIndicatorWrap']").parentNode.scrollHeight;
		}
		
		if (!settings.captureevents){
			return;
		}
		
		var target2 = document.querySelector('[class*="DivBottomStickyMessageContainer'); 
		
		if (!target2){
			return;
		}
		
		if (window.location.href.includes("livecenter")){
			//
		} else if ((!(window.location.pathname.includes("@") && window.location.pathname.includes("live")))){
			return;
		}
		
		if (target2.set123){
			return;
		} else {
			target2.set123 = true;
		}
		// class="tiktok-1dvdrb1-DivChatRoomMessage-StyledLikeMessageItem e12fsz0m0"
		console.log("STARTED SOCIAL STREAM - events");
		
		
		var onMutationsObserved2= function(mutations) {
			if (settings.captureevents){
				mutations.forEach(function(mutation) {
					if (mutation.addedNodes.length) {
						for (var i = 0, len = mutation.addedNodes.length; i < len; i++) {
							
							try {
								if ( mutation.addedNodes[i].nodeName == "DIV"){
									setTimeout(function(ele2){
										var typeOfEvent = ele2.querySelector("[data-e2e]");
										if (typeOfEvent){
											if (!settings.capturejoinedevent){
												if (typeOfEvent.dataset.e2e == "enter-message"){ // xxx joined the room
													return;
												}
											}
											//console.warn(ele2);
											processMessage(ele2.cloneNode(true), typeOfEvent.dataset.e2e || true); // event
										} else {
											//console.warn(ele2);
											processMessage(ele2.cloneNode(true), true); // event
										}
									},200, mutation.addedNodes[i]);
								} 
							} catch(e){}
						}
					}
				});
			}
		};
		
		var config2 = { childList: true, subtree: true };
		var observer2 = new MutationObserver(onMutationsObserved2);
		observer2.observe(target2, config2);
		
	}
	
	setInterval(start,2000);
	setInterval(start2,2000);
	

	var settings = {};
	var isExtensionOn = false;
	// settings.textonlymode
	// settings.captureevents

	
	try{
		chrome.runtime.sendMessage(chrome.runtime.id, { "getSettings": true }, function(response){  // {"state":isExtensionOn,"streamID":channel, "settings":settings}
			if ("settings" in response){
				settings = response.settings;
			}
			if ("state" in response){
				isExtensionOn = response.state;
			}
		});
	} catch(e){}


	var pokeMe = setInterval(function(){
		try{
			//if (chrome.runtime.id !== 1){
				chrome.runtime.sendMessage(chrome.runtime.id, { "pokeMe": true }, function(response){  // {"state":isExtensionOn,"streamID":channel, "settings":settings}
					console.log("POKED");
				});
			//}
		} catch(e){}
	},1000*60*59);
	
	var videosMuted = false;
	
	try{
		chrome.runtime.onMessage.addListener(
			function (request, sender, sendResponse) {
				try{
					if ("focusChat" == request){
						if (document.querySelector('.public-DraftEditorPlaceholder-inner')){
							document.querySelector(".public-DraftEditorPlaceholder-inner").focus();
							sendResponse(true);
							clearInterval(pokeMe);
							pokeMe = setInterval(function(){
								chrome.runtime.sendMessage(chrome.runtime.id, { "pokeMe": true }, function(response){  // {"state":isExtensionOn,"streamID":channel, "settings":settings}
									
								});
							},1000*60*60);
						} else if (document.querySelector("[contenteditable='true'][placeholder]")){
							
							document.querySelector("[contenteditable='true'][placeholder]").focus();
							sendResponse(true);
							setTimeout(function(){
								if (document.querySelector("[contenteditable='true'][placeholder]").textContent == ""){
									document.querySelector("[contenteditable='true'][placeholder]").innerHTML = "";
								}
							},300);
							clearInterval(pokeMe);
							pokeMe = setInterval(function(){
								chrome.runtime.sendMessage(chrome.runtime.id, { "pokeMe": true }, function(response){  // {"state":isExtensionOn,"streamID":channel, "settings":settings}
									
								});
							},1000*60*60);
						} else {
							sendResponse(false);
						}
						return;
					}
					if (typeof request === "object"){
						if ("state" in request){
							isExtensionOn = request.state;
						}
						if ("settings" in request){
							settings = request.settings;
							sendResponse(true);
							return;
						}
						if ("muteWindow" in request){
							
							if (request.muteWindow){
								clearInterval(videosMuted);
								
								videosMuted = setInterval(function(){
									document.querySelectorAll("video").forEach(v=>{
										v.muted = true;
										v.pause();
									});
								},1000);
								document.querySelectorAll("video").forEach(v=>{
									v.muted = true;
									v.pause();
								});
								sendResponse(true);
								return;
							} else {
								if (videosMuted){
									clearInterval(videosMuted);
									document.querySelectorAll("video").forEach(v=>{
										v.muted = false;
										v.play();
									});
								} else {
									clearInterval(videosMuted);
								}
								videosMuted = false;
								sendResponse(true);
								return;
							}
						}
					}
				} catch(e){	}
				
				sendResponse(false);
			}
		);
	} catch(e){}

	
})();