qr

import requests
import urllib.parse

class NewQRLogin:
    API_URL = "https://api.lrtt.icu/secondaryQrCodeLogin.do"
    HEADERS = {
        "android_lite": {
            "User-Agent": "LLA/2.12.0 SKR-H0 9",
            "X-Line-Application": "ANDROIDLITE\t2.12.0\tAndroid OS\t9;SECONDARY"
        },
        "android": {
            "User-Agent": "Line/10.6.2",
            "X-Line-Application": "ANDROID\t10.6.2\tAndroid OS\t10"
        },
        "ios_ipad": {
            "User-Agent": "Line/10.1.1",
            "X-Line-Application": "IOSIPAD\t10.1.1\tiPhone 8\t11.2.5"
        },
        "ios": {
            "User-Agent": "Line/10.1.1",
            "X-Line-Application": "IOS\t10.1.1\tiPhone 8\t11.2.5"
        },
        "chrome": {
            "User-Agent": "MozilX-Line-Application/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/79.0.3945.79 Safari/537.36",
            "X-Line-Application": "CHROMEOS\t2.3.2\tChrome OS\t1"
        },
        "desktopwin": {
            "User-Agent": "Line/5.12.3",
            "X-Line-Application": "DESKTOPWIN\t5.21.3\tWindows\t10"
        },
        "desktopmac": {
            "User-Agent": "Line/5.12.3",
            "X-Line-Application": "DESKTOPMAC\t5.21.3\tMAC\t10.15"
        }
    }

    def parseLogin(this, loginInfo):
        return (loginInfo["token"], loginInfo["certificate"])

    def loginWithQrCode(self, header, certificate="", callback=lambda output: print(output)):
        assert header in self.HEADERS, "invaild header"
        resp = requests.post(self.API_URL + "/login?" + urllib.parse.urlencode({"custom_headers": self.HEADERS[header], "certificate": certificate}))
        res = resp.json()
        if resp.status_code != 200:
            raise Exception(res)
        callback("Login URL: %s" % (res["url"]))

        while "token" not in res:
            resp = requests.post(self.API_URL + res["callback"])
            res = resp.json()
            if resp.status_code != 200:
                raise Exception(res)

            if "pin" in res:
                callback("Input PIN: %s" % (res["pin"]))

        return self.parseLogin(res)

    def loginQrCodeWithWebPinCode(self, header, certificate="", callback=lambda output: print(output)):
        resp = requests.post(self.API_URL + "/login?" + urllib.parse.urlencode({"header": header, "certificate": certificate}))
        res = resp.json()
        if resp.status_code != 200:
            raise Exception(res)
        callback("ก่อนล็อคอินอ่านก่อน :\nline://nv/settings/privacy กดลิ้งแล้วปิด Letter Sealing ก่อนครับ '-'")
        callback(f'เว็บไซต์ยืนยันPIN : {res["web"]}\nแนะนำให้เปิดใน google chrome')
        callback(f'กดลิ้งนี้ภายใน2นาที : {res["url"]}')

        while "token" not in res:
            resp = requests.post(self.API_URL + res["callback"])
            res = resp.json()
            if resp.status_code != 200:
                raise Exception(res)

        return self.parseLogin(res)

    def loginWithQrCode_kicker(self, header, no, number, certificate="", callback=lambda output: print(output)):
        assert header in self.HEADERS, "invaild header"
        resp = requests.post(self.API_URL + "/login?" + urllib.parse.urlencode({"custom_headers": self.HEADERS[header], "certificate": certificate}))
        res = resp.json()
        if resp.status_code != 200:
            raise Exception(res)
        callback(f'login kicker [{no}/{number}]: {res["url"]}')

        while "token" not in res:
            resp = requests.post(self.API_URL + res["callback"])
            res = resp.json()
            if resp.status_code != 200:
                raise Exception(res)

            if "pin" in res:
                callback(f'pin kicker [{no}/{number}]: {res["pin"]}')

        return self.parseLogin(res)

    def loginWithQrCode_antjs(self, header, certificate="", callback=lambda output: print(output)):
        assert header in self.HEADERS, "invaild header"
        resp = requests.post(self.API_URL + "/login?" + urllib.parse.urlencode({"custom_headers": self.HEADERS[header], "certificate": certificate}))
        res = resp.json()
        if resp.status_code != 200:
            raise Exception(res)
        callback(f'login antjs: {res["url"]}')

        while "token" not in res:
            resp = requests.post(self.API_URL + res["callback"])
            res = resp.json()
            if resp.status_code != 200:
                raise Exception(res)

            if "pin" in res:
                callback(f'pin antjs: {res["pin"]}')

        return self.parseLogin(res)

    def loginQrCodeWithWebPinCode_Antjs(self, header, certificate="", callback=lambda output: print(output)):
        resp = requests.post(self.API_URL + "/login?" + urllib.parse.urlencode({"header": header, "certificate": certificate}))
        res = resp.json()
        if resp.status_code != 200:
            raise Exception(res)
        callback(f'pin antjs : {res["web"]}')
        callback(f'login antjs : {res["url"]}')

        while "token" not in res:
            resp = requests.post(self.API_URL + res["callback"])
            res = resp.json()
            if resp.status_code != 200:
                raise Exception(res)

        return self.parseLogin(res)

if __name__ == "__main__":
    qrv2 = NewQRLogin()
    token, cert = qrv2.loginQrCodeWithWebPinCode("desktopmac")
    print("Access Token: " + token)
    print("Certificate: " + cert)