import { StockData, Market, StockPrefix, RawStockResponse, AStockData, HKStockData, USStockData } from './types.js';
const BASE_URL = 'https://qt.gtimg.cn';
// 根据市场获取前缀
function getPrefix(market: Market): StockPrefix {
switch (market) {
case 'sh': return 'sh';
case 'sz': return 'sz';
case 'hk': return 'r_hk';
case 'us': return 's_us';
}
}
// 获取股票数据
export async function fetchStock(code: string, market: Market): Promise<StockData> {
const prefix = getPrefix(market);
const url = `${BASE_URL}/?q=${prefix}${code}`;
const response = await fetch(url);
if (!response.ok) {
throw new Error(`Failed to fetch stock data: ${response.statusText}`);
}
const text = await response.text();
const rawData = parseRawResponse(text);
if (!rawData) {
throw new Error(`No data returned for ${prefix}${code}`);
}
return parseStockData(rawData, market);
}
// 解析原始响应
function parseRawResponse(text: string): RawStockResponse | null {
// 格式: v_sh600000="..." 或 v_s_usAAPL="..."
const match = text.match(/v_(sh|sz|r_hk|s_us)([a-zA-Z0-9._]+)="([^"]+)"/);
if (!match) return null;
return {
key: match[1] + match[2],
data: match[3]
};
}
// 解析股票数据
function parseStockData(raw: RawStockResponse, market: Market): StockData {
// 不过滤空字段，保留原始索引
const fields = raw.data.split('~');
if (fields.length < 10) {
throw new Error(`Invalid stock data: too few fields`);
}
const marketCode = parseInt(fields[0]);
// 判断市场类型
if (market === 'sh' || market === 'sz' || marketCode === 1 || marketCode === 51) {
return parseAStock(fields, market === 'sz' ? 'sz' : 'sh');
} else if (market === 'hk' || marketCode === 100) {
return parseHKStock(fields);
} else if (market === 'us' || marketCode === 200) {
return parseUSStock(fields);
}
throw new Error(`Unknown market type: ${marketCode}`);
}
// 解析A股数据
function parseAStock(fields: string[], market: 'sh' | 'sz'): AStockData {
const getField = (idx: number): string => fields[idx] || '';
const parseNum = (idx: number, defaultVal: number = 0): number => {
const val = getField(idx);
return val ? parseFloat(val) || defaultVal : defaultVal;
};
const result: AStockData = {
market,
name: getField(1),
code: getField(2),
currentPrice: parseNum(3),
yesterdayClose: parseNum(4),
open: parseNum(5),
volume: parseNum(6),
outside: parseNum(7),
inside: parseNum(8),
datetime: getField(30),
change: parseNum(31),
changePercent: parseNum(32),
high: parseNum(33),
low: parseNum(34),
amount: parseNum(37),
turnoverRate: parseNum(38),
peTtm: parseNum(39),
amplitude: parseNum(43),
totalMarketCap: parseNum(44),
floatMarketCap: parseNum(45),
volumeRatio: parseNum(49),
avgPrice: parseNum(51),
peDynamic: parseNum(52),
peStatic: parseNum(53),
floatingShares: parseNum(72),
totalShares: parseNum(73),
currency: getField(82) || 'CNY',
};
// 解析10档买卖盘
const asks: Array<{ price: number; volume: number }> = [];
const bids: Array<{ price: number; volume: number }> = [];
// 卖盘: 9-10(价格,量), 11-12, 13-14, 15-16, 17-18 (卖5-卖1)
for (let i = 9; i <= 18; i += 2) {
if (fields[i] && fields[i + 1]) {
asks.push({
price: parseFloat(fields[i]),
volume: parseInt(fields[i + 1])
});
}
}
// 买盘: 19-20(价格,量), 21-22, 23-24, 25-26, 27-28 (买1-买5)
for (let i = 19; i <= 28; i += 2) {
if (fields[i] && fields[i + 1]) {
bids.push({
price: parseFloat(fields[i]),
volume: parseInt(fields[i + 1])
});
}
}
result.bidAsk = { asks, bids };
return result;
}
// 解析港股数据
function parseHKStock(fields: string[]): HKStockData {
const getField = (idx: number): string => fields[idx] || '';
const parseNum = (idx: number, defaultVal: number = 0): number => {
const val = getField(idx);
return val ? parseFloat(val) || defaultVal : defaultVal;
};
return {
market: 'hk',
name: getField(1),
code: getField(2),
currentPrice: parseNum(3),
yesterdayClose: parseNum(4),
open: parseNum(5),
volume: parseNum(6),
datetime: getField(30),
change: parseNum(31),
changePercent: parseNum(32),
high: parseNum(33),
low: parseNum(34),
amount: parseNum(37),
pe: parseNum(39),
floatingShares: parseNum(69),
totalShares: parseNum(70),
currency: getField(75) || 'HKD',
};
}
// 解析美股数据
function parseUSStock(fields: string[]): USStockData {
const getField = (idx: number): string => fields[idx] || '';
const parseNum = (idx: number, defaultVal: number = 0): number => {
const val = getField(idx);
return val ? parseFloat(val) || defaultVal : defaultVal;
};
return {
market: 'us',
name: getField(1),
code: getField(2),
currentPrice: parseNum(3),
change: parseNum(4),
changePercent: parseNum(5),
volume: parseNum(6),
amount: parseNum(7),
marketCap: parseNum(8),
currency: 'USD',
};
}
// 获取多只股票数据
export async function fetchStocks(codes: Array<{ code: string; market: Market }>): Promise<StockData[]> {
const results: StockData[] = [];
for (const { code, market } of codes) {
try {
const data = await fetchStock(code, market);
results.push(data);
} catch (error) {
console.error(`Failed to fetch ${market}${code}:`, error);
}
}
return results;
}
