WorkOrder.php 72 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008
  1. <?php
  2. namespace app\api\controller;
  3. use app\common\controller\Api;
  4. use app\job\ImageJob;
  5. use app\service\AIGatewayService;
  6. use app\service\ImageService;
  7. use think\App;
  8. use think\Db;
  9. use think\Exception;
  10. use think\Log;
  11. use think\Queue;
  12. use think\queue\job\Redis;
  13. use think\Request;
  14. use app\api\controller\Common;
  15. class WorkOrder extends Api{
  16. protected $noNeedLogin = ['*'];
  17. protected $noNeedRight = ['*'];
  18. public function index(){echo '访问成功';}
  19. /**
  20. * AI队列入口处理 出图接口
  21. * 此方法处理图像转换为文本的请求,将图像信息存入队列以供后续处理。
  22. */
  23. public function imageToText()
  24. {
  25. $params = $this->request->param();
  26. $service = new ImageService();
  27. $service->handleImage($params);
  28. $this->success('任务成功提交至队列');
  29. }
  30. /**
  31. * task_id:查询获取任务图片
  32. */
  33. public function GetImageStatus(){
  34. $params = $this->request->param();
  35. $taskId = $params['task_id'];
  36. if (empty($taskId)) {
  37. $res = [
  38. 'code' => 1,
  39. 'msg' => '任务ID不能为空'
  40. ];
  41. return json($res);
  42. }
  43. //从Redis中获取任务状态
  44. $redis = getTaskRedis();
  45. $taskData = $redis->get("img_to_img_task:{$taskId}");
  46. if (!$taskData) {
  47. $taskData = $redis->get("text_to_image_task:{$taskId}");
  48. }
  49. if (!$taskData) {
  50. $res = [
  51. 'code' => 1,
  52. 'msg' => '任务不存在或已过期',
  53. ];
  54. return json($res);
  55. }
  56. $taskInfo = json_decode($taskData, true);
  57. if (!empty($taskInfo['image'])) {
  58. $taskInfo['image'] = Common::ossFullUrl((string)$taskInfo['image']);
  59. }
  60. if (!empty($taskInfo['image_url'])) {
  61. $taskInfo['image_url'] = Common::ossFullUrl((string)$taskInfo['image_url']);
  62. }
  63. $res = [
  64. 'code' => 0,
  65. 'msg' => '查询成功',
  66. 'data' => $taskInfo
  67. ];
  68. return json($res);
  69. }
  70. /**
  71. * 支持的AI任务类型枚举(前端传入的status_val需在该列表内)
  72. * 键:前端传入值,值:方法后缀( '前端传入值' => '方法后缀')
  73. */
  74. private static $AI_TASK_TYPES = [
  75. '图生文' => 'ImgToText',
  76. '文生文' => 'TextToText',
  77. '文生图' => 'TextToImg',
  78. '图生图' => 'ImgToImg'
  79. ];
  80. /**
  81. * AI模型接口统一调用入口
  82. * @description 接收前端AI请求,校验任务类型合法性,分发至对应处理方法,统一异常捕获
  83. * @return \think\response\Json 标准化JSON响应
  84. */
  85. public function callAIModelApi()
  86. {
  87. try {
  88. // 1. 获取并校验入参
  89. $params = $this->request->param();
  90. $statusVal = $this->validateAndGetStatusVal($params);
  91. // 2. 映射并校验处理方法
  92. $method = $this->getHandleMethod($statusVal);
  93. // 3. 执行对应处理逻辑并返回响应
  94. return $this->$method($params);
  95. } catch (\InvalidArgumentException $e) {
  96. // 参数/方法异常(用户侧错误)
  97. return $this->jsonResponse(1, $e->getMessage());
  98. } catch (\Throwable $e) {
  99. // 系统异常(服务侧错误)
  100. \think\Log::error('AI接口处理异常:' . $e->getMessage() . ' | 任务类型:' . ($params['status_val'] ?? '未知') . ' | 异常行:' . $e->getLine());
  101. return $this->jsonResponse(1, '服务异常,请稍后重试');
  102. }
  103. }
  104. // -------------------------- 私有核心方法(通用逻辑) --------------------------
  105. /**
  106. * 校验并获取合法的任务类型
  107. * @param array $params 前端入参
  108. * @return string 合法的status_val
  109. * @throws \InvalidArgumentException 任务类型不合法时抛出
  110. */
  111. private function validateAndGetStatusVal(array $params): string
  112. {
  113. $statusVal = trim($params['status_val'] ?? '');
  114. // 空值校验
  115. if (empty($statusVal)) {
  116. throw new \InvalidArgumentException('任务类型不能为空');
  117. }
  118. // 合法性校验
  119. if (!array_key_exists($statusVal, self::$AI_TASK_TYPES)) {
  120. throw new \InvalidArgumentException('不支持的任务类型:' . $statusVal);
  121. }
  122. return $statusVal;
  123. }
  124. /**
  125. * 获取任务对应的处理方法名
  126. * @param string $statusVal 合法的任务类型
  127. * @return string 处理方法名(如handleAiImgToText)
  128. * @throws \InvalidArgumentException 方法未实现时抛出
  129. */
  130. private function getHandleMethod(string $statusVal): string
  131. {
  132. $methodSuffix = self::$AI_TASK_TYPES[$statusVal];
  133. $method = 'handleAi' . $methodSuffix;
  134. if (!method_exists($this, $method)) {
  135. throw new \InvalidArgumentException('任务类型暂未实现:' . $statusVal);
  136. }
  137. return $method;
  138. }
  139. /**
  140. * 通用JSON响应封装
  141. * @param int $code 响应码(0=成功,1=失败)
  142. * @param string $msg 响应信息
  143. * @param array $data 响应数据(可选)
  144. * @return \think\response\Json
  145. */
  146. private function jsonResponse(int $code, string $msg, array $data = []): \think\response\Json
  147. {
  148. $response = [
  149. 'code' => $code,
  150. 'msg' => $msg,
  151. 'time' => date('Y-m-d H:i:s')
  152. ];
  153. if (!empty($data)) {
  154. $response['data'] = $data;
  155. }
  156. return json($response);
  157. }
  158. /**
  159. * 任务类接口统一响应(带task_id的场景)
  160. * @param array $result 业务处理结果(需包含success字段)
  161. * @param string $failMsg 失败提示语
  162. * @return \think\response\Json
  163. */
  164. private function jsonTaskResponse(array $result, string $failMsg = '任务提交失败'): \think\response\Json
  165. {
  166. $isSuccess = isset($result['success']) && $result['success'];
  167. $data = $isSuccess ? ['task_id' => $result['task_id'] ?? ''] : [];
  168. $msg = $isSuccess ? ($result['message'] ?? '提交成功') : ($result['message'] ?? $failMsg);
  169. return $this->jsonResponse($isSuccess ? 0 : 1, $msg, $data);
  170. }
  171. // -------------------------- 业务处理方法(按任务类型拆分) --------------------------
  172. /**
  173. * 图生文任务处理:提交队列并返回提示
  174. * @param array $params 前端入参
  175. * @return \think\response\Json
  176. */
  177. private function handleAiImgToText(array $params): \think\response\Json
  178. {
  179. (new ImageService())->handleImgToText($params);
  180. return $this->jsonResponse(0, '正在优化提示词,请稍等.....');
  181. }
  182. /**
  183. * 文生文任务处理:生成话术并返回内容(支持产品内容更新)
  184. * @param array $params 前端入参
  185. * @return \think\response\Json
  186. */
  187. private function handleAiTextToText(array $params): \think\response\Json
  188. {
  189. //构造生成提示词
  190. $promptTemplate = "\n请根据上述内容生成一段完整的话术,要求:\n"
  191. . "1. 内容必须是连贯的一段话,不要使用列表、分隔线或其他结构化格式\n"
  192. . "2. 不要包含非文本元素的描述\n"
  193. . "3. 不要添加任何额外的引导语、解释或开场白\n"
  194. . "4. 禁忌:不添加无关形容词,不修改产品核心信息,语言流畅自然";
  195. $prompt = ($params['prompt'] ?? '') . $promptTemplate;
  196. // 调用服务层生成内容
  197. $result = (new ImageService())->handleTextToText(
  198. $params['status_val'],
  199. $prompt,
  200. $params['model']
  201. );
  202. if (empty($result['success'])) {
  203. return $this->jsonResponse(1, $result['message'] ?? '生成失败');
  204. }
  205. $content = $result['data'] ?? '';
  206. //区分业务场景处理
  207. $isProductImageGeneration = ($params['status_type'] ?? '') === 'ProductImageGeneration';
  208. $isProductTemplateReplace = ($params['status_type'] ?? '') === 'ProductTemplateReplace';
  209. if (!$isProductImageGeneration && !$isProductTemplateReplace) {
  210. Db::name('product')->where('id', $params['id'])->update(['content' => $content]);
  211. }
  212. return $this->jsonResponse(0, '优化成功', ['content' => $content]);
  213. }
  214. /**
  215. * 文生图任务处理:提交任务并返回task_id
  216. * @param array $params 前端入参
  217. * @return \think\response\Json
  218. */
  219. private function handleAiTextToImg(array $params): \think\response\Json
  220. {
  221. $serviceResult = (new ImageService())->handleTextToImg($params);
  222. return $this->jsonTaskResponse($serviceResult, '文生图任务提交失败');
  223. }
  224. /**
  225. * 图生图任务处理:提交任务并返回task_id
  226. * @param array $params 前端入参
  227. * @return \think\response\Json
  228. */
  229. private function handleAiImgToImg(array $params): \think\response\Json
  230. {
  231. $serviceResult = (new ImageService())->handleImgToImg($params);
  232. return $this->jsonTaskResponse($serviceResult, '图生图任务提交失败');
  233. }
  234. /**
  235. * 即梦AI--创建视频任务接口
  236. * 支持:单张首帧图 / 首尾双帧图
  237. * 图片入参:JSON/base64(first_image/last_image)、form-data 文件、或 http(s) URL
  238. */
  239. public function Create_ImgToVideo()
  240. {
  241. $apiUrl = 'https://ark.cn-beijing.volces.com/api/v3/contents/generations/tasks';
  242. $apiKey = 'ark-1ca8aa97-3663-4bc7-8c53-d4ab516883f1-d2339';
  243. $params = $this->mergeRequestParams();
  244. $prompt = trim((string)($params['prompt'] ?? ''));
  245. if ($prompt === '') {
  246. return json(['code' => 0, 'msg' => 'prompt 不能为空']);
  247. }
  248. $firstError = '';
  249. $firstFrame = $this->resolveFrameImagePayload($params, 'first', $firstError);
  250. if (($firstFrame['api_url'] ?? '') === '') {
  251. $hint = Common::isOssEnabled()
  252. ? '请传 first_image(JSON 内 data:image/...;base64,... 或 form-data 文件),并查看 runtime/log'
  253. : '请在 application/config.php 配置 oss(accessKeyId、endpoint、bucket、host)';
  254. $detail = $firstError !== '' ? '(' . $firstError . ')' : '';
  255. return json([
  256. 'code' => 0,
  257. 'msg' => '首帧图片无效或上传失败。' . $hint . $detail,
  258. 'data' => [
  259. 'has_first_image' => isset($params['first_image']) && $params['first_image'] !== '',
  260. 'content_type' => $this->request->contentType(),
  261. 'upload_env' => $this->getPhpUploadEnvInfo(),
  262. ],
  263. ]);
  264. }
  265. $lastFrame = $this->resolveFrameImagePayload($params, 'last');
  266. $firstImageUrl = $firstFrame['api_url'];
  267. $lastImageUrl = $lastFrame['api_url'];
  268. $content = $this->buildImgToVideoContent($prompt, $firstImageUrl, $lastImageUrl);
  269. $data = [
  270. 'model' => 'doubao-seedance-1-5-pro-251215',
  271. 'content' => $content,
  272. 'generate_audio' => filter_var($params['generate_audio'] ?? true, FILTER_VALIDATE_BOOLEAN),
  273. 'ratio' => $params['ratio'] ?? $params['aspect_ratio'] ?? $params['size'] ?? 'adaptive',
  274. 'duration' => (int)($params['duration'] ?? $params['seconds'] ?? 5),
  275. 'watermark' => filter_var($params['watermark'] ?? false, FILTER_VALIDATE_BOOLEAN),
  276. ];
  277. $ch = curl_init();
  278. curl_setopt($ch, CURLOPT_URL, $apiUrl);
  279. curl_setopt($ch, CURLOPT_POST, true);
  280. curl_setopt($ch, CURLOPT_POSTFIELDS, json_encode($data, JSON_UNESCAPED_UNICODE));
  281. curl_setopt($ch, CURLOPT_HTTPHEADER, [
  282. 'Content-Type: application/json',
  283. 'Authorization: Bearer ' . $apiKey
  284. ]);
  285. curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
  286. curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
  287. curl_setopt($ch, CURLOPT_TIMEOUT, 60);
  288. $response = curl_exec($ch);
  289. if (curl_errno($ch)) {
  290. $error = curl_error($ch);
  291. curl_close($ch);
  292. return json(['code' => 0, 'msg' => 'Curl 错误: ' . $error]);
  293. }
  294. curl_close($ch);
  295. $responseData = json_decode($response, true);
  296. if (isset($responseData['error'])) {
  297. $msg = $responseData['error']['message'] ?? 'API 请求失败';
  298. return json(['code' => 0, 'msg' => 'API 错误: ' . $msg]);
  299. }
  300. $taskId = $responseData['id'] ?? '';
  301. if ($taskId === '') {
  302. return json(['code' => 0, 'msg' => '获取任务 ID 失败']);
  303. }
  304. $firstImageUrl = $this->finalizeFrameImageToOss($firstFrame, $taskId);
  305. $lastImageUrl = $this->finalizeFrameImageToOss($lastFrame, $taskId);
  306. $firstImageDb = $this->toDbUploadPath($firstImageUrl);
  307. $lastImageDb = $this->toDbUploadPath($lastImageUrl);
  308. $videoData = [
  309. 'video_id' => $taskId,
  310. 'prompt' => $prompt,
  311. 'first_image_url' => $firstImageDb,
  312. 'last_image_url' => $lastImageDb,
  313. 'model' => $data['model'],
  314. 'seconds' => (string)$data['duration'],
  315. 'size' => (string)$data['ratio'],
  316. 'sys_rq' => date('Y-m-d H:i:s'),
  317. ];
  318. try {
  319. Db::name('video')->insert($videoData);
  320. } catch (Exception $e) {
  321. return json([
  322. 'code' => 0,
  323. 'msg' => '任务已创建,但数据库保存失败',
  324. 'data' => [
  325. 'task_id' => $taskId,
  326. 'error_message' => $e->getMessage(),
  327. ],
  328. ]);
  329. }
  330. return json([
  331. 'code' => 1,
  332. 'msg' => '任务创建成功',
  333. 'data' => [
  334. 'task_id' => $taskId,
  335. 'video_id' => $taskId,
  336. 'status' => $responseData['status'] ?? '',
  337. 'created_at' => $responseData['created_at'] ?? '',
  338. 'mode' => $lastImageUrl !== '' ? 'first_last_frame' : 'single_frame',
  339. 'first_image_url' => $this->toPublicMediaUrl($firstImageDb),
  340. 'last_image_url' => $this->toPublicMediaUrl($lastImageDb),
  341. ]
  342. ]);
  343. }
  344. /**
  345. * 即梦AI--获取视频接口(先 GET 查询任务状态,succeeded 后再落盘/OSS/入库)
  346. */
  347. public function Get_ImgToVideo()
  348. {
  349. $apiUrl = 'https://ark.cn-beijing.volces.com/api/v3/contents/generations/tasks';
  350. $apiKey = 'ark-1ca8aa97-3663-4bc7-8c53-d4ab516883f1-d2339';
  351. $params = $this->request->param();
  352. $taskId = trim((string)($params['task_id'] ?? $params['video_id'] ?? ''));
  353. if ($taskId === '') {
  354. return json(['code' => 0, 'msg' => '任务 ID 不能为空']);
  355. }
  356. // 1. 先根据任务 ID 查询即梦任务状态(GET /tasks/{id})
  357. $queryResult = $this->fetchVolcImgToVideoTask($taskId, $apiUrl, $apiKey);
  358. if (!$queryResult['ok']) {
  359. return json([
  360. 'code' => 0,
  361. 'msg' => '查询任务失败: ' . $queryResult['error'],
  362. 'data' => [
  363. 'task_id' => $taskId,
  364. 'http_code' => $queryResult['http_code'],
  365. 'task' => $queryResult['data'],
  366. ],
  367. ]);
  368. }
  369. $taskData = $queryResult['data'];
  370. $taskStatus = (string)($taskData['status'] ?? '');
  371. // 2. 终态(failed / cancelled / expired)直接返回,不进入下载/入库
  372. $terminalMsg = $this->resolveVolcTaskTerminalFailure($taskStatus);
  373. if ($terminalMsg !== null) {
  374. return json([
  375. 'code' => 0,
  376. 'msg' => $terminalMsg,
  377. 'data' => $this->formatVolcTaskQueryPayload($taskData),
  378. ]);
  379. }
  380. // 3. queued / running 等未完成态则轮询(最多 30 次,间隔 5 秒)
  381. if (!$this->isVolcVideoTaskSucceeded($taskStatus)) {
  382. $pollResult = $this->pollVolcImgToVideoTaskUntilDone($taskId, $apiUrl, $apiKey, 30, 5);
  383. if (!$pollResult['ok']) {
  384. return json([
  385. 'code' => 0,
  386. 'msg' => $pollResult['error'],
  387. 'data' => $this->formatVolcTaskQueryPayload($pollResult['data']),
  388. ]);
  389. }
  390. $taskData = $pollResult['data'];
  391. $taskStatus = (string)($taskData['status'] ?? '');
  392. }
  393. if (!$this->isVolcVideoTaskSucceeded($taskStatus)) {
  394. return json([
  395. 'code' => 0,
  396. 'msg' => '轮询超时,当前状态:' . $this->getVolcTaskStatusMessage($taskStatus),
  397. 'data' => $this->formatVolcTaskQueryPayload($taskData),
  398. ]);
  399. }
  400. // 4. 事务处理:下载视频 → 本地/OSS → 更新数据库
  401. $videoUrl = $this->extractVideoUrlFromVolcTask($taskData);
  402. if ($videoUrl === '') {
  403. return json([
  404. 'code' => 0,
  405. 'msg' => '获取视频 URL 失败',
  406. 'data' => $this->formatVolcTaskQueryPayload($taskData),
  407. ]);
  408. }
  409. $fileName = $this->sanitizeTaskIdSegment($taskId) . '.mp4';
  410. $saveDir = $this->buildTaskMediaLocalDir($taskId);
  411. if (!is_dir($saveDir) && !@mkdir($saveDir, 0755, true) && !is_dir($saveDir)) {
  412. return json(['code' => 0, 'msg' => '创建视频目录失败', 'data' => ['saveDir' => $saveDir]]);
  413. }
  414. $savePath = $saveDir . $fileName;
  415. $videoContent = $this->downloadRemoteFile($videoUrl);
  416. if ($videoContent === false) {
  417. return json(['code' => 0, 'msg' => '下载视频失败', 'data' => ['videoUrl' => $videoUrl]]);
  418. }
  419. if (file_put_contents($savePath, $videoContent) === false) {
  420. return json(['code' => 0, 'msg' => '保存视频失败', 'data' => ['savePath' => $savePath]]);
  421. }
  422. $objectKey = $this->buildTaskMediaObjectKey($taskId, $fileName);
  423. $upload = $this->uploadToOSS($savePath, $objectKey);
  424. $ossObjectKey = (string)($upload['object_key'] ?? $objectKey);
  425. $webUrlDb = $this->toDbUploadPath($ossObjectKey);
  426. if ($webUrlDb === '' || stripos($webUrlDb, 'uploads/') !== 0) {
  427. $fullUrl = $upload['success']
  428. ? ((string)($upload['url'] ?? '') !== '' ? $upload['url'] : Common::ossFullUrl($objectKey))
  429. : ($this->buildPublicUploadUrl($savePath) ?: Common::ossFullUrl($objectKey));
  430. $webUrlDb = $this->toDbUploadPath((string)$fullUrl);
  431. if (!$upload['success']) {
  432. Log::write('[Get_ImgToVideo] OSS上传失败,入库路径=' . $webUrlDb, 'error');
  433. }
  434. }
  435. try {
  436. Db::name('video')->where('video_id', $taskId)->update(['web_url' => $webUrlDb]);
  437. } catch (Exception $e) {
  438. return json([
  439. 'code' => 0,
  440. 'msg' => '视频已生成,但数据库更新失败',
  441. 'data' => array_merge($this->formatVolcTaskQueryPayload($taskData), [
  442. 'web_url' => $this->toPublicMediaUrl($webUrlDb),
  443. 'error_message' => $e->getMessage(),
  444. ]),
  445. ]);
  446. }
  447. return json([
  448. 'code' => 1,
  449. 'msg' => '视频获取成功',
  450. 'data' => array_merge($this->formatVolcTaskQueryPayload($taskData), [
  451. 'task_id' => $taskId,
  452. 'video_id' => $taskId,
  453. 'web_url' => $this->toPublicMediaUrl($webUrlDb),
  454. 'oss_object_key' => $ossObjectKey,
  455. 'oss_uploaded' => $upload['success'] ?? false,
  456. 'local_path' => $webUrlDb,
  457. 'source_video_url' => $videoUrl,
  458. ]),
  459. ]);
  460. }
  461. /**
  462. * GET 查询即梦图生视频任务:/api/v3/contents/generations/tasks/{task_id}
  463. * @return array{ok:bool,data:array,error:string,http_code:int}
  464. */
  465. private function fetchVolcImgToVideoTask(string $taskId, string $apiUrl, string $apiKey): array
  466. {
  467. $empty = ['ok' => false, 'data' => [], 'error' => '', 'http_code' => 0];
  468. $queryUrl = rtrim($apiUrl, '/') . '/' . str_replace('%2F', '/', rawurlencode($taskId));
  469. $ch = curl_init();
  470. curl_setopt($ch, CURLOPT_URL, $queryUrl);
  471. curl_setopt($ch, CURLOPT_HTTPGET, true);
  472. curl_setopt($ch, CURLOPT_HTTPHEADER, [
  473. 'Content-Type: application/json',
  474. 'Authorization: Bearer ' . $apiKey,
  475. ]);
  476. curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
  477. curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
  478. curl_setopt($ch, CURLOPT_TIMEOUT, 60);
  479. $response = curl_exec($ch);
  480. if (curl_errno($ch)) {
  481. $empty['error'] = curl_error($ch);
  482. curl_close($ch);
  483. return $empty;
  484. }
  485. $httpCode = (int)curl_getinfo($ch, CURLINFO_HTTP_CODE);
  486. curl_close($ch);
  487. $data = json_decode((string)$response, true);
  488. if (!is_array($data)) {
  489. return ['ok' => false, 'data' => [], 'error' => '任务响应解析失败', 'http_code' => $httpCode];
  490. }
  491. if (isset($data['error'])) {
  492. $errMsg = is_array($data['error'])
  493. ? (string)($data['error']['message'] ?? json_encode($data['error'], JSON_UNESCAPED_UNICODE))
  494. : (string)$data['error'];
  495. return ['ok' => false, 'data' => $data, 'error' => $errMsg, 'http_code' => $httpCode];
  496. }
  497. if ($httpCode < 200 || $httpCode >= 300) {
  498. return ['ok' => false, 'data' => $data, 'error' => 'HTTP ' . $httpCode, 'http_code' => $httpCode];
  499. }
  500. return ['ok' => true, 'data' => $data, 'error' => '', 'http_code' => $httpCode];
  501. }
  502. /**
  503. * 轮询任务直至 succeeded,或遇到 failed/cancelled/expired
  504. * @return array{ok:bool,data:array,error:string}
  505. */
  506. private function pollVolcImgToVideoTaskUntilDone(
  507. string $taskId,
  508. string $apiUrl,
  509. string $apiKey,
  510. int $maxPolls = 30,
  511. int $intervalSeconds = 5
  512. ): array {
  513. $pollCount = 0;
  514. $lastData = [];
  515. while ($pollCount < $maxPolls) {
  516. sleep($intervalSeconds);
  517. $pollCount++;
  518. $result = $this->fetchVolcImgToVideoTask($taskId, $apiUrl, $apiKey);
  519. if (!$result['ok']) {
  520. return ['ok' => false, 'data' => $result['data'], 'error' => $result['error']];
  521. }
  522. $lastData = $result['data'];
  523. $status = (string)($lastData['status'] ?? '');
  524. $terminalMsg = $this->resolveVolcTaskTerminalFailure($status);
  525. if ($terminalMsg !== null) {
  526. return ['ok' => false, 'data' => $lastData, 'error' => $terminalMsg];
  527. }
  528. if ($this->isVolcVideoTaskSucceeded($status)) {
  529. return ['ok' => true, 'data' => $lastData, 'error' => ''];
  530. }
  531. }
  532. $lastStatus = (string)($lastData['status'] ?? '');
  533. return [
  534. 'ok' => false,
  535. 'data' => $lastData,
  536. 'error' => '轮询超时,当前状态:' . $this->getVolcTaskStatusMessage($lastStatus),
  537. ];
  538. }
  539. /**
  540. * 即梦任务 status 中文说明(官方:queued/running/cancelled/succeeded/failed/expired)
  541. */
  542. private function getVolcTaskStatusMessage(string $status): string
  543. {
  544. $map = [
  545. 'queued' => '排队中',
  546. 'running' => '任务运行中',
  547. 'cancelled' => '任务已取消',
  548. 'succeeded' => '任务成功',
  549. 'failed' => '任务失败',
  550. 'expired' => '任务超时',
  551. 'completed' => '任务成功',
  552. ];
  553. return $map[$status] ?? ('未知状态(' . $status . ')');
  554. }
  555. /**
  556. * 是否为可继续轮询的进行中状态
  557. */
  558. private function isVolcVideoTaskPending(string $status): bool
  559. {
  560. return in_array($status, ['queued', 'running'], true);
  561. }
  562. /**
  563. * 终态且不可下载:failed / cancelled / expired
  564. */
  565. private function resolveVolcTaskTerminalFailure(string $status): ?string
  566. {
  567. $map = [
  568. 'failed' => '任务执行失败',
  569. 'cancelled' => '任务已取消',
  570. 'expired' => '任务超时',
  571. ];
  572. return $map[$status] ?? null;
  573. }
  574. private function isVolcVideoTaskSucceeded(string $status): bool
  575. {
  576. return in_array($status, ['succeeded', 'completed'], true);
  577. }
  578. /**
  579. * 从即梦任务响应中取视频地址(优先 content.video_url)
  580. */
  581. private function extractVideoUrlFromVolcTask(array $taskData): string
  582. {
  583. $content = $taskData['content'] ?? null;
  584. if (is_array($content) && !empty($content['video_url'])) {
  585. return trim((string)$content['video_url']);
  586. }
  587. return trim((string)(
  588. $taskData['output']['video_url']
  589. ?? $taskData['video_url']
  590. ?? ''
  591. ));
  592. }
  593. /**
  594. * 格式化任务查询结果(与即梦 GET 响应字段对齐)
  595. */
  596. private function formatVolcTaskQueryPayload(array $taskData): array
  597. {
  598. $status = (string)($taskData['status'] ?? '');
  599. return [
  600. 'id' => $taskData['id'] ?? '',
  601. 'model' => $taskData['model'] ?? '',
  602. 'status' => $status,
  603. 'status_text' => $this->getVolcTaskStatusMessage($status),
  604. 'is_pending' => $this->isVolcVideoTaskPending($status),
  605. 'is_succeeded' => $this->isVolcVideoTaskSucceeded($status),
  606. 'content' => $taskData['content'] ?? null,
  607. 'usage' => $taskData['usage'] ?? null,
  608. 'created_at' => $taskData['created_at'] ?? null,
  609. 'updated_at' => $taskData['updated_at'] ?? null,
  610. 'resolution' => $taskData['resolution'] ?? '',
  611. 'ratio' => $taskData['ratio'] ?? '',
  612. 'duration' => $taskData['duration'] ?? null,
  613. 'framespersecond' => $taskData['framespersecond'] ?? null,
  614. 'generate_audio' => $taskData['generate_audio'] ?? null,
  615. ];
  616. }
  617. /**
  618. * 下载远程文件(视频等)
  619. * @return string|false
  620. */
  621. private function downloadRemoteFile(string $url)
  622. {
  623. $ch = curl_init();
  624. curl_setopt($ch, CURLOPT_URL, $url);
  625. curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
  626. curl_setopt($ch, CURLOPT_FOLLOWLOCATION, true);
  627. curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
  628. curl_setopt($ch, CURLOPT_SSL_VERIFYHOST, 0);
  629. curl_setopt($ch, CURLOPT_TIMEOUT, 300);
  630. $body = curl_exec($ch);
  631. if (curl_errno($ch)) {
  632. Log::write('[downloadRemoteFile] ' . curl_error($ch) . ' | url=' . $url, 'error');
  633. curl_close($ch);
  634. return false;
  635. }
  636. $httpCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
  637. curl_close($ch);
  638. if ($httpCode < 200 || $httpCode >= 300 || $body === false) {
  639. return false;
  640. }
  641. return $body;
  642. }
  643. //获取视频列表
  644. public function Getvideolist(){
  645. if (!$this->request->isGet()) {
  646. $this->error('请求方式错误');
  647. }
  648. $params = $this->request->param();
  649. $search = input('search', '');
  650. $page = isset($params['page']) ? (int)$params['page'] : 1;
  651. $limit = isset($params['limit']) ? (int)$params['limit'] : 50;
  652. $where = [];
  653. if (!empty($search)) {
  654. $where['prompt'] = ['like', '%' . $search . '%'];
  655. }
  656. $list = Db::name('video')->where('mod_rq', null)
  657. ->where($where)
  658. ->order('id desc')
  659. ->limit(($page - 1) * $limit, $limit)
  660. ->select();
  661. foreach ($list as &$row) {
  662. if (!empty($row['web_url'])) {
  663. $row['web_url'] = $this->toPublicMediaUrl((string)$row['web_url']);
  664. }
  665. if (!empty($row['first_image_url'])) {
  666. $row['first_image_url'] = $this->toPublicMediaUrl((string)$row['first_image_url']);
  667. }
  668. if (!empty($row['last_image_url'])) {
  669. $row['last_image_url'] = $this->toPublicMediaUrl((string)$row['last_image_url']);
  670. }
  671. }
  672. unset($row);
  673. $total = Db::name('video')->where('mod_rq', null)
  674. ->where($where)
  675. ->count();
  676. $res['code'] = 0;
  677. $res['msg'] = '成功';
  678. $res['count'] = $total;
  679. $res['data'] = $list;
  680. return json($res);
  681. }
  682. /**
  683. * 文生视频/图生视频接口
  684. */
  685. //文生视频
  686. public function video(){
  687. $apiUrl = 'https://ark.cn-beijing.volces.com/api/v3/contents/generations/tasks';
  688. $apiKey = 'ark-1ca8aa97-3663-4bc7-8c53-d4ab516883f1-d2339';
  689. $params = $this->request->param();
  690. $postData = [
  691. 'prompt' => $params['prompt'],
  692. 'model' => 'doubao-seedance-1-5-pro-251215',
  693. 'seconds' => $params['seconds'],
  694. 'size' => $params['size'],
  695. ];
  696. // 初始化CURL
  697. $ch = curl_init();
  698. curl_setopt($ch, CURLOPT_URL, $apiUrl);
  699. curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
  700. curl_setopt($ch, CURLOPT_POST, true);
  701. curl_setopt($ch, CURLOPT_POSTFIELDS, $postData);
  702. curl_setopt($ch, CURLOPT_HTTPHEADER, [
  703. 'Authorization: Bearer ' . $apiKey
  704. ]);
  705. curl_setopt($ch, CURLOPT_TIMEOUT, 300);
  706. curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
  707. curl_setopt($ch, CURLOPT_SSL_VERIFYHOST, 0);
  708. curl_setopt($ch, CURLOPT_HEADER, true); // 获取响应头
  709. curl_setopt($ch, CURLOPT_VERBOSE, true); // 启用详细输出以进行调试
  710. // 创建临时文件来捕获详细的cURL输出
  711. $verbose = fopen('php://temp', 'w+');
  712. curl_setopt($ch, CURLOPT_STDERR, $verbose);
  713. // 执行请求
  714. $response = curl_exec($ch);
  715. //HTTP状态码
  716. $httpCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
  717. // 获取详细的cURL调试信息
  718. rewind($verbose);
  719. //CURL调试信息
  720. $verboseLog = stream_get_contents($verbose);
  721. fclose($verbose);
  722. // 分离头部和主体
  723. $header_size = curl_getinfo($ch, CURLINFO_HEADER_SIZE);
  724. //响应头部
  725. $header = substr($response, 0, $header_size);
  726. //响应主体
  727. $body = substr($response, $header_size);
  728. // 检查CURL错误
  729. $curlError = curl_error($ch);
  730. curl_close($ch);
  731. $responseData = json_decode($body, true);
  732. echo "<pre>";
  733. print_r($responseData);
  734. echo "<pre>";die;
  735. // 检查API是否返回了错误信息
  736. if (isset($responseData['error'])) {
  737. $errorMessage = isset($responseData['error']['message']) ? $responseData['error']['message'] : 'API请求失败';
  738. return json([
  739. 'code' => 1,
  740. 'msg' => '视频生成请求失败',
  741. 'data' => [
  742. 'error_type' => isset($responseData['error']['type']) ? $responseData['error']['type'] : 'unknown',
  743. 'error_code' => isset($responseData['error']['code']) ? $responseData['error']['code'] : 'unknown',
  744. 'error_message' => $errorMessage
  745. ]
  746. ]);
  747. }
  748. // 检查是否有自定义错误格式
  749. if (isset($responseData['code']) && $responseData['code'] === 'fail_to_fetch_task' && isset($responseData['message'])) {
  750. return json([
  751. 'code' => 1,
  752. 'msg' => '视频生成请求失败',
  753. 'data' => [
  754. 'error_code' => $responseData['code'],
  755. 'error_message' => $responseData['message']
  756. ]
  757. ]);
  758. }
  759. // 检查是否存在id字段
  760. if (!isset($responseData['id'])) {
  761. return json([
  762. 'code' => 1,
  763. 'msg' => '无法获取视频ID',
  764. 'data' => [
  765. 'response_data' => $responseData,
  766. 'http_code' => $httpCode
  767. ]
  768. ]);
  769. }
  770. $videoData = [
  771. 'video_id' => $responseData['id'],
  772. 'prompt' => $postData['prompt'],
  773. 'model' => $postData['model'],
  774. 'seconds' => $postData['seconds'],
  775. 'size' => $postData['size'],
  776. 'sys_rq' => date("Y-m-d H:i:s")
  777. ];
  778. // 尝试插入数据
  779. try {
  780. $res = Db::name('video')->insert($videoData);
  781. return json([
  782. 'code' => 0,
  783. 'msg' => '视频正在生成中',
  784. 'data ' => [
  785. 'video_id' => $responseData['id'],
  786. 'insert_result' => $res
  787. ]
  788. ]);
  789. } catch (Exception $e) {
  790. return json([
  791. 'code' => 1,
  792. 'msg' => '数据库操作失败',
  793. 'data' => [
  794. 'error_message' => $e->getMessage()
  795. ]
  796. ]);
  797. }
  798. }
  799. /**
  800. * 获取视频内容
  801. * 下载已完成的视频内容
  802. */
  803. public function videoContent(){
  804. // 从请求参数获取video_id,如果没有则使用默认值
  805. $video_id = input('get.video_id');
  806. $apiKey = '';
  807. // 1. 先检查视频状态
  808. $statusUrl = 'https://chatapi.onechats.ai/v1/videos/' . $video_id;
  809. $statusData = $this->fetchVideoStatus($statusUrl, $apiKey);
  810. // 检查视频状态
  811. if ($statusData['status'] !== 'completed') {
  812. return json([
  813. 'code' => 202,
  814. 'msg' => '视频尚未生成完成',
  815. 'data' => [
  816. 'video_id' => $video_id,
  817. 'status' => $statusData['status'],
  818. 'progress' => $statusData['progress'],
  819. 'created_at' => $statusData['created_at'],
  820. 'message' => '请稍后再试,视频仍在' . ($statusData['status'] === 'queued' ? '排队中' : '处理中')
  821. ]
  822. ]);
  823. }
  824. // 2. 视频生成完成,准备下载
  825. $apiUrl = 'https://chatapi.onechats.ai/v1/videos/' . $video_id . '/content';
  826. // 获取可选的variant参数
  827. $variant = $this->request->get('variant', '');
  828. if (!empty($variant)) {
  829. $apiUrl .= '?variant=' . urlencode($variant);
  830. }
  831. // 创建保存目录
  832. $saveDir = ROOT_PATH . 'public' . DS . 'uploads' . DS . 'videos' . DS . date('Ymd');
  833. if (!is_dir($saveDir)) {
  834. mkdir($saveDir, 0755, true);
  835. }
  836. // 生成唯一文件名
  837. $filename = $video_id . '.mp4';
  838. $localPath = DS . 'uploads' . DS . 'videos' . DS . date('Ymd') . DS . $filename;
  839. $fullPath = $saveDir . DS . $filename;
  840. // 3. 下载视频
  841. $videoData = $this->downloadVideo($apiUrl, $apiKey);
  842. // 4. 保存视频文件
  843. if (file_put_contents($fullPath, $videoData) === false) {
  844. throw new Exception('视频保存失败');
  845. }
  846. // 确保路径使用正斜杠,并只保存相对路径部分
  847. $localPath = str_replace('\\', '/', $localPath);
  848. // 移除开头的斜杠,确保路径格式为uploads/videos/...
  849. $savePath = ltrim($localPath, '/');
  850. // 将正确格式的文件路径存入数据库
  851. Db::name('video')->where('video_id', $video_id)->update([
  852. 'web_url' => $savePath
  853. ]);
  854. // 返回成功响应
  855. return json([
  856. 'code' => 0,
  857. 'msg' => '视频下载成功',
  858. 'data' => [
  859. 'video_id' => $video_id,
  860. 'local_path' => $localPath,
  861. 'web_url' => $savePath,
  862. 'file_size' => filesize($fullPath)
  863. ]
  864. ]);
  865. }
  866. /**
  867. * 获取视频状态
  868. */
  869. private function fetchVideoStatus($url, $apiKey) {
  870. $ch = curl_init();
  871. curl_setopt($ch, CURLOPT_URL, $url);
  872. curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
  873. curl_setopt($ch, CURLOPT_HTTPGET, true);
  874. curl_setopt($ch, CURLOPT_HTTPHEADER, [
  875. 'Authorization: Bearer ' . $apiKey,
  876. 'Accept: application/json'
  877. ]);
  878. curl_setopt($ch, CURLOPT_TIMEOUT, 30);
  879. curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
  880. curl_setopt($ch, CURLOPT_SSL_VERIFYHOST, 0);
  881. $response = curl_exec($ch);
  882. $httpCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
  883. $error = curl_error($ch);
  884. curl_close($ch);
  885. if ($error) {
  886. throw new Exception('获取视频状态失败: ' . $error);
  887. }
  888. if ($httpCode < 200 || $httpCode >= 300) {
  889. throw new Exception('获取视频状态失败,HTTP状态码: ' . $httpCode);
  890. }
  891. $data = json_decode($response, true);
  892. if (!is_array($data)) {
  893. throw new Exception('视频状态数据格式错误');
  894. }
  895. return $data;
  896. }
  897. /**
  898. * 下载视频文件
  899. */
  900. private function downloadVideo($url, $apiKey) {
  901. $ch = curl_init();
  902. curl_setopt($ch, CURLOPT_URL, $url);
  903. curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
  904. curl_setopt($ch, CURLOPT_HTTPGET, true);
  905. curl_setopt($ch, CURLOPT_HTTPHEADER, [
  906. 'Authorization: Bearer ' . $apiKey
  907. ]);
  908. curl_setopt($ch, CURLOPT_TIMEOUT, 300);
  909. curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
  910. curl_setopt($ch, CURLOPT_SSL_VERIFYHOST, 0);
  911. $response = curl_exec($ch);
  912. $httpCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
  913. $error = curl_error($ch);
  914. curl_close($ch);
  915. if ($error) {
  916. throw new Exception('视频下载失败: ' . $error);
  917. }
  918. if ($httpCode < 200 || $httpCode >= 300) {
  919. throw new Exception('视频下载失败,HTTP状态码: ' . $httpCode);
  920. }
  921. return $response;
  922. }
  923. private function sendPostRequest($url, $data, $apiKey)
  924. {
  925. $ch = curl_init();
  926. curl_setopt($ch, CURLOPT_URL, $url);
  927. curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
  928. curl_setopt($ch, CURLOPT_POST, true);
  929. curl_setopt($ch, CURLOPT_HTTPHEADER, [
  930. 'Authorization: Bearer ' . $apiKey,
  931. 'Accept: application/json',
  932. 'Content-Type: application/json'
  933. ]);
  934. curl_setopt($ch, CURLOPT_POSTFIELDS, json_encode($data));
  935. curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
  936. curl_setopt($ch, CURLOPT_SSL_VERIFYHOST, false);
  937. curl_setopt($ch, CURLOPT_TIMEOUT, 60); // 延长超时时间
  938. $response = curl_exec($ch);
  939. $httpCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
  940. $error = curl_error($ch);
  941. curl_close($ch);
  942. return [
  943. 'response' => $response,
  944. 'http_code' => $httpCode,
  945. 'error' => $error
  946. ];
  947. }
  948. /**
  949. * 查询模版
  950. */
  951. public function product_template()
  952. {
  953. $params = $this->request->param();
  954. if (!$this->request->isGet()) {
  955. $this->error('请求方法错误');
  956. }
  957. $page = isset($params['page']) ? (int)$params['page'] : 1;
  958. $limit = isset($params['limit']) ? (int)$params['limit'] : 30;
  959. $where = [];
  960. if (!empty($params['search'])) {
  961. $where['id|chinese_description|template_name|style'] = ['like', '%' . $params['search'] . '%'];
  962. }
  963. // toexamine / release:用 isset + 非空串,避免 empty(0) 导致「未发布」无法筛选;release 建议库中为 0/1
  964. if (isset($params['toexamine']) && $params['toexamine'] !== '') {
  965. $where['toexamine'] = $params['toexamine'];
  966. }
  967. if (isset($params['release']) && $params['release'] !== '') {
  968. $where['release'] = is_numeric($params['release']) ? (int) $params['release'] : $params['release'];
  969. }
  970. $isSuperAdmin = false;
  971. if (!empty($params['sys_id']) && $params['sys_id'] == '超级管理员') {
  972. $isSuperAdmin = true;
  973. }
  974. if (!$isSuperAdmin && !empty($params['sys_id'])) {
  975. $where['sys_id'] = ['like', '%' . $params['sys_id'] . '%'];
  976. $products = Db::name('product_template')->order('id desc')->where($where)
  977. ->whereNull('mod_rq')
  978. ->limit(($page - 1) * $limit, $limit)
  979. ->select();
  980. }else{
  981. $products = Db::name('product_template')->order('id desc')->where($where)
  982. ->whereNull('mod_rq')
  983. ->limit(($page - 1) * $limit, $limit)
  984. ->select();
  985. }
  986. $total = Db::name('product_template')->where($where)
  987. ->whereNull('mod_rq')
  988. ->count();
  989. foreach ($products as &$item) {
  990. if (!empty($item['template_image_url'])) {
  991. $item['template_image_url'] = Common::ossFullUrl((string)$item['template_image_url']);
  992. }
  993. if (!empty($item['thumbnail_image'])) {
  994. $item['thumbnail_image'] = Common::ossFullUrl((string)$item['thumbnail_image']);
  995. }
  996. }
  997. unset($item);
  998. return json([
  999. 'code' => 0,
  1000. 'msg' => '请求成功',
  1001. 'count' => $total,
  1002. 'data' => $products
  1003. ]);
  1004. }
  1005. /**
  1006. * 获取 AI 模型配置
  1007. * status 1 = 启用 0 = 禁用(同一模型内的优先级,数值越小越优先)
  1008. * model_type 支持多能力逗号间隔(如 文生图,图生图),传参精确匹配某一能力
  1009. * 可选参数:manage=1 时返回全部(含禁用),用于管理端
  1010. */
  1011. public function GetAIModel(){
  1012. $params = $this->request->param();
  1013. $query = Db::name('ai_model');
  1014. if (empty($params['manage'])) {
  1015. $query->where('status', '1');
  1016. }
  1017. if (!empty($params['model_type'])) {
  1018. $query->whereRaw('FIND_IN_SET(:mt, model_type) > 0', ['mt' => trim($params['model_type'])]);
  1019. }
  1020. if (!empty($params['supplier'])) {
  1021. $query->where('supplier', 'like', '%' . $params['supplier'] . '%');
  1022. }
  1023. if (!empty($params['model_name'])) {
  1024. $query->where('model_name|model_alias', 'like', '%' . $params['model_name'] . '%');
  1025. }
  1026. $list = $query
  1027. ->field('id,model_alias,model_group,model_name,model_type,sort,status,supplier')
  1028. ->order('sort ASC, id ASC')->select();
  1029. return json([
  1030. 'code' => 0,
  1031. 'msg' => '成功',
  1032. 'data' => $list
  1033. ]);
  1034. }
  1035. /**
  1036. * 新增 AI 模型配置
  1037. * POST: status, supplier, api_key, api_url, model_group, model_name, model_alias, model_type, sort
  1038. * model_type 多能力用逗号间隔,如:文生图,图生图
  1039. */
  1040. public function AddAIModel(){
  1041. $params = $this->request->param();
  1042. $required = ['api_url', 'api_key', 'model_name', 'model_type'];
  1043. foreach ($required as $k) {
  1044. if (empty(trim($params[$k] ?? ''))) {
  1045. return json(['code' => 1, 'msg' => $k . ' 不能为空']);
  1046. }
  1047. }
  1048. $insert = [
  1049. 'status' => $params['status'],
  1050. 'supplier' => trim($params['supplier']),
  1051. 'api_key' => trim($params['api_key']),
  1052. 'api_url' => trim($params['api_url']),
  1053. 'model_group' => trim($params['model_group']),
  1054. 'model_name' => trim($params['model_name']),
  1055. 'model_alias' => trim($params['model_alias']),
  1056. 'model_type' => trim($params['model_type']),
  1057. 'sort' => isset($params['sort']) ? intval($params['sort']) : 0,
  1058. ];
  1059. try {
  1060. Db::name('ai_model')->insert($insert);
  1061. return json(['code' => 0, 'msg' => '新增成功']);
  1062. } catch (\Exception $e) {
  1063. return json(['code' => 1, 'msg' => '新增失败: ' . $e->getMessage()]);
  1064. }
  1065. }
  1066. /**
  1067. * 修改 AI 模型配置
  1068. * POST: id(必填), 其余字段同新增
  1069. */
  1070. public function UpdateAIModel(){
  1071. $params = $this->request->param();
  1072. if (empty($params['id'])) {
  1073. return json(['code' => 1, 'msg' => 'id 不能为空']);
  1074. }
  1075. $id = intval($params['id']);
  1076. $exists = Db::name('ai_model')->where('id', $id)->find();
  1077. if (!$exists) {
  1078. return json(['code' => 1, 'msg' => '记录不存在']);
  1079. }
  1080. $update = [];
  1081. $fields = ['status', 'supplier', 'api_key', 'api_url', 'model_group', 'model_name', 'model_alias', 'model_type', 'sort'];
  1082. foreach ($fields as $f) {
  1083. if (array_key_exists($f, $params)) {
  1084. $update[$f] = $f === 'sort' ? intval($params[$f]) : trim($params[$f] ?? '');
  1085. }
  1086. }
  1087. if (empty($update)) {
  1088. return json(['code' => 1, 'msg' => '无有效修改字段']);
  1089. }
  1090. try {
  1091. Db::name('ai_model')->where('id', $id)->update($update);
  1092. return json(['code' => 0, 'msg' => '修改成功']);
  1093. } catch (\Exception $e) {
  1094. return json(['code' => 1, 'msg' => '修改失败: ' . $e->getMessage()]);
  1095. }
  1096. }
  1097. /**
  1098. * 用于获取所有产品记录
  1099. **/
  1100. public function GetProductList(){
  1101. $params = $this->request->param();
  1102. $page = max(1, intval($params['page'] ?? 1));
  1103. $pageSize = min(100, max(1, intval($params['limit'] ?? 30)));
  1104. // 构建查询条件
  1105. $where = [];
  1106. if (!empty($params['search'])) {
  1107. $where['prompt|model'] = ['like', '%' . $params['search'] . '%'];
  1108. }
  1109. if (!empty($params['sys_id'])) {
  1110. $where['sys_id'] = ['like', '%' . $params['sys_id'] . '%'];
  1111. }
  1112. $prompt = Db::name('product_image_generate')->field('prompt')->where($where)->group('prompt')->order('id desc')->select();
  1113. $data = Db::name('product_image_generate')->where($where)->order('id desc')->page($page, $pageSize)->select();
  1114. $count = Db::name('product_image_generate')->field('prompt')->where($where)->order('id desc')->count();
  1115. foreach ($data as &$item) {
  1116. if (!empty($item['generated_image']) || !empty($item['product_img']) || !empty($item['reference_image'])) {
  1117. $item['generated_image'] = Common::ossFullUrl((string)$item['generated_image']);
  1118. $item['product_img'] = Common::ossFullUrl((string)$item['product_img']);
  1119. $item['reference_image'] = Common::ossFullUrl((string)$item['reference_image']);
  1120. }
  1121. }
  1122. unset($item);
  1123. return json([
  1124. 'code' => 0,
  1125. 'msg' => '成功',
  1126. 'count' => $count,
  1127. 'prompt' => $prompt,
  1128. 'data' => $data
  1129. ]);
  1130. }
  1131. /**
  1132. * 查询队列列表
  1133. * 统计文件对应的队列情况
  1134. */
  1135. public function get_queue_logs()
  1136. {
  1137. $params = $this->request->param('old_image_file', '');
  1138. $queue_logs = Db::name('queue_logs')
  1139. ->where('old_image_file', $params)
  1140. ->order('id desc')
  1141. ->select();
  1142. $result = []; //初始化变量,避免未定义错误
  1143. foreach ($queue_logs as &$log) {
  1144. $taskId = $log['id'];
  1145. $statusCount = Db::name('image_task_log')
  1146. ->field('status, COUNT(*) as count')
  1147. ->where('task_id', $taskId)
  1148. ->where('mod_rq', null)
  1149. ->group('status')
  1150. ->select();
  1151. $log['已完成数量'] = 0;
  1152. $log['处理中数量'] = 0;
  1153. $log['排队中的数量'] = 0;
  1154. $log['失败数量'] = 0;
  1155. foreach ($statusCount as $item) {
  1156. switch ($item['status']) {
  1157. case 0:
  1158. $log['排队中的数量'] = $item['count'];
  1159. break;
  1160. case 1:
  1161. $log['处理中数量'] = $item['count'];
  1162. break;
  1163. case 2:
  1164. $log['已完成数量'] = $item['count'];
  1165. break;
  1166. case -1:
  1167. $log['失败数量'] = $item['count'];
  1168. break;
  1169. }
  1170. }
  1171. // if ($log['排队中的数量'] >$log['已完成数量']) {
  1172. // $result[] = $log;
  1173. // }
  1174. if ($log['排队中的数量']) {
  1175. $result[] = $log;
  1176. }
  1177. // if ($log['处理中数量'] >= 0) {
  1178. // $result[] = $log;
  1179. // }
  1180. }
  1181. return json([
  1182. 'code' => 0,
  1183. 'msg' => '查询成功',
  1184. 'data' => $result,
  1185. 'count' => count($result)
  1186. ]);
  1187. }
  1188. /**
  1189. * 查询总队列状态(统计当前处理的数据量)
  1190. */
  1191. public function queueStats()
  1192. {
  1193. $statusList = Db::name('image_task_log')
  1194. ->field('status, COUNT(*) as total')
  1195. ->where('mod_rq', null)
  1196. ->where('create_time', '>=', date('Y-m-d 00:00:00'))
  1197. ->group('status')
  1198. ->select();
  1199. $statusCount = [];
  1200. foreach ($statusList as $item) {
  1201. $statusCount[$item['status']] = $item['total'];
  1202. }
  1203. // 总数为所有状态和
  1204. $total = array_sum($statusCount);
  1205. //获取队列当前状态
  1206. $statusText = Db::name('queue_logs')->order('id desc')->value('status');
  1207. return json([
  1208. 'code' => 0,
  1209. 'msg' => '获取成功',
  1210. 'data' => [
  1211. '总任务数' => $total,
  1212. '待处理' => $statusCount[0] ?? 0,
  1213. '处理中' => $statusCount[1] ?? 0,
  1214. '成功' => $statusCount[2] ?? 0,
  1215. '失败' => $statusCount[-1] ?? 0,
  1216. '当前状态' => $statusText
  1217. ]
  1218. ]);
  1219. }
  1220. /**
  1221. * 获取 Redis 连接实例
  1222. * @return \Redis|null Redis 实例或 null(如果连接失败)
  1223. */
  1224. private function getRedisConnection()
  1225. {
  1226. if (!class_exists('\Redis')) {
  1227. return null;
  1228. }
  1229. return getTaskRedis();
  1230. }
  1231. /**
  1232. * 显示当前运行中的队列监听进程
  1233. */
  1234. public function viewQueueStatus()
  1235. {
  1236. $redis = $this->getRedisConnection();
  1237. if (!$redis) {
  1238. return json([
  1239. 'code' => 1,
  1240. 'msg' => 'Redis扩展未安装或未启用',
  1241. 'data' => null
  1242. ]);
  1243. }
  1244. $key = 'queues:imgtotxt';
  1245. // 判断 key 是否存在,避免报错
  1246. if (!$redis->exists($key)) {
  1247. return json([
  1248. 'code' => 0,
  1249. 'msg' => '查询成功,队列为空',
  1250. 'count' => 0,
  1251. 'tasks_preview' => []
  1252. ]);
  1253. }
  1254. $count = $redis->lLen($key);
  1255. $list = $redis->lRange($key, 0, 9);
  1256. // 解码 JSON 内容,确保每一项都有效
  1257. $parsed = array_filter(array_map(function ($item) {
  1258. return json_decode($item, true);
  1259. }, $list), function ($item) {
  1260. return !is_null($item);
  1261. });
  1262. return json([
  1263. 'code' => 0,
  1264. 'msg' => '查询成功',
  1265. 'count' => $count,
  1266. 'tasks_preview' => $parsed
  1267. ]);
  1268. }
  1269. /**
  1270. * 清空队列并删除队列日志记录
  1271. */
  1272. public function stopQueueProcesses()
  1273. {
  1274. Db::name('image_task_log')
  1275. ->where('log', '队列中')
  1276. ->whereOr('status', 1)
  1277. ->where('create_time', '>=', date('Y-m-d 00:00:00'))
  1278. ->update([
  1279. 'status' => "-1",
  1280. 'log' => '清空取消队列',
  1281. 'mod_rq' => date('Y-m-d H:i:s')
  1282. ]);
  1283. Db::name('image_task_log')
  1284. ->whereLike('log', '%处理中%')
  1285. ->where('create_time', '>=', date('Y-m-d 00:00:00'))
  1286. ->update([
  1287. 'status' => "-1",
  1288. 'log' => '清空取消队列',
  1289. 'mod_rq' => date('Y-m-d H:i:s')
  1290. ]);
  1291. $redis = $this->getRedisConnection();
  1292. if (!$redis) {
  1293. return json([
  1294. 'code' => 1,
  1295. 'msg' => 'Redis扩展未安装或未启用',
  1296. 'data' => null
  1297. ]);
  1298. }
  1299. $key_txttoimg = 'queues:txttoimg:reserved';
  1300. $key_txttotxt = 'queues:txttotxt:reserved';
  1301. $key_imgtotxt = 'queues:imgtotxt:reserved';
  1302. $key_imgtoimg = 'queues:imgtoimg:reserved';
  1303. // 清空 Redis 队列
  1304. $redis->del($key_txttoimg);
  1305. $redis->del($key_txttotxt);
  1306. $redis->del($key_imgtotxt);
  1307. $redis->del($key_imgtoimg);
  1308. $count = $redis->lLen($key_txttoimg) + $redis->lLen($key_txttotxt) + $redis->lLen($key_imgtotxt) + $redis->lLen($key_imgtoimg);
  1309. return json([
  1310. 'code' => 0,
  1311. 'msg' => '成功停止队列任务'
  1312. ]);
  1313. }
  1314. /**
  1315. *获取服务器URL地址和端口 IP地址:端口
  1316. * 用于获取图片路径拼接时
  1317. **/
  1318. public function GetHttpUrl(){
  1319. $data = Db::name('http_url')->find();
  1320. $fullUrl = "http://" . $data['baseUrl'] . ":" . $data['port'];
  1321. $res = [
  1322. 'code' => 0,
  1323. 'msg' => '成功',
  1324. 'data' => [
  1325. 'id' => $data['id'],
  1326. 'full_url' => $fullUrl,
  1327. 'baseUrl' => $data['baseUrl'],
  1328. 'port' => $data['port']
  1329. ]
  1330. ];
  1331. return json($res);
  1332. }
  1333. /**
  1334. * 构建图生视频 content:单帧无 role,双帧带 first_frame / last_frame
  1335. */
  1336. private function buildImgToVideoContent(string $prompt, string $firstImageUrl, string $lastImageUrl): array
  1337. {
  1338. $content = [
  1339. ['type' => 'text', 'text' => $prompt],
  1340. ];
  1341. if ($lastImageUrl !== '') {
  1342. $content[] = [
  1343. 'type' => 'image_url',
  1344. 'image_url' => ['url' => $firstImageUrl],
  1345. 'role' => 'first_frame',
  1346. ];
  1347. $content[] = [
  1348. 'type' => 'image_url',
  1349. 'image_url' => ['url' => $lastImageUrl],
  1350. 'role' => 'last_frame',
  1351. ];
  1352. return $content;
  1353. }
  1354. $content[] = [
  1355. 'type' => 'image_url',
  1356. 'image_url' => ['url' => $firstImageUrl],
  1357. ];
  1358. return $content;
  1359. }
  1360. /** 图生视频创建前,帧图暂存 OSS 子目录(AI 返回 id 后会迁移到正式目录) */
  1361. private const FRAME_OSS_STAGING_ID = '_staging';
  1362. /**
  1363. * 任务 id 用于路径的安全片段
  1364. */
  1365. private function sanitizeTaskIdSegment(string $taskId): string
  1366. {
  1367. $segment = preg_replace('/[\\\\\/:*?"<>|]/u', '_', trim($taskId));
  1368. return $segment !== '' ? $segment : '_unknown';
  1369. }
  1370. /**
  1371. * 图生视频任务媒体本地目录:public/uploads/videos/{日期}/{taskId}/
  1372. */
  1373. private function buildTaskMediaLocalDir(string $taskId): string
  1374. {
  1375. return str_replace('\\', '/', ROOT_PATH . 'public/uploads/videos/' . date('Ymd') . '/' . $this->sanitizeTaskIdSegment($taskId) . '/');
  1376. }
  1377. /**
  1378. * 图生视频任务媒体 OSS 对象键:uploads/videos/{日期}/{taskId}/{文件名}(帧图与生成视频同目录)
  1379. */
  1380. private function buildTaskMediaObjectKey(string $taskId, string $fileName): string
  1381. {
  1382. return 'uploads/videos/' . date('Ymd') . '/' . $this->sanitizeTaskIdSegment($taskId) . '/' . ltrim($fileName, '/');
  1383. }
  1384. /**
  1385. * 解析帧图片(含本地路径与暂存 OSS 信息,供创建任务后按 AI id 归档)
  1386. * @return array{type:string,api_url:string,local_path:string,staging_object_key:string}
  1387. */
  1388. private function resolveFrameImagePayload(array $params, string $role, string &$error = ''): array
  1389. {
  1390. $empty = ['type' => 'empty', 'api_url' => '', 'local_path' => '', 'staging_object_key' => ''];
  1391. $prefix = $role === 'last' ? 'last' : 'first';
  1392. $error = '';
  1393. $taskId = trim((string)($params['task_id'] ?? $params['video_id'] ?? ''));
  1394. $ossTaskId = $taskId !== '' ? $taskId : self::FRAME_OSS_STAGING_ID;
  1395. $uploadedFile = $this->request->file("{$prefix}_image");
  1396. if (!empty($uploadedFile)) {
  1397. $fileList = is_array($uploadedFile) ? $uploadedFile : [$uploadedFile];
  1398. foreach ($fileList as $file) {
  1399. if (!$file) {
  1400. continue;
  1401. }
  1402. $fileError = '';
  1403. $result = $this->uploadFrameFileToOss($file, $prefix, $fileError, $ossTaskId);
  1404. if (($result['url'] ?? '') !== '') {
  1405. return [
  1406. 'type' => 'local',
  1407. 'api_url' => $result['url'],
  1408. 'local_path' => $result['local_path'] ?? '',
  1409. 'staging_object_key' => $result['object_key'] ?? '',
  1410. ];
  1411. }
  1412. if ($fileError !== '') {
  1413. $error = $fileError;
  1414. }
  1415. }
  1416. }
  1417. $base64Keys = ["{$prefix}_image", "{$prefix}_image_base64"];
  1418. foreach ($base64Keys as $key) {
  1419. if (empty($params[$key]) || !is_string($params[$key])) {
  1420. continue;
  1421. }
  1422. $raw = trim($params[$key]);
  1423. if ($raw === '' || strlen($raw) < 50) {
  1424. continue;
  1425. }
  1426. $result = $this->uploadBase64ImageToOss($raw, $prefix, $ossTaskId, $error);
  1427. if (($result['url'] ?? '') !== '') {
  1428. return [
  1429. 'type' => 'local',
  1430. 'api_url' => $result['url'],
  1431. 'local_path' => $result['local_path'] ?? '',
  1432. 'staging_object_key' => $result['object_key'] ?? '',
  1433. ];
  1434. }
  1435. }
  1436. $urlKey = "{$prefix}_image_url";
  1437. if (!empty($params[$urlKey])) {
  1438. $url = trim((string)$params[$urlKey]);
  1439. if (stripos($url, 'http://') === 0 || stripos($url, 'https://') === 0) {
  1440. return ['type' => 'external', 'api_url' => $url, 'local_path' => '', 'staging_object_key' => ''];
  1441. }
  1442. }
  1443. return $empty;
  1444. }
  1445. /**
  1446. * AI 返回任务 id 后,将本地帧图上传到 uploads/videos/{日期}/{taskId}/(与生成视频同目录)
  1447. */
  1448. private function finalizeFrameImageToOss(array $payload, string $taskId): string
  1449. {
  1450. if (($payload['type'] ?? '') === 'external') {
  1451. return (string)($payload['api_url'] ?? '');
  1452. }
  1453. if (($payload['type'] ?? '') !== 'local') {
  1454. return '';
  1455. }
  1456. $localPath = (string)($payload['local_path'] ?? '');
  1457. if ($localPath === '' || !is_file($localPath)) {
  1458. return (string)($payload['api_url'] ?? '');
  1459. }
  1460. $objectKey = $this->buildTaskMediaObjectKey($taskId, basename($localPath));
  1461. $stagingKey = (string)($payload['staging_object_key'] ?? '');
  1462. if ($stagingKey !== '' && $stagingKey === $objectKey) {
  1463. return (string)($payload['api_url'] ?? '') ?: Common::ossFullUrl($objectKey);
  1464. }
  1465. $upload = $this->uploadToOSS($localPath, $objectKey);
  1466. if (!$upload['success']) {
  1467. Log::write('[finalizeFrameImageToOss] OSS上传失败: ' . $objectKey, 'error');
  1468. return (string)($payload['api_url'] ?? '');
  1469. }
  1470. if ($stagingKey !== '' && $stagingKey !== $objectKey) {
  1471. Common::deleteOssObject($stagingKey);
  1472. }
  1473. return $upload['url'] !== '' ? $upload['url'] : Common::ossFullUrl($objectKey);
  1474. }
  1475. /**
  1476. * 解析帧图片:form-data 文件 > base64 > http(s) URL
  1477. */
  1478. private function resolveFrameImageUrl(array $params, string $role, string &$error = ''): string
  1479. {
  1480. $payload = $this->resolveFrameImagePayload($params, $role, $error);
  1481. return (string)($payload['api_url'] ?? '');
  1482. }
  1483. /**
  1484. * form-data 上传的图片落盘并同步 OSS
  1485. * @return array{url:string,local_path:string,object_key:string}
  1486. */
  1487. private function uploadFrameFileToOss($file, string $roleLabel, string &$error = '', string $taskId = ''): array
  1488. {
  1489. $empty = ['url' => '', 'local_path' => '', 'object_key' => ''];
  1490. if ($taskId === '') {
  1491. $taskId = self::FRAME_OSS_STAGING_ID;
  1492. }
  1493. $error = '';
  1494. $ext = $this->resolveUploadedImageExt($file);
  1495. if ($ext === '') {
  1496. $error = '不支持的图片格式';
  1497. Log::write('[uploadFrameFileToOss] ' . $error, 'error');
  1498. return $empty;
  1499. }
  1500. $uploadInfo = $file->getInfo();
  1501. $tmpPath = isset($uploadInfo['tmp_name']) ? (string)$uploadInfo['tmp_name'] : '';
  1502. if ($tmpPath === '' || !is_file($tmpPath)) {
  1503. $tmpPath = (string)($file->getRealPath() ?: '');
  1504. }
  1505. if ($tmpPath === '' || !is_file($tmpPath)) {
  1506. $uploadErr = $uploadInfo['error'] ?? UPLOAD_ERR_NO_FILE;
  1507. $error = '未接收到上传文件(错误码' . $uploadErr . ')';
  1508. Log::write('[uploadFrameFileToOss] ' . $error, 'error');
  1509. return $empty;
  1510. }
  1511. $saveDir = $this->buildTaskMediaLocalDir($taskId);
  1512. if (!is_dir($saveDir)) {
  1513. mkdir($saveDir, 0755, true);
  1514. }
  1515. $saveFileName = $roleLabel . '_' . str_replace('.', '', uniqid('', true)) . '.' . $ext;
  1516. $localFullPath = $saveDir . $saveFileName;
  1517. $saved = false;
  1518. if (method_exists($file, 'isValid') && $file->isValid()) {
  1519. $info = $file->move($saveDir, $saveFileName);
  1520. if ($info) {
  1521. $localFullPath = $saveDir . $info->getFilename();
  1522. $saved = true;
  1523. }
  1524. }
  1525. if (!$saved && !@copy($tmpPath, $localFullPath)) {
  1526. $moveErr = method_exists($file, 'getError') ? (string)$file->getError() : 'copy失败';
  1527. $error = '图片保存失败: ' . $moveErr;
  1528. Log::write('[uploadFrameFileToOss] ' . $error, 'error');
  1529. return $empty;
  1530. }
  1531. $objectKey = $this->buildTaskMediaObjectKey($taskId, basename($localFullPath));
  1532. $upload = $this->uploadToOSS($localFullPath, $objectKey);
  1533. if (!$upload['success']) {
  1534. $error = 'OSS上传失败';
  1535. Log::write('[uploadFrameFileToOss] ' . $error . ' | ' . $objectKey, 'error');
  1536. return $empty;
  1537. }
  1538. $url = $upload['url'] !== '' ? $upload['url'] : Common::ossFullUrl($objectKey);
  1539. return [
  1540. 'url' => $url,
  1541. 'local_path' => $localFullPath,
  1542. 'object_key' => $upload['object_key'] ?? $objectKey,
  1543. ];
  1544. }
  1545. /**
  1546. * 解析上传图片扩展名
  1547. */
  1548. private function resolveUploadedImageExt($file): string
  1549. {
  1550. $name = $file->getInfo('name') ?? '';
  1551. $ext = strtolower(pathinfo($name, PATHINFO_EXTENSION));
  1552. if ($ext === 'jpeg') {
  1553. $ext = 'jpg';
  1554. }
  1555. $allowed = ['jpg', 'png', 'gif', 'webp', 'bmp'];
  1556. if ($ext && in_array($ext, $allowed, true)) {
  1557. return $ext;
  1558. }
  1559. $mime = method_exists($file, 'getMime') ? strtolower((string)$file->getMime()) : '';
  1560. $map = [
  1561. 'image/jpeg' => 'jpg',
  1562. 'image/png' => 'png',
  1563. 'image/gif' => 'gif',
  1564. 'image/webp' => 'webp',
  1565. 'image/bmp' => 'bmp',
  1566. ];
  1567. return $map[$mime] ?? '';
  1568. }
  1569. /**
  1570. * base64 图片落盘并上传 OSS(前端传 data:image/...;base64,...)
  1571. * @return array{url:string,local_path:string,object_key:string}
  1572. */
  1573. private function uploadBase64ImageToOss(string $base64Input, string $roleLabel, string $taskId = '', string &$error = ''): array
  1574. {
  1575. $empty = ['url' => '', 'local_path' => '', 'object_key' => ''];
  1576. if ($taskId === '') {
  1577. $taskId = self::FRAME_OSS_STAGING_ID;
  1578. }
  1579. $error = '';
  1580. $parsed = $this->parseBase64Image($base64Input);
  1581. if ($parsed === null) {
  1582. $error = '图片Base64解析失败';
  1583. Log::write('[uploadBase64ImageToOss] ' . $error, 'error');
  1584. return $empty;
  1585. }
  1586. [$ext, $imageData] = $parsed;
  1587. $saveDir = $this->buildTaskMediaLocalDir($taskId);
  1588. if (!is_dir($saveDir)) {
  1589. mkdir($saveDir, 0755, true);
  1590. }
  1591. $fileName = $roleLabel . '_' . str_replace('.', '', uniqid('', true)) . '.' . $ext;
  1592. $localFullPath = $saveDir . $fileName;
  1593. if (file_put_contents($localFullPath, $imageData) === false) {
  1594. Log::write('[uploadBase64ImageToOss] 本地保存失败: ' . $localFullPath, 'error');
  1595. return $empty;
  1596. }
  1597. $objectKey = $this->buildTaskMediaObjectKey($taskId, $fileName);
  1598. $upload = $this->uploadToOSS($localFullPath, $objectKey);
  1599. if (!$upload['success']) {
  1600. $fallbackUrl = $this->buildPublicUploadUrl($localFullPath);
  1601. if ($fallbackUrl !== '') {
  1602. Log::write('[uploadBase64ImageToOss] OSS失败,回退本站URL: ' . $fallbackUrl, 'warning');
  1603. return [
  1604. 'url' => $fallbackUrl,
  1605. 'local_path' => $localFullPath,
  1606. 'object_key' => $objectKey,
  1607. ];
  1608. }
  1609. Log::write('[uploadBase64ImageToOss] OSS上传失败: ' . $objectKey, 'error');
  1610. return $empty;
  1611. }
  1612. $url = $upload['url'] !== '' ? $upload['url'] : Common::ossFullUrl($objectKey);
  1613. return [
  1614. 'url' => $url,
  1615. 'local_path' => $localFullPath,
  1616. 'object_key' => $upload['object_key'] ?? $objectKey,
  1617. ];
  1618. }
  1619. /**
  1620. * 解析 base64 图片(避免对大字符串整段正则,防止服务器 PCRE 超限)
  1621. * @return array{0:string,1:string}|null [扩展名, 二进制内容]
  1622. */
  1623. private function parseBase64Image(string $base64Input): ?array
  1624. {
  1625. $base64Input = trim($base64Input);
  1626. if ($base64Input === '') {
  1627. return null;
  1628. }
  1629. $ext = 'jpg';
  1630. $rawBase64 = $base64Input;
  1631. $prefix = 'data:image/';
  1632. if (stripos($base64Input, $prefix) === 0) {
  1633. $semi = stripos($base64Input, ';base64,');
  1634. if ($semi === false) {
  1635. return null;
  1636. }
  1637. $mimePart = strtolower(substr($base64Input, strlen($prefix), $semi - strlen($prefix)));
  1638. if ($mimePart === 'jpeg') {
  1639. $mimePart = 'jpg';
  1640. }
  1641. if (in_array($mimePart, ['jpg', 'png', 'gif', 'webp', 'bmp'], true)) {
  1642. $ext = $mimePart;
  1643. }
  1644. $rawBase64 = substr($base64Input, $semi + 8);
  1645. }
  1646. $rawBase64 = preg_replace('/\s+/', '', $rawBase64);
  1647. if ($rawBase64 === '') {
  1648. return null;
  1649. }
  1650. $imageData = base64_decode($rawBase64, true);
  1651. if ($imageData === false || strlen($imageData) < 100) {
  1652. return null;
  1653. }
  1654. return [$ext, $imageData];
  1655. }
  1656. /**
  1657. * 合并请求参数(兼容浏览器 application/json 提交 first_image/last_image base64)
  1658. */
  1659. private function mergeRequestParams(): array
  1660. {
  1661. $params = $this->request->param();
  1662. if (!empty($params['first_image']) || !empty($params['last_image']) || !empty($params['prompt'])) {
  1663. return $params;
  1664. }
  1665. $content = $this->request->getInput();
  1666. if (!is_string($content) || $content === '') {
  1667. return $params;
  1668. }
  1669. $json = json_decode($content, true);
  1670. if (!is_array($json)) {
  1671. return $params;
  1672. }
  1673. return array_merge($params, $json);
  1674. }
  1675. /**
  1676. * 将本地文件上传到阿里云 OSS
  1677. *
  1678. * @param string $file 本地完整路径,或 public 下相对路径(如 uploads/videos/20260604/xxx.mp4)
  1679. * @param string $objectKey OSS 对象键;传 uploads/ 开头或带扩展名则视为完整键,否则作为子目录标识(如 task_id)
  1680. * @return array{success:bool,object_key:string,url:string}
  1681. */
  1682. private function uploadToOSS(string $file, string $objectKey = ''): array
  1683. {
  1684. $localFullPath = $this->resolvePublicLocalPath($file);
  1685. if ($localFullPath === '') {
  1686. Log::write('[uploadToOSS] 本地文件不存在: ' . $file, 'error');
  1687. return ['success' => false, 'object_key' => '', 'url' => ''];
  1688. }
  1689. $fileName = basename($localFullPath);
  1690. if ($objectKey !== '' && (strpos($objectKey, 'uploads/') === 0 || preg_match('/\.[a-z0-9]{1,8}$/i', $objectKey))) {
  1691. $ossObjectKey = Common::normalizeOssObjectKey($objectKey);
  1692. } else {
  1693. $segments = ['uploads', 'videos', date('Ymd')];
  1694. if ($objectKey !== '') {
  1695. $segments[] = preg_replace('/[\\\\\/:*?"<>|]/u', '_', $objectKey);
  1696. }
  1697. $segments[] = $fileName;
  1698. $ossObjectKey = implode('/', $segments);
  1699. }
  1700. $success = Common::uploadLocalFileToOss($localFullPath, $ossObjectKey);
  1701. return [
  1702. 'success' => $success,
  1703. 'object_key' => $ossObjectKey,
  1704. 'url' => $success ? Common::ossFullUrl($ossObjectKey) : '',
  1705. ];
  1706. }
  1707. /**
  1708. * 入库路径:仅保留 uploads/ 及后面部分
  1709. */
  1710. private function toDbUploadPath(string $path): string
  1711. {
  1712. $path = trim(str_replace('\\', '/', $path));
  1713. if ($path === '') {
  1714. return '';
  1715. }
  1716. if (stripos($path, 'http://') === 0 || stripos($path, 'https://') === 0) {
  1717. $mark = '/uploads/';
  1718. $pos = stripos($path, $mark);
  1719. if ($pos !== false) {
  1720. return ltrim(substr($path, $pos + 1), '/');
  1721. }
  1722. return $path;
  1723. }
  1724. $path = ltrim($path, '/');
  1725. if (stripos($path, 'uploads/') === 0) {
  1726. return $path;
  1727. }
  1728. $pos = stripos($path, 'uploads/');
  1729. if ($pos !== false) {
  1730. return substr($path, $pos);
  1731. }
  1732. return $path;
  1733. }
  1734. /**
  1735. * 接口返回:库内相对路径转完整 URL
  1736. */
  1737. private function toPublicMediaUrl(string $dbPath): string
  1738. {
  1739. $dbPath = trim($dbPath);
  1740. if ($dbPath === '') {
  1741. return '';
  1742. }
  1743. if (stripos($dbPath, 'http://') === 0 || stripos($dbPath, 'https://') === 0) {
  1744. return $dbPath;
  1745. }
  1746. return Common::ossFullUrl($this->toDbUploadPath($dbPath));
  1747. }
  1748. /**
  1749. * 将 public 下本地文件转为可公网访问的站点 URL(OSS 不可用时的回退)
  1750. */
  1751. private function buildPublicUploadUrl(string $localFullPath): string
  1752. {
  1753. $publicRoot = str_replace('\\', '/', ROOT_PATH . 'public/');
  1754. $path = str_replace('\\', '/', $localFullPath);
  1755. if (strpos($path, $publicRoot) !== 0) {
  1756. return '';
  1757. }
  1758. $relative = ltrim(substr($path, strlen($publicRoot)), '/');
  1759. return $relative === '' ? '' : rtrim($this->request->domain(), '/') . '/' . $relative;
  1760. }
  1761. /**
  1762. * @return array{summary:string,content_length:int,post_max_size:string,upload_max_filesize:string,has_first_image_file:bool}
  1763. */
  1764. private function getPhpUploadEnvInfo(): array
  1765. {
  1766. $contentLength = (int)($_SERVER['CONTENT_LENGTH'] ?? 0);
  1767. $postMax = (string)ini_get('post_max_size');
  1768. $uploadMax = (string)ini_get('upload_max_filesize');
  1769. $hasFile = !empty($this->request->file('first_image'));
  1770. $summary = 'CONTENT_LENGTH=' . $contentLength
  1771. . ';post_max_size=' . $postMax
  1772. . ';upload_max_filesize=' . $uploadMax
  1773. . ';has_first_image_file=' . ($hasFile ? 'yes' : 'no');
  1774. return [
  1775. 'summary' => $summary,
  1776. 'content_length' => $contentLength,
  1777. 'post_max_size' => $postMax,
  1778. 'upload_max_filesize' => $uploadMax,
  1779. 'has_first_image_file' => $hasFile,
  1780. ];
  1781. }
  1782. /**
  1783. * 解析为 public 目录下的本地绝对路径
  1784. */
  1785. private function resolvePublicLocalPath(string $file): string
  1786. {
  1787. $file = str_replace('\\', '/', trim($file));
  1788. if ($file === '') {
  1789. return '';
  1790. }
  1791. if (is_file($file)) {
  1792. return $file;
  1793. }
  1794. $publicPath = str_replace('\\', '/', ROOT_PATH . 'public/' . ltrim($file, '/'));
  1795. return is_file($publicPath) ? $publicPath : '';
  1796. }
  1797. }