Files
my-assistant/.agents/skills/dingtalk-api/scripts/list-user-parent-departments.ts

179 lines
5.0 KiB
TypeScript
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
// 获取指定用户的所有父部门列表
// 用法: ts-node scripts/list-user-parent-departments.ts <userId> [--debug]
// 需要设置环境变量 DINGTALK_APP_KEY 和 DINGTALK_APP_SECRET
export {};
const https = require('https');
const { default: dingtalkOauth2_1_0, GetAccessTokenRequest } = require('@alicloud/dingtalk/oauth2_1_0');
const { Config } = require('@alicloud/openapi-client');
interface SuccessResult {
success: boolean;
userId: string;
parentIdList: number[];
}
interface ErrorResult {
success: false;
userId: string;
error: {
code: string;
message: string;
};
}
function createConfig(): any {
const config = new Config({});
config.protocol = "https";
config.regionId = "central";
return config;
}
async function getAccessToken(appKey: string, appSecret: string): Promise<string> {
const config = createConfig();
const client = new dingtalkOauth2_1_0(config);
const request = new GetAccessTokenRequest({
appKey: appKey,
appSecret: appSecret,
});
try {
const response = await client.getAccessToken(request);
const accessToken = response.body?.accessToken;
if (!accessToken) {
throw new Error('获取 access_token 失败:响应中未包含 token');
}
return accessToken;
} catch (err: any) {
throw new Error(`获取 access_token 失败: ${err.message || err}`);
}
}
async function dingtalkRequest(accessToken: string, method: string, path: string, body?: any): Promise<any> {
return new Promise((resolve, reject) => {
const options = {
hostname: 'oapi.dingtalk.com',
path: `${path}?access_token=${accessToken}`,
method,
headers: {
'Content-Type': 'application/json',
} as Record<string, string>,
};
const req = https.request(options, (res: any) => {
let data = '';
res.on('data', (chunk: string) => data += chunk);
res.on('end', () => {
try {
const parsed = JSON.parse(data);
if (parsed.errcode !== undefined && parsed.errcode !== 0) {
reject({ code: parsed.errcode, message: parsed.errmsg });
} else if (res.statusCode && res.statusCode >= 400) {
reject(parsed);
} else {
resolve(parsed);
}
} catch {
reject(new Error(`Invalid JSON response: ${data}`));
}
});
});
req.on('error', reject);
if (body) req.write(JSON.stringify(body));
req.end();
});
}
async function listUserParentDepartments(accessToken: string, userId: string, debug: boolean = false): Promise<void> {
try {
const response = await dingtalkRequest(
accessToken,
'POST',
'/topapi/v2/department/listparentbyuser',
{ userid: userId }
);
if (debug) {
console.error('\n=== 调试信息 ===');
console.error('完整响应:', JSON.stringify(response, null, 2));
console.error('==============\n');
}
const result: SuccessResult = {
success: true,
userId: userId,
parentIdList: response.result || [],
};
console.log(JSON.stringify(result, null, 2));
} catch (err: any) {
const errorResult: ErrorResult = {
success: false,
userId: userId,
error: {
code: err.code || 'UNKNOWN_ERROR',
message: err.message || err.msg || JSON.stringify(err),
}
};
console.error(JSON.stringify(errorResult, null, 2));
process.exit(1);
}
}
async function main(): Promise<void> {
const args = process.argv.slice(2);
const debug = args.includes('--debug');
const filteredArgs = args.filter(arg => arg !== '--debug');
const appKey = process.env.DINGTALK_APP_KEY;
const appSecret = process.env.DINGTALK_APP_SECRET;
if (!appKey || !appSecret) {
console.error(JSON.stringify({
success: false,
error: {
code: 'MISSING_CREDENTIALS',
message: '缺少钉钉应用凭证,请设置环境变量 DINGTALK_APP_KEY 和 DINGTALK_APP_SECRET',
usage: 'export DINGTALK_APP_KEY=your-app-key && export DINGTALK_APP_SECRET=your-app-secret && ts-node scripts/list-user-parent-departments.ts <userId> [--debug]'
}
}, null, 2));
process.exit(1);
}
if (filteredArgs.length < 1) {
console.error(JSON.stringify({
success: false,
error: {
code: 'INVALID_ARGUMENTS',
message: '参数错误:需要提供用户 IDuserId',
usage: 'ts-node scripts/list-user-parent-departments.ts <userId> [--debug]'
}
}, null, 2));
process.exit(1);
}
const userId = filteredArgs[0];
try {
console.error('正在获取 access_token...');
const accessToken = await getAccessToken(appKey, appSecret);
console.error('access_token 获取成功,正在获取用户的父部门列表...');
await listUserParentDepartments(accessToken, userId, debug);
} catch (err: any) {
console.error(JSON.stringify({
success: false,
error: {
code: 'AUTH_FAILED',
message: err.message || '认证失败',
}
}, null, 2));
process.exit(1);
}
}
main();