forked from QuantumNous/new-api
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathapi.ts
More file actions
112 lines (91 loc) · 3.63 KB
/
Copy pathapi.ts
File metadata and controls
112 lines (91 loc) · 3.63 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
/*
Copyright (C) 2023-2026 QuantumNous
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as
published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
For commercial licensing, please contact support@quantumnous.com
*/
import { api } from '@/lib/api'
import { buildQueryParams } from './lib/utils'
import type {
GetLogsParams,
GetLogsResponse,
GetLogStatsParams,
GetLogStatsResponse,
GetMidjourneyLogsParams,
GetTaskLogsParams,
UserInfo,
} from './types'
// ============================================================================
// Generic API Helpers
// ============================================================================
function buildApiPath(endpoint: string, isAdmin: boolean): string {
return isAdmin ? endpoint : `${endpoint}/self`
}
async function fetchLogs<T>(
endpoint: string,
params: T,
isAdmin: boolean
): Promise<GetLogsResponse> {
const paramRecord = params as unknown as Record<string, unknown>
const queryParams = buildQueryParams({
p: paramRecord.p || 1,
page_size: paramRecord.page_size || 20,
...params,
})
const path = buildApiPath(endpoint, isAdmin)
const res = await api.get(`${path}?${queryParams}`)
return res.data
}
async function fetchLogStats<T>(
endpoint: string,
params: T,
isAdmin: boolean
): Promise<GetLogStatsResponse> {
const queryParams = buildQueryParams(
params as unknown as Record<string, unknown>
)
const path = buildApiPath(endpoint, isAdmin)
const res = await api.get(`${path}/stat?${queryParams}`)
return res.data
}
// ============================================================================
// Common Log APIs
// ============================================================================
export const getAllLogs = (params: GetLogsParams = {}) =>
fetchLogs('/api/log', params, true)
export const getUserLogs = (
params: Omit<GetLogsParams, 'username' | 'channel'> = {}
) => fetchLogs('/api/log', params, false)
export const getLogStats = (params: GetLogStatsParams = {}) =>
fetchLogStats('/api/log', params, true)
export const getUserLogStats = (
params: Omit<GetLogStatsParams, 'username' | 'channel'> = {}
) => fetchLogStats('/api/log', params, false)
export async function getUserInfo(
userId: number
): Promise<{ success: boolean; message?: string; data?: UserInfo }> {
const res = await api.get(`/api/user/${userId}`)
return res.data
}
// ============================================================================
// MjProxy (Drawing) Logs API
// ============================================================================
export const getAllMidjourneyLogs = (params: GetMidjourneyLogsParams) =>
fetchLogs('/api/mj', params, true)
export const getUserMidjourneyLogs = (params: GetMidjourneyLogsParams) =>
fetchLogs('/api/mj', params, false)
// ============================================================================
// Task Logs API
// ============================================================================
export const getAllTaskLogs = (params: GetTaskLogsParams) =>
fetchLogs('/api/task', params, true)
export const getUserTaskLogs = (params: GetTaskLogsParams) =>
fetchLogs('/api/task', params, false)