ArkCard-Linebot/main.py

import uvicorn
import fastapi
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import HTMLResponse
from linebot import LineBotApi, WebhookHandler
from linebot.models import (
    MessageEvent, TextMessage, TextSendMessage, FollowEvent, TemplateSendMessage, ButtonsTemplate, URITemplateAction,
)
import dataset
import requests
import json
import qrcode
# from PIL import Image
# import base64, io
from random import randrange
import models
import datetime as dt

app = fastapi.FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=['*'],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# bot config
line_bot_api = LineBotApi("SJT7VPT4RMQFLcS27jQBy3FcC24gtDrkcwJWZ5Xzqesr5T78LOKudHEJzt0k3b2S7n4KPwf27J7DVz2c8NQ4plSaaQylEeB1cYrfejaE/RPG/lCIQBYe4iBTzo26s4i2PcmT89837per/lTyvhVIKAdB04t89/1O/w1cDnyilFU=")
handler = WebhookHandler("411ae3ef7e766739ed2c2c27b249d010")

# callback event
@app.post("/callback")
async def callback(request: fastapi.Request):
    signature = request.headers['X-Line-Signature']
    body = await request.body()
    handler.handle(body.decode('utf-8'), signature)
    return 'OK'

# follow event
@handler.add(FollowEvent)
def handle_follow(event):
    # get user id when follow
    real_user_id = event.source.user_id

    # db connect and search
    db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/arkcard?charset=utf8mb4')
    table = db['users']
    result1 = table.find_one(userid=real_user_id)

    # 都存在db的話
    if result1:
        db.close()
        line_bot_api.reply_message(
            event.reply_token,
            TextSendMessage(text='很高興再見到您！'))

    # 建立全新使用者
    else:
        # create user account api
        url = 'https://nft-api-staging.joyso.io/api/v1/accounts'
        headers = {'Authorization': 'Basic bmZ0OmMxOTEzOWMzYjM3YjdjZWU3ZmY3OTFiZGU3NzdjZWNl'}

        # setup for temp use (unique id)
        rand_num = str(randrange(99999))
        user_id = event.source.user_id + rand_num
        data = 'uid=' + user_id
        r = requests.post(url=url, headers=headers, data=data)

        # extract the account address
        dict_str = json.loads(r.text)
        user_account = dict_str['account']
        user_address = user_account['address']

        # generate qr code from user id
        qr = qrcode.QRCode(
            version=1,
            box_size=10,
            border=5)
        qr.add_data(user_address)
        qr.make(fit=True)
        img_qr = qr.make_image(fill='black', back_color='white')
        filename = "/var/www/ArkCard-Linebot/ArkCard-web/qrcode/" + real_user_id + '.png'
        img_save = img_qr.save(filename)

        # add to db
        data = dict(userid=real_user_id, useraddress=user_address)
        table.insert(data)

        db.close()

        line_bot_api.reply_message(
            event.reply_token,
            TextSendMessage(text='歡迎加入好友'))

# message handler
@handler.add(MessageEvent, message=TextMessage)
def message(event):
    if '我要發送' in event.message.text:
        button_template_message = ButtonsTemplate(
            title=' ',
            text='點擊並打開收藏的NFT，可以選擇想要發送的NFT給對方！',
            actions=[
                URITemplateAction(
                    label='打開發送頁',
                    uri='https://ark.cards/collect.html?' + event.source.user_id),])
        line_bot_api.reply_message(
            event.reply_token,
            TemplateSendMessage(
                alt_text="Receive",
                template=button_template_message))

    elif '我要接收' in  event.message.text:
        button_template_message = ButtonsTemplate(
            title=' ',
            text='點擊並打開接收頁面，即可分享接收地址給對方！',
            actions=[
                URITemplateAction(
                    label='打開接收頁',
                    uri='https://ark.cards/qr-code.html?' + event.source.user_id),])
        line_bot_api.reply_message(
            event.reply_token,
            TemplateSendMessage(
                alt_text="Receive",
                template=button_template_message))

    elif 'NFT商店' in event.message.text:
        button_template_message = ButtonsTemplate(
            title=' ',
            text='點擊並打開NFT商品頁，就可以購買您所想要的NFT商品哦！',
            actions=[
                URITemplateAction(
                    label='打開NFT商品頁',
                    uri='https://ark.cards/shop.html?' + event.source.user_id),])
        line_bot_api.reply_message(
            event.reply_token,
            TemplateSendMessage(
                alt_text="Receive",
                template=button_template_message))

    elif 'NFT收藏' in event.message.text:
        button_template_message = ButtonsTemplate(
            title=' ',
            text='點擊並打開收藏的NFT，可以查看收到的NFT！',
            actions=[
                URITemplateAction(
                    label='打開收藏頁',
                    uri='https://ark.cards/collect.html?' + event.source.user_id), ])
        line_bot_api.reply_message(
            event.reply_token,
            TemplateSendMessage(
                alt_text="Receive",
                template=button_template_message))
    else:
        button_template_message = ButtonsTemplate(
            title=' ',
            text='更多的服務內容，歡迎請上我們的官網！',
            actions=[
                URITemplateAction(
                    label='ArkCard的官網',
                    uri='https://ark.cards'),])
        line_bot_api.reply_message(
            event.reply_token,
            TemplateSendMessage(
                alt_text="Receive",
                template=button_template_message))

