@sokeai/cli 1.0.41 → 1.0.54

package/skills/oss-upload/upload.py

@@ -0,0 +1,355 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+阿里云OSS文件上传工具
+通过授客AI接口获取签名后上传文件
+仅使用Python标准库，无需安装第三方依赖
+"""
+
+import os
+import sys
+import json
+import urllib.request
+import urllib.parse
+import urllib.error
+import mimetypes
+import uuid
+import time
+
+
+MAX_FILE_SIZE = 2 * 1024 * 1024 * 1024  # 2GB
+
+def load_config():
+    """加载配置文件获取access_token"""
+    config_path = os.path.expanduser("~/.soke-cli/config.json")
+    if not os.path.exists(config_path):
+        print("错误: 配置文件不存在，请先运行 'soke-cli config init'")
+        sys.exit(1)
+
+    with open(config_path, 'r', encoding='utf-8') as f:
+        config = json.load(f)
+
+    if 'UserToken' not in config or not config['UserToken']:
+        print("错误: 未找到用户令牌，请先运行 'soke-cli auth login' 登录")
+        sys.exit(1)
+
+    return config
+
+
+def get_upload_signature(access_token, api_base_url, max_retries=3):
+    """从接口获取上传签名和配置"""
+    params = urllib.parse.urlencode({
+        'lang': 'zh',
+        'utcoffset': '-28800'
+    })
+    signature_url = f"{api_base_url}/skills/uploadFile/signature?{params}"
+
+    headers = {
+        'Authorization': f'{access_token}',
+        'Content-Type': 'application/x-www-form-urlencoded;charset=UTF-8'
+    }
+
+    for attempt in range(max_retries):
+        try:
+            if attempt > 0:
+                print(f"正在重试... ({attempt + 1}/{max_retries})")
+                time.sleep(2 ** attempt)  # 指数退避: 2s, 4s, 8s
+            else:
+                print("正在获取上传签名...")
+
+            req = urllib.request.Request(signature_url, method='POST', headers=headers)
+
+            with urllib.request.urlopen(req, timeout=30) as response:
+                response_data = response.read().decode('utf-8')
+                result = json.loads(response_data)
+
+            # 检查响应格式
+            # 支持多种格式:
+            # 1. {"code": 0, "message": "success", "data": {...}}
+            # 2. {"code": "get_success", "status": "ok", "message": "获取成功", "data": {...}}
+            # 3. {"success": true, "err_code": 0, "err_message": "success", "data": {...}}
+            code = result.get('code')
+            status = result.get('status', '')
+            success = result.get('success')
+            err_code = result.get('err_code')
+
+            # 判断是否成功
+            is_success = (
+                (isinstance(code, int) and code == 0) or  # 数字格式
+                (isinstance(code, str) and code in ['get_success', 'success']) or  # 字符串格式
+                (status == 'ok') or  # 通过status判断
+                (success is True and err_code == 0)  # success格式
+            )
+
+            if not is_success:
+                print(f"✗ 获取签名失败: {result.get('message', result.get('err_message', '未知错误'))}")
+                return None
+
+            # 提取签名数据
+            data = result.get('data', {})
+            if not data:
+                print("✗ 响应数据为空")
+                return None
+
+            print("✓ 签名获取成功")
+            return data
+
+        except urllib.error.HTTPError as e:
+            error_msg = f"HTTP {e.code} {e.reason}"
+            try:
+                error_body = e.read().decode('utf-8')
+            except:
+                error_body = ""
+
+            # 503/502/504 等服务端错误可以重试
+            if e.code in [502, 503, 504] and attempt < max_retries - 1:
+                print(f"✗ {error_msg} - 服务暂时不可用，将自动重试...")
+                continue
+            else:
+                print(f"✗ HTTP错误: {error_msg}")
+                if error_body:
+                    print(f"响应内容: {error_body}")
+                if e.code == 503:
+                    print("\n提示: 503错误通常表示服务端暂时不可用，请稍后重试")
+                    print("可能的原因:")
+                    print("  1. 服务正在维护或重启")
+                    print("  2. 服务负载过高")
+                    print("  3. 网络连接问题")
+                return None
+
+        except urllib.error.URLError as e:
+            if attempt < max_retries - 1:
+                print(f"✗ 网络错误: {str(e.reason)} - 将自动重试...")
+                continue
+            else:
+                print(f"✗ 请求失败: {str(e.reason)}")
+                return None
+
+        except json.JSONDecodeError as e:
+            print(f"✗ 响应解析失败: {str(e)}")
+            return None
+
+        except Exception as e:
+            print(f"✗ 发生错误: {str(e)}")
+            return None
+
+    print(f"✗ 已重试 {max_retries} 次，仍然失败")
+    return None
+
+
+def create_multipart_form_data(fields, file_path):
+    """创建multipart/form-data格式的请求体"""
+    boundary = f'----WebKitFormBoundary{uuid.uuid4().hex[:16]}'
+
+    body = []
+
+    # 添加表单字段
+    for key, value in fields.items():
+        body.append(f'--{boundary}'.encode())
+        body.append(f'Content-Disposition: form-data; name="{key}"'.encode())
+        body.append(b'')
+        body.append(str(value).encode())
+
+    # 添加文件
+    filename = os.path.basename(file_path)
+    mimetype = mimetypes.guess_type(file_path)[0] or 'application/octet-stream'
+
+    body.append(f'--{boundary}'.encode())
+    body.append(f'Content-Disposition: form-data; name="file"; filename="{filename}"'.encode())
+    body.append(f'Content-Type: {mimetype}'.encode())
+    body.append(b'')
+
+    with open(file_path, 'rb') as f:
+        body.append(f.read())
+
+    body.append(f'--{boundary}--'.encode())
+    body.append(b'')
+
+    content_type = f'multipart/form-data; boundary={boundary}'
+    body_bytes = b'\r\n'.join(body)
+
+    return content_type, body_bytes
+
+
+def upload_file(file_path, object_name, signature_data, corp_id):
+    """上传文件到OSS"""
+    if not os.path.exists(file_path):
+        print(f"错误: 文件不存在: {file_path}")
+        sys.exit(1)
+
+    # 从签名数据中提取上传配置
+    upload_url = signature_data.get('host')
+    access_key_id = signature_data.get('accessid') or signature_data.get('OSSAccessKeyId')
+    policy = signature_data.get('policy')
+    signature = signature_data.get('signature')
+    callback = signature_data.get('callback', '')
+    dir_prefix = signature_data.get('dir', '')
+
+    if not all([upload_url, access_key_id, policy, signature]):
+        print("✗ 签名数据不完整")
+        print(f"签名数据: {json.dumps(signature_data, indent=2, ensure_ascii=False)}")
+        return False
+
+    # 获取文件信息
+    file_size = os.path.getsize(file_path)
+    file_name = os.path.basename(file_path)
+    file_ext = os.path.splitext(file_name)[1].lstrip('.')  # 移除点号，如 pdf
+
+    # 构建完整的对象路径：CorpID/dir_prefix/object_name
+    if corp_id:
+        # 如果有CorpID，添加为顶层目录
+        if dir_prefix:
+            # 移除dir_prefix开头的斜杠（如果有）
+            dir_prefix = dir_prefix.lstrip('/')
+            full_path = f"{corp_id}/{dir_prefix}{object_name}"
+        else:
+            full_path = f"{corp_id}/{object_name}"
+    else:
+        # 如果没有CorpID，使用原逻辑
+        if dir_prefix and not object_name.startswith(dir_prefix):
+            full_path = dir_prefix + object_name
+        else:
+            full_path = object_name
+
+    # 准备表单数据
+    form_fields = {
+        'key': full_path,
+        'policy': policy,
+        'OSSAccessKeyId': access_key_id,
+        'signature': signature,
+        'success_action_status': '200',
+    }
+
+    # 如果有callback，添加到表单
+    if callback:
+        form_fields['callback'] = callback
+
+    # 显示上传信息
+    print(f"\n正在上传文件: {file_path}")
+    print(f"文件名称: {file_name}")
+    print(f"文件大小: {file_size} 字节")
+    print(f"文件后缀: {file_ext if file_ext else '(无后缀)'}")
+    if corp_id:
+        print(f"企业ID: {corp_id}")
+    print(f"目标路径: {full_path}")
+    print(f"上传地址: {upload_url}")
+
+    try:
+        # 创建multipart/form-data请求体
+        content_type, body = create_multipart_form_data(form_fields, file_path)
+
+        headers = {
+            'Content-Type': content_type,
+            'Content-Length': str(len(body))
+        }
+
+        req = urllib.request.Request(upload_url, data=body, headers=headers, method='POST')
+
+        with urllib.request.urlopen(req, timeout=300) as response:
+            response_code = response.getcode()
+            response_body = response.read().decode('utf-8')
+
+        if response_code == 200:
+            file_url = f"{upload_url.rstrip('/')}/{full_path}"
+
+            print(f"\n✓ 上传成功!")
+            print(f"=" * 60)
+            print(f"文件信息:")
+            print(f"  - 文件名: {file_name}")
+            print(f"  - 大小: {file_size} 字节")
+            print(f"  - 后缀: {file_ext if file_ext else '(无后缀)'}")
+            print(f"  - OSS路径: {full_path}")
+            print(f"  - 访问URL: {file_url}")
+            print(f"=" * 60)
+
+            # 如果有callback响应，打印出来
+            if response_body:
+                try:
+                    callback_result = json.loads(response_body)
+                    if callback_result:
+                        print(f"\n回调响应: {json.dumps(callback_result, indent=2, ensure_ascii=False)}")
+                except:
+                    pass
+
+            return True
+        else:
+            print(f"\n✗ 上传失败!")
+            print(f"状态码: {response_code}")
+            print(f"响应内容: {response_body}")
+            return False
+
+    except urllib.error.HTTPError as e:
+        print(f"\n✗ 上传失败: HTTP {e.code} {e.reason}")
+        try:
+            error_body = e.read().decode('utf-8')
+            print(f"响应内容: {error_body}")
+        except:
+            pass
+        return False
+    except urllib.error.URLError as e:
+        print(f"\n✗ 上传失败: {str(e.reason)}")
+        return False
+    except Exception as e:
+        print(f"\n✗ 发生错误: {str(e)}")
+        import traceback
+        traceback.print_exc()
+        return False
+
+
+def main():
+    """主函数"""
+    if len(sys.argv) < 2:
+        print("用法: python upload.py <本地文件路径> [OSS对象名]")
+        print("示例: python upload.py ./document.pdf uploads/2024/document.pdf")
+        sys.exit(1)
+
+    file_path = sys.argv[1]
+
+    # 检查文件是否存在
+    if not os.path.exists(file_path):
+        print(f"错误: 文件不存在: {file_path}")
+        sys.exit(1)
+
+    # 检查是否是文件（不是目录）
+    if not os.path.isfile(file_path):
+        print(f"错误: 路径不是文件: {file_path}")
+        sys.exit(1)
+
+    # 检查文件大小（不能超过2GB）
+    file_size = os.path.getsize(file_path)
+    if file_size > MAX_FILE_SIZE:
+        print(f"错误: 文件大小超过限制")
+        print(f"  当前大小: {file_size / (1024 * 1024 * 1024):.2f} GB ({file_size} 字节)")
+        print(f"  最大限制: 2 GB ({MAX_FILE_SIZE} 字节)")
+        sys.exit(1)
+
+    # 如果未指定对象名，使用文件名
+    if len(sys.argv) >= 3:
+        object_name = sys.argv[2]
+    else:
+        object_name = os.path.basename(file_path)
+
+    # 加载配置
+    config = load_config()
+    access_token = config.get('UserToken')
+    api_base_url = config.get('APIBaseURL', 'https://app2651.eapps.dingtalkcloud.com/NewSoke')
+    corp_id = config.get('CorpID', '')
+
+    if not corp_id:
+        print("警告: 未找到CorpID配置，文件将上传到根目录")
+        print("建议在配置文件中添加 CorpID 字段")
+
+    # 获取上传签名
+    signature_data = get_upload_signature(access_token, api_base_url)
+    if not signature_data:
+        print("✗ 无法获取上传签名")
+        sys.exit(1)
+
+    # 上传文件
+    success = upload_file(file_path, object_name, signature_data, corp_id)
+
+    sys.exit(0 if success else 1)
+
+
+if __name__ == '__main__':
+    main()

package/skills/soke-assign/README.md

@@ -0,0 +1,310 @@
+# 授客指派管理 Skill
+
+通用的学习资源指派系统，支持将课程、考试、学习地图等学习资源指派给部门和用户。
+
+## 功能概述
+
+这个 skill 提供了完整的指派管理功能，包括：
+
+- 📋 **部门查询**: 查询可用的部门列表，支持搜索和分页
+- 👥 **用户查询**: 查询可用的用户列表，支持多种搜索方式
+- ✅ **创建指派**: 将学习资源指派给部门或用户
+- 📊 **查询指派**: 查询已创建的指派记录，支持多维度筛选
+
+## 支持的模块
+
+- **course** - 课程指派
+- **exam** - 考试指派
+- **learning_map** - 学习地图指派
+- 其他模块（根据系统配置）
+
+## 支持的指派类型
+
+- **dept（部门）**: 部门下所有成员都可以学习
+- **user（用户）**: 只有指定的用户可以学习
+
+## 快速开始
+
+### 创建课程指派
+
+```bash
+# 1. 查询部门列表，获取部门ID
+soke-cli assign +list-departments --keyword "研发"
+
+# 2. 查询用户列表（可选）
+soke-cli assign +list-users --dept-id "130110584"
+
+# 3. 创建指派
+soke-cli assign +create \
+  --module "course" \
+  --target-id "E08464D0-D567-4A01-8FAB-BD41AFFDB271" \
+  --dept-ids "130110584" \
+  --dept-names "研发部" \
+  --learn-type "required" \
+  --message-content "新课程已为您安排，请及时学习！"
+
+# 4. 验证指派
+soke-cli assign +list \
+  --module "course" \
+  --target-id "E08464D0-D567-4A01-8FAB-BD41AFFDB271"
+```
+
+### 创建考试指派
+
+```bash
+# 指派考试给部门
+soke-cli assign +create \
+  --module "exam" \
+  --target-id "EXAM-UUID-12345" \
+  --dept-ids "130110584" \
+  --dept-names "研发部" \
+  --learn-type "required" \
+  --message-content "年度考核考试已开放，请按时参加！"
+```
+
+## 主要命令
+
+### 查询命令
+| 命令 | 说明 |
+|------|------|
+| `+list-departments` | 查询部门列表（获取部门ID和名称） |
+| `+list-users` | 查询用户列表（获取用户ID和姓名） |
+| `+list` | 查询指派列表 |
+
+### 操作命令
+| 命令 | 说明 |
+|------|------|
+| `+create` | 创建指派 |
+
+## 使用场景
+
+### 场景1: 为新员工培训课程创建指派
+
+```bash
+# 1. 查询部门列表
+soke-cli assign +list-departments --keyword "研发"
+# 输出：部门ID: 130110584, 部门名称: 研发部
+
+# 2. 查询该部门的用户
+soke-cli assign +list-users --dept-id "130110584"
+# 输出：用户列表
+
+# 3. 创建课程指派（指派给整个部门）
+soke-cli assign +create \
+  --module "course" \
+  --target-id "E08464D0-D567-4A01-8FAB-BD41AFFDB271" \
+  --dept-ids "130110584" \
+  --dept-names "研发部" \
+  --learn-type "required" \
+  --message-content "新员工培训课程已为您安排，请在一周内完成学习！"
+
+# 4. 验证指派是否成功
+soke-cli assign +list \
+  --module "course" \
+  --target-id "E08464D0-D567-4A01-8FAB-BD41AFFDB271"
+```
+
+### 场景2: 为特定用户创建指派
+
+```bash
+# 1. 搜索用户
+soke-cli assign +list-users --keyword "张三" --search-target "name"
+# 输出：用户ID: 1241453468881932, 姓名: 张三
+
+# 2. 创建指派
+soke-cli assign +create \
+  --module "course" \
+  --target-id "E08464D0-D567-4A01-8FAB-BD41AFFDB271" \
+  --user-ids "1241453468881932" \
+  --user-names "张三" \
+  --learn-type "required"
+```
+
+### 场景3: 批量指派给多个部门
+
+```bash
+# 1. 查询多个部门
+soke-cli assign +list-departments --keyword "战区"
+
+# 2. 批量创建指派
+soke-cli assign +create \
+  --module "course" \
+  --target-id "E08464D0-D567-4A01-8FAB-BD41AFFDB271" \
+  --dept-ids "130110584,130110585,130110586" \
+  --dept-names "安徽战区,江苏战区,浙江战区" \
+  --learn-type "required"
+```
+
+### 场景4: 创建多维度指派
+
+```bash
+# 同时指派给部门和用户
+soke-cli assign +create \
+  --module "exam" \
+  --target-id "EXAM-UUID-12345" \
+  --dept-ids "130110584,130110585" \
+  --dept-names "研发部,测试部" \
+  --user-ids "1241453468881932" \
+  --user-names "张三" \
+  --learn-type "required" \
+  --message-content "年度考核考试已开放，请按时参加！"
+```
+
+### 场景5: 查询和统计指派情况
+
+```bash
+# 查询特定课程的所有指派
+soke-cli assign +list \
+  --module "course" \
+  --target-id "E08464D0-D567-4A01-8FAB-BD41AFFDB271"
+
+# 查询部门指派
+soke-cli assign +list \
+  --module "course" \
+  --target-id "E08464D0-D567-4A01-8FAB-BD41AFFDB271" \
+  --assign-type "dept"
+
+# 查询用户指派
+soke-cli assign +list \
+  --module "course" \
+  --target-id "E08464D0-D567-4A01-8FAB-BD41AFFDB271" \
+  --assign-type "user"
+
+# 按关键词搜索
+soke-cli assign +list \
+  --module "course" \
+  --keyword "研发"
+```
+
+## 参数说明
+
+### 查询部门列表参数
+
+| 参数 | 类型 | 必填 | 默认值 | 说明 |
+|------|------|------|--------|------|
+| --page | int | 否 | 1 | 页码 |
+| --limit | int | 否 | 10 | 每页数量 |
+| --parent-id | string | 否 | 1 | 父部门ID |
+| --keyword | string | 否 | - | 搜索关键词 |
+
+### 查询用户列表参数
+
+| 参数 | 类型 | 必填 | 默认值 | 说明 |
+|------|------|------|--------|------|
+| --dept-id | string | 否 | - | 部门ID |
+| --keyword | string | 否 | - | 搜索关键词 |
+| --search-target | string | 否 | name | 搜索目标：name/phone/job_number |
+| --page | int | 否 | 1 | 页码 |
+| --limit | int | 否 | 20 | 每页数量 |
+
+### 创建指派参数
+
+| 参数 | 类型 | 必填 | 默认值 | 说明 |
+|------|------|------|--------|------|
+| --module | string | 是 | - | 模块类型：course/exam/learning_map |
+| --target-id | string | 是 | - | 目标ID |
+| --learn-type | string | 否 | required | 学习类型：required/optional |
+| --dept-ids | string | 否 | - | 部门ID列表，逗号分隔 |
+| --dept-names | string | 否 | - | 部门名称列表，逗号分隔 |
+| --user-ids | string | 否 | - | 用户ID列表，逗号分隔 |
+| --user-names | string | 否 | - | 用户名称列表，逗号分隔 |
+| --message-content | string | 否 | 默认消息 | 通知消息内容 |
+| --message-switch | int | 否 | 1 | 消息开关：0关闭/1开启 |
+
+### 查询指派列表参数
+
+| 参数 | 类型 | 必填 | 默认值 | 说明 |
+|------|------|------|--------|------|
+| --target-id | string | 否 | - | 目标ID |
+| --module | string | 否 | - | 模块类型 |
+| --learn-type | string | 否 | - | 学习类型 |
+| --assign-type | string | 否 | - | 指派类型：dept/user/group/position/role |
+| --keyword | string | 否 | - | 搜索关键词 |
+| --page | int | 否 | 1 | 页码 |
+| --limit | int | 否 | 10 | 每页数量 |
+
+## 注意事项
+
+1. **指派前准备**
+   - 确保目标资源已创建（如课程已发布）
+   - 使用 `+list-departments` 和 `+list-users` 查询可用的部门和用户
+   - 记录部门ID和用户ID用于创建指派
+
+2. **批量指派**
+   - 支持同时指派给多个对象
+   - 使用逗号分隔多个ID和名称
+   - ID和名称的顺序要一一对应
+
+3. **必修 vs 选修**
+   - **必修**（required）：用户必须完成学习
+   - **选修**（optional）：用户可选择性学习
+
+4. **消息通知**
+   - 默认开启消息通知（message-switch=1）
+   - 可自定义消息内容（message-content）
+   - 可通过 message-switch=0 关闭通知
+
+5. **权限要求**
+   - 需要具有指派的读写权限
+   - 只能查询和操作权限范围内的部门和用户
+
+6. **搜索方式**
+   - 部门搜索：按部门名称搜索
+   - 用户搜索：支持按姓名、手机号、工号搜索
+
+## 错误处理
+
+### 权限不足
+
+```bash
+# 错误信息: Permission denied
+# 解决方案: 检查是否已登录，并确认账号有相应权限
+soke-cli auth login
+```
+
+### 目标资源不存在
+
+```bash
+# 错误信息: Target not found
+# 解决方案: 检查目标ID是否正确，确认资源已创建
+```
+
+### 部门或用户不存在
+
+```bash
+# 错误信息: Department/User not found
+# 解决方案: 使用 +list-departments 或 +list-users 查询正确的ID
+```
+
+### 参数格式错误
+
+```bash
+# 错误信息: Invalid parameter format
+# 解决方案: 检查ID和名称列表是否一一对应，逗号分隔是否正确
+```
+
+## 相关文档
+
+- [SKILL.md](./SKILL.md) - 完整的 Skill 文档和 API 参考
+- [使用文档](../../docs/ASSIGN_USAGE.md) - 详细的使用指南和示例
+- [课程管理 Skill](../soke-course/SKILL.md) - 课程创建和管理
+- [考试管理 Skill](../soke-exam/SKILL.md) - 考试创建和管理
+- [共享配置 Skill](../soke-shared/SKILL.md) - 认证和配置管理
+- [API 接口文档](../../docs/API_PROXY_INTERFACES_2.0.md) - 完整的 API 接口说明
+
+## 工作流程图
+
+```
+指派工作流
+├── 1. 查询可用对象
+│   ├── +list-departments (查询部门)
+│   └── +list-users (查询用户)
+├── 2. 创建指派
+│   └── +create (指派给部门/用户/群组/职位/角色)
+└── 3. 验证和查询
+    └── +list (查询指派记录)
+```
+
+## 版本历史
+
+- v1.0.0 (2026-05-28): 初始版本，支持部门查询、用户查询、创建指派、查询指派列表