FastMovieAI AI短剧创作平台二开基线

- 源码: xhadmincn/FastMovieAI (Apache 2.0) - 二开: Docker部署, Swoole改Select, route.php修复, 补update/VERSION - vendor/示例资源已gitignore
2026-08-26 18:37:36 +08:00
commit eae151fd57
888 changed files with 105571 additions and 0 deletions
@@ -0,0 +1,92 @@
+<?php
+
+namespace plugin\model\process\video;
+
+use plugin\control\expose\helper\Uploads;
+use plugin\model\app\model\PluginModelTask;
+use plugin\model\app\model\PluginModelTaskResult;
+use plugin\model\utils\enum\ModelScene;
+use plugin\model\utils\enum\ModelTaskStatus;
+use plugin\model\utils\enum\ModelType;
+use plugin\notification\expose\helper\Push;
+use plugin\shortplay\app\model\PluginShortplayDramaStoryboard;
+use support\Log;
+use think\facade\Db;
+use Workerman\Coroutine;
+use Workerman\Crontab\Crontab;
+use Workerman\Timer;
+
+class VideoTransfer
+{
+    public function onWorkerStart($worker)
+    {
+        $id = $worker->id;
+        new Crontab('*/5 * * * * *', function () use ($id) {
+            try {
+                if ($id) {
+                    Timer::sleep(0.3 * $id);
+                }
+                Coroutine::create(function () {
+                    $PluginModelTask = PluginModelTask::where(['status' => ModelTaskStatus::WAIT_DOWNLOAD['value'], 'model_type' => ModelType::TOVIDEO['value']])->order('last_heartbeat asc,id asc')->lock(true)->find();
+                    if (!$PluginModelTask) {
+                        return;
+                    }
+                    $PluginModelTask->status = ModelTaskStatus::DOWNLOADING['value'];
+                    $PluginModelTask->last_heartbeat = date('Y-m-d H:i:s', strtotime('+5 seconds'));
+                    $PluginModelTask->save();
+                    $PluginModelTaskResult = PluginModelTaskResult::where('task_id', $PluginModelTask->id)->find();
+                    if (!$PluginModelTaskResult) {
+                        return;
+                    }
+                    try {
+                        $ModelScene = ModelScene::get($PluginModelTask->scene);
+                        $classify = Uploads::getClassify($PluginModelTask->channels_uid, 'uploads/' . $PluginModelTask->scene, $ModelScene['label']);
+                        $result = Uploads::download($PluginModelTask->channels_uid, $PluginModelTaskResult->video, $classify);
+                    } catch (\Throwable $th) {
+                        $PluginModelTask = PluginModelTask::where('id', $PluginModelTask->id)->find();
+                        $PluginModelTask->status = ModelTaskStatus::WAIT_DOWNLOAD['value'];
+                        $PluginModelTask->save();
+                        Log::error("视频转存处理失败：" . $th->getMessage(), $th->getTrace());
+                        return;
+                    }
+                    $pushData = [
+                        'task_id' => $PluginModelTask->id,
+                        'id' => $PluginModelTask->alias_id,
+                        'video' => $result->file_url
+                    ];
+                    $event = 'generate' . strtolower(str_replace('_', '', $PluginModelTask->scene));
+                    Db::startTrans();
+                    try {
+                        $PluginModelTask = PluginModelTask::where('id', $PluginModelTask->id)->find();
+                        $PluginModelTask->status = ModelTaskStatus::SUCCESS['value'];
+                        $PluginModelTask->save();
+                        $PluginModelTaskResult = PluginModelTaskResult::where('task_id', $PluginModelTask->id)->find();
+                        $PluginModelTaskResult->video_path = $result->file_name;
+                        $PluginModelTaskResult->save();
+                        switch ($PluginModelTask->scene) {
+                            case ModelScene::STORYBOARD_VIDEO['value']:
+                                $PluginShortplayDramaStoryboard = PluginShortplayDramaStoryboard::where('id', $PluginModelTask->alias_id)->find();
+                                $PluginShortplayDramaStoryboard->video = $result->file_name;
+                                $PluginShortplayDramaStoryboard->use_material_type='video';
+                                $PluginShortplayDramaStoryboard->save();
+                                $event = 'generatestoryboard';
+                                $pushData['event'] = ModelScene::STORYBOARD_VIDEO['value'];
+                                break;
+                        }
+                        Db::commit();
+                    } catch (\Throwable $th) {
+                        Db::rollback();
+                        Log::error("视频转存保存失败：" . $th->getMessage(), $th->getTrace());
+                    }
+                    Push::send([
+                        'uid' => $PluginModelTask->uid,
+                        'channels_uid' => $PluginModelTask->channels_uid,
+                        'event' => $event
+                    ], $pushData);
+                });
+            } catch (\Throwable $th) {
+                Log::error("视频转存定时任务失败：" . $th->getMessage(), $th->getTrace());
+            }
+        });
+    }
+}