# nft collection api
@app.get("/collection/{userid}")
def collection(userid):
    # 連到ownership表單去找，並回傳json
    # db connect
    db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/arkcard?charset=utf8mb4')
    table1 = db['ownership']
    nft = {}
    i = 0
    if not table1.find_one(userid=userid):
        db.close()
        return "error: user don't have any nft"
    else:
        results = db.query("SELECT nftid, amount FROM arkcard.ownership WHERE userid IN (SELECT userid FROM arkcard.ownership WHERE userid = '"+userid+"');")
        for item in results:
            nft[i] = item
            i += 1
        db.close()
        return nft

# receive handler
@app.get("/receive/{userid}")
def receive(userid):
    # 確定要傳送的對象，到user去找到它
    # db connect
    db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/arkcard?charset=utf8mb4')
    table = db['users']

    if not table.find_one(userid=userid):
        db.close()
        return "ERROR: User Not Found"
    else:
        result = table.find_one(userid=userid)
        db.close()
        return {"userid": result['userid'], "address": result['address']}

# send handler
@app.post("/send")
async def receive(userModel : models.TransactionNft):
    # 從網頁上選定好商品，在選定好數量後輸入收方地址;將把該商品轉到收方，並存下交易記錄
    # 從ownership找到該userid的擁有人，再把擁有人userid改為收方，並在trans留下記錄
    # db connect
    db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/arkcard?charset=utf8mb4')
    table1 = db['users']
    table2 = db['ownership']
    table3 = db['transactions']

    # find the userid from the selected nft
    fromuser = userModel.fromuser
    address = userModel.address
    result1 = table1.find_one(address=address)
    result2 = table1.find_one(userid=fromuser)
    # 例:
    # {
    #   "fromuser": "a01",
    #   "address": "0x4cd0ea8b1bdb5ab9249d96ccf3d8a0d3ada2bc76",
    #   "dic": {"10":"2", "20":"3"}
    # }
    # 第一個為nftid, 第二個為amount，可以同時輸入多組nft給同一人
    the_list = userModel.dic
    fr_token = "from token"
    _hash = "hash"

    # confirm if the user exist
    # 如果沒有這個接受者，回傳錯誤
    if not result1:
        db.close()
        return {'msg': 'user not found'}
    # 確認發送方的nft夠不夠發放
    else:
        for nftid, amount in the_list.items():
            print("nft: "+nftid+"，數量:"+ amount)
            # 直接針對取到的值去做確認數量和更新表單，最後再記錄
            # 確認sender的總數
            result3 = db.query('SELECT SUM(amount) AS total FROM arkcard.ownership WHERE userid="' + fromuser + '" AND nftid ="' + nftid + '";')

            # 把總數撈出來，存到total
            for row in result3:
                rows = row
            total = rows['total']

            # 如果總數不夠，就報錯回傳
            if total == None or total < int(amount):
                db.close()
                return {"messge: NFT not enough"}

            # 總數夠，在ownership刪去數量，並再新增ownership一筆
            else:
                # 今天到這，需要找出怎麼扣除總數的方式，目前找到的總數，會更新到資料欄位，導致原本才100個的，會被更新成499個，因為全加到一個
                # 找接收者的userid
                to_user = result1['userid']
                old_address = result2['address']
                print(total)
                new_amount = total - int(amount)
                # 更新原user總數
                result4 = db.query('DELETE FROM arkcard.ownership WHERE userid="' + fromuser + '" AND nftid ="' + nftid + '";')
                result5 = db.query('UPDATE arkcard.ownership SET amount = amount - "' + amount + '" WHERE userid = "' + fromuser + '" AND nftid = "' + nftid + '" LIMIT 1;')
                new_data1 = dict(userid=fromuser, address=old_address, nftid=nftid, amount=new_amount)
                table2.insert(new_data1)
                # 增加新user數量
                new_data2 = dict(userid=to_user, address=address, nftid=nftid, amount=int(amount))
                table2.insert(new_data2)

                # 在transactions記錄from和to, 以及時間等等
                now = dt.datetime.now()
                trans_data = dict(to_token=address, time=now, amount=int(amount), userid=fromuser, from_token=fr_token, hash=_hash)
                table3.insert(trans_data)
        db.close()
        return {"messge: NFT 夠"}

# shop handler
@app.get("/shop")
def shop():
    # 為了顯示正確的nft網頁和數量，直接顯示nft表出來
    # db connect
    db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/arkcard?charset=utf8mb4')
    result1 = db.query('SELECT nftid, launched, amount, type, title, context FROM arkcard.nft WHERE launched="y" AND nftid NOT IN(SELECT nftid FROM arkcard.nft_to_event);')
    nfts = {}
    i = 0
    for row in result1:
        nfts[i] = row
        i += 1
    return nfts
    db.close()

@app.post("/buy")
async def buy(userModel : models.BuyNft):
    # db connect
    db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/arkcard?charset=utf8mb4')
    table1 = db['users']
    table2 = db['ownership']
    table3 = db['nft']
    table4 = db['transactions']
    shops = "shop"
    hash = "hash"

    # input
    the_list = userModel.dic
    address = userModel.address
    # 例:
    # {
    #   "address": "d04",
    #   "dic": {"1":"2","2":"3"}
    # }

    result1 = table1.find_one(address=address)

    # 找不到該使用者就回報錯誤
    if not result1:
        db.close()
        return "該用戶不存在！如果有疑問，請洽網站的服務信箱！"
    else:
        for nftid, amount in the_list.items():
            # 先user找到該使用者的userid留著備用
            userid = result1['userid']

            # 更新到ownership
            owner_data = [dict(userid=userid, address=address, nftid=nftid, amount=int(amount))]
            table2.insert_many(owner_data)

            # 更新nft數量
            result3 = table3.find_one(nftid=nftid)
            nft_total = result3['amount']
            total = int(nft_total) - int(amount)
            nft_data = dict(nftid=nftid, amount=int(total))
            table3.update(nft_data, ['nftid'])

            # 最後再來處理transactions
            now = dt.datetime.now()
            trans_data = dict(to_token=address, time=now, amount=int(amount), userid=userid, from_token=shops, hash=hash)
            table4.insert(trans_data)
        db.close()
        return "您已購買成功！"

@app.post("/event")
async def nftdrops(userModel : models.NftDrops):
    # db connect
    db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/arkcard?charset=utf8mb4')
    table1 = db['nft_drop_list']

    # input
    userid = userModel.userid
    email = userModel.email

    the_current_event = eventPeriod()

    if the_current_event == 0:
        return "目前沒有任何進行中的活動"
    else:
        # 再針對有上架的 event 填寫客戶到 nftdroplist
        now = dt.datetime.now()
        result1 = table1.find_one(userid=userid)
        if not result1:
            for key, item in the_current_event.items():
                table1.insert(dict(eventid=item, userid=userid, email=email, time=now))
            db.close()
            return "完成加入"
        else:
            db.close()
            return "該資料已存在"

@app.post("/eventPeriod")
def eventPeriod():
    # db connect
    db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/arkcard?charset=utf8mb4')
    # 設定event始終期間，與上架與否
    table3 = db['event']
    now = dt.datetime.now()
    # 先去找 event 表，確認在期間之內
    result = db.query("SELECT * FROM arkcard.event WHERE start < NOW() and NOW() < end and launched='yes';")
    rows = {}
    i = 0
    for row in result:
        rows[i] = row['eventid']
        i += 1
    return rows
    db.close()




if __name__ == '__main__':
    uvicorn.run("main:app", host="0.0.0.0", port=8228, reload=True,ssl_context=('/etc/letsencrypt/live/ark.cards/fullchain.pem', '/etc/letsencrypt/live/ark.cards/privkey.pem'))