import os, time, datetime, struct, pathlib, sys
from datetime import datetime
from pathlib import Path
from pythonnet import load

load("coreclr")
import clr, System

##透過Clr引用系統標準函式
clr.AddReference('System.Collections')
from System.Collections.Generic import List

##宣告增加模組、DLL的路徑(windows可抓取當前路徑 Linux跟MAC需指定路徑)
sys.path.append(Path(pathlib.Path(__file__).parent.resolve()))
if sys.platform == "win32":
    os.add_dll_directory(Path(pathlib.Path(__file__).parent.resolve()))

##透過Clr引用YuantaSparkAPI.dll
##pythonnet引用元件不用加附檔名
try:
    clr.AddReference("YuantaSparkAPI")
except Exception as e:
    print(f"Error loading YuantaSparkAPI: {e}")
from YuantaOneAPI import YuantaSparkAPITrader, enumLogType, enumQuoteIndexType, enumMarketType, enumEnvironmentMode, enumQuoteFiveTickIndexType, FiveTickA

# 建立 API 物件
objYuantaSparkAPI = YuantaSparkAPITrader()
objYuantaSparkAPI.SetLogType(enumLogType.COMMON)

using System;
using System.Collections.Generic;
using System.Text;
using System.Threading;
using YuantaOneAPI;

YuantaSparkAPITrader objYuantaSparkAPI = new YuantaSparkAPITrader();
string Account = "S98875005091";
string Password = "1234";

enumEnvironmentMode enumEvenMode = enumEnvironmentMode.UAT;

objYuantaSparkAPI.OnResponse += objApi_OnResponse;
objYuantaSparkAPI.SetLogType(enumLogType.ALL);

objYuantaSparkAPI.Open(enumEvenMode);
Thread.Sleep(1000);
objYuantaSparkAPI.Login(Account, Password);
Thread.Sleep(1000);

List<FiveTickA> lstFiveTick = new List<FiveTickA>();

FiveTickA stkInfo = new FiveTickA
{
    MarketType = enumMarketType.TWSE,
    StockCode = "2330"
};

lstFiveTick.Add(stkInfo);

objYuantaSparkAPI.SubscribeFiveTickA(Account, lstFiveTick);
Thread.Sleep(2000);

