1
function promisify(original, customPromise) {
2
  // 使用自定义Promise实现或原生Promise
3
  const PromiseImpl = customPromise || Promise;
4

5
  return function(...args) {
6
    return new PromiseImpl((resolve, reject) => {
7
      // 创建回调函数
8
      const callback = (err, ...values) => {
9
        if (err) return reject(err);
10

11
        // 多参数处理：单值直接返回，多值返回对象
12
        resolve(values.length <= 1 ? values : values);
13
      };
14

15
      // 添加回调到参数列表
16
      try {
17
        // 保留this上下文
18
        original.apply(this, [...args, callback]);
19
      } catch (error) {
20
        // 处理同步错误
21
        reject(error);
22
      }
23
    });
24
  };
25
}
26

27
// 高级版本：支持多参数命名
28
promisify.custom = Symbol('promisify.custom');
29

30
function promisifyAdvanced(original, customPromise) {
31
  const PromiseImpl = customPromise || Promise;
32

33
  // 处理自定义promisify函数
34
  if (original[promisify.custom]) {
35
    return function(...args) {
36
      return PromiseImpl.resolve().then(() =>
37
        original[promisify.custom](...args)
38
      );
39
    };
40
  }
41

42
  return function(...args) {
43
    return new PromiseImpl((resolve, reject) => {
44
      // 防止多次调用
45
      let called = false;
46

47
      const callback = (err, ...values) => {
48
        if (called) return;
49
        called = true;
50

51
        if (err) return reject(err);
52

53
        // 处理多参数回调
54
        if (values.length === 0) return resolve(undefined);
55
        if (values.length === 1) return resolve(values);
56

57
        // 创建结果对象
58
        const result = {};
59
        const callbackParams = original.toString().match(/function\s.*?$$(.*?)$$/).split(',');
60
        const callbackArgNames = callbackParams.slice(-1).replace(/\s+/g, '').split(')').split('=').split(' ');
61

62
        if (callbackArgNames.length > 1) {
63
          // 使用命名参数
64
          const [, ...paramNames] = callbackArgNames;
65
          paramNames.forEach((name, index) => {
66
            result[name] = values[index];
67
          });
68
        } else {
69
          // 使用索引参数
70
          values.forEach((value, index) => {
71
            result[`arg${index}`] = value;
72
          });
73
        }
74

75
        resolve(result);
76
      };
77

78
      try {
79
        // 调用原始函数
80
        original.apply(this, [...args, callback]);
81
      } catch (error) {
82
        if (!called) {
83
          called = true;
84
          reject(error);
85
        }
86
      }
87
    });
88
  };
89
}