def on_response(intMark, dwIndex, strIndex, objHandle, objValue):
    try:
        result = ''
        match intMark:
            case 0:  # 系統回應資訊
                result = str(objValue)

            case 1:  # 查詢回應資訊
                match strIndex:
                    case 'Login':
                        loginResult = objValue
                        status = loginResult.LoginStatus
                        strMsgCode = status.MsgCode # 訊息代碼
                        strMsgContent = status.MsgContent # 訊息內容
                        intCount = status.Count # 筆數
                        result = '{0},{1},帳號筆數:{2}\r\n'.format(strMsgCode,strMsgContent, str(intCount))
                        if strMsgCode == '0001' or strMsgCode == '00001' or intCount > 0 :
                            for i in objValue.LoginList:
                                result += f"{i.Account},{i.Name},{i.InvestorID},{i.SellerNo}\n"

            case 2:  # 訂閱回應資訊
                match strIndex:
                    case 'SubscribeFiveTickA':
                        # 五檔訂閱結果
                        fResult = objValue

                        result += '五檔訂閱結果:\r\n'
                        result += '{0},{1},{2},{3},'.format(fResult.Key,str(fResult.MarketType),fResult.StkCode,str(fResult.IndexFlag))
                        match fResult.IndexFlag:
                            case enumQuoteFiveTickIndexType.IndexFlag20:
                                #flag20 = FiveTick_Flag20()
                                flag20 = fResult.IndexFlag_20
                                result +='{0},{1},{2},{3},{4},{5},{6},{7},{8},{9}\r\n'.format(
                                    str(flag20.Price1),str(flag20.Price2),str(flag20.Price3),str(flag20.Price4),str(flag20.Price5),str(flag20.Vol1),str(flag20.Vol2),str(flag20.Vol3, str(flag20.Vol4),str(flag20.Vol5)))
                            case enumQuoteFiveTickIndexType.IndexFlag21:
                                #flag21 = FiveTick_Flag21()
                                flag21 = fResult.IndexFlag_21
                                result +='{0},{1},{2},{3},{4},{5},{6},{7},{8},{9}\r\n'.format(
                                    str(flag21.Price1),str(flag21.Price2),str(flag21.Price3),str(flag21.Price4),str(flag21.Price5),str(flag21.Vol1),str(flag21.Vol2),str(flag21.Vol3),str(flag21.Vol4),str(flag21.Vol5))
                            case enumQuoteFiveTickIndexType.IndexFlag42:
                                #flag42 = FiveTick_Flag42()
                                flag42 = fResult.IndexFlag_42
                                result +='{0},{1},{2},{3},{4},{5},{6},{7},{8},{9}\r\n'.format(
                                    str(flag42.Price1),str(flag42.Price2),str(flag42.Price3),str(flag42.Price4),str(flag42.Price5),str(flag42.Vol1),str(flag42.Vol2),str(flag42.Vol3),str(flag42.Vol4),str(flag42.Vol5))
                            case enumQuoteFiveTickIndexType.IndexFlag43:
                                #flag43 = FiveTick_Flag43()
                                flag43 = fResult.IndexFlag_43
                                result +='{0},{1},{2},{3},{4},{5},{6},{7},{8},{9}\r\n'.format(
                                    str(flag43.Price1),str(flag43.Price2),str(flag43.Price3),str(flag43.Price4),str(flag43.Price5),str(flag43.Vol1),str(flag43.Vol2),str(flag43.Vol3),str(flag43.Vol4),str(flag43.Vol5))
                            case enumQuoteFiveTickIndexType.IndexFlag50:
                                #flag50 = FiveTick_Flag50()
                                flag50 = fResult.IndexFlag_50
                                result +='{0},{1},{2},{3},{4},{5},{6},{7},{8},{9},{10},{11},{12},{13},{14},{15},{16},{17},{18},{19}\r\n'.format(
                                    str(flag50.BuyPrice1),str(flag50.BuyPrice2),str(flag50.BuyPrice3),str(flag50.BuyPrice4),str(flag50.BuyPrice5),str(flag50.BuyVol1),str(flag50.BuyVol2),str(flag50.BuyVol3),str(flag50.BuyVol4),
                                    str(flag50.BuyVol5),str(flag50.SellPrice1),str(flag50.SellPrice2),str(flag50.SellPrice3),str(flag50.SellPrice4),str(flag50.SellPrice5),str(flag50.SellVol1),str(flag50.SellVol2),str(flag50.SellVol3),
                                    str(flag50.SellVol4),str(flag50.SellVol5))
                            case enumQuoteFiveTickIndexType.IndexFlag51:
                                #flag51 = FiveTick_Flag51()
                                flag51 = fResult.IndexFlag_51
                                result +='{0},{1},{2},{3},{4},{5},{6},{7},{8},{9},{10},{11},{12},{13},{14},{15},{16},{17},{18},{19}\r\n'.format(
                                    str(flag51.BuyPrice1),str(flag51.BuyPrice2),str(flag51.BuyPrice3),str(flag51.BuyPrice4),str(flag51.BuyPrice5),str(flag51.BuyVol1),str(flag51.BuyVol2),str(flag51.BuyVol3),str(flag51.BuyVol4),
                                    str(flag51.BuyVol5),str(flag51.SellPrice1),str(flag51.SellPrice2),str(flag51.SellPrice3),str(flag51.SellPrice4),str(flag51.SellPrice5),str(flag51.SellVol1),str(flag51.SellVol2),str(flag51.SellVol3),
                                    str(flag51.SellVol4),str(flag51.SellVol5))
                            case _:
                                result += str(fResult.Value)+'\r\n'

        # 輸出結果
        if result:
            print('##================================================##\n')
            print(result)
    except Exception as error:
        print(f"處理回應時發生錯誤: {error}")

objYuantaSparkAPI.OnResponse += on_response
#測試環境帳號:UAT 正式環境:PROD
objYuantaSparkAPI.Open(enumEnvironmentMode.UAT)
time.sleep(2)
objYuantaSparkAPI.Login('S98875005091', '1234')
time.sleep(2)

fivetickList = List[FiveTickA]()
fivetick = FiveTickA()
fivetick.MarketType = enumMarketType.TWSE
fivetick.StockCode = '2330'
fivetickList.Add(fivetick)

#最佳五檔行情訂閱
objYuantaSparkAPI.SubscribeFiveTickA('S98875005091',fivetickList)

#測試環境傳送後要休息一下
time.sleep(10)

# 保持程式運行
while True:
    time.sleep(1)

void objApi_OnResponse(int intMark, uint dwIndex, string strIndex, object objHandle, object objValue)
{
    string strResult = "";
    try
    {
        if (intMark == 0)
        {
            Console.WriteLine(Convert.ToString(objValue));
            return;
        }

        if (intMark == 1)
        {
            if (strIndex == "Login")
            {
                var result = (LoginResult)objValue;

                string strMsgCode = result.LoginStatus.MsgCode;
                string strMsgContent = result.LoginStatus.MsgContent;
                int intCount = result.LoginStatus.Count;

                strResult += $"{strMsgCode}, {strMsgContent}{Environment.NewLine}";
                if (strMsgCode == "0001" || strMsgCode == "00001")
                {
                    strResult += $"帳號筆數: {intCount.ToString()}{Environment.NewLine}";
                    result.LoginList.ForEach(r => strResult += $"{r.Account},{r.Name},{r.InvestorID},{r.SellerNo}\r\n");

                }
                else
                {
                    Account = "";
                }

                Console.WriteLine("\n======================");
                Console.WriteLine(strResult.ToString());
                Console.WriteLine("======================\n");
                return;
            }
        }

        if (intMark == 2)
        {
            if (strIndex == "SubscribeFiveTickA")
            {
                var result = (FiveTickAResult)objValue;

                try
                {
                    strResult += "五檔訂閱結果: \r\n";
                    strResult += $"{result.Key},{result.MarketType},{result.StkCode},{result.IndexFlag}:";

                    string datas = string.Empty;
                    dynamic flag1 = null;
                    dynamic flag2 = null;

                    switch (result.IndexFlag)
                    {
                        case enumQuoteFiveTickIndexType.IndexFlag20:
                            flag1 = result.IndexFlag_20;
                            break;
                        case enumQuoteFiveTickIndexType.IndexFlag21:
                            flag1 = result.IndexFlag_21;
                            break;
                        case enumQuoteFiveTickIndexType.IndexFlag42:
                            flag1 = result.IndexFlag_42;
                            break;
                        case enumQuoteFiveTickIndexType.IndexFlag43:
                            flag1 = result.IndexFlag_43;
                            break;
                        case enumQuoteFiveTickIndexType.IndexFlag50:
                            flag2 = result.IndexFlag_50;
                            break;
                        case enumQuoteFiveTickIndexType.IndexFlag51:
                            flag2 = result.IndexFlag_51;
                            break;
                        default:
                            datas += result.Value;
                            break;
                    }

                    if (flag1 != null)
                    {
                        datas = string.Empty;
                        datas += $"{flag1.Price1},{flag1.Price2},{flag1.Price3},{flag1.Price4},{flag1.Price5}," +
                                $"{flag1.Vol1},{flag1.Vol2},{flag1.Vol3},{flag1.Vol4},{flag1.Vol5}";
                    }
                    else if (flag2 != null)
                    {
                        datas = string.Empty;
                        datas += $"{flag2.BuyPrice1},{flag2.BuyPrice2},{flag2.BuyPrice3},{flag2.BuyPrice4},{flag2.BuyPrice5}," +
                                $"{flag2.BuyVol1},{flag2.BuyVol2},{flag2.BuyVol3},{flag2.BuyVol4},{flag2.BuyVol5}," +
                                $"{flag2.SellPrice1},{flag2.SellPrice2},{flag2.SellPrice3},{flag2.SellPrice4},{flag2.SellPrice5}," +
                                $"{flag2.SellVol1},{flag2.SellVol2},{flag2.SellVol3},{flag2.SellVol4},{flag2.SellVol5}";
                    }

                    strResult += datas + "\r\n";
                }
                catch
                {
                    strResult = "";
                }

                Console.WriteLine("\n======================");
                Console.WriteLine(strResult);
                Console.WriteLine("======================\n");
                return;
            }

            Console.WriteLine($"[{strIndex}] {Convert.ToString(objValue)}");
        }
    }
    catch (Exception exc)
    {
        Console.WriteLine("OnResponse Error: " + exc.Message);
    }
}