AiModelController.php 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160
  1. <?php
  2. namespace App\Http\Controllers;
  3. use App\Http\Requests\StoreAiModelRequest;
  4. use App\Http\Requests\UpdateAiModelRequest;
  5. use App\Models\AiModel;
  6. use Illuminate\Http\Request;
  7. use App\Http\Api\AuthApi;
  8. use Illuminate\Support\Str;
  9. use Illuminate\Support\Facades\Log;
  10. use App\Http\Api\StudioApi;
  11. use App\Http\Resources\AiModelResource;
  12. class AiModelController extends Controller
  13. {
  14. /**
  15. * Display a listing of the resource.
  16. *
  17. * @return \Illuminate\Http\Response
  18. */
  19. public function index(Request $request)
  20. {
  21. //
  22. $user = AuthApi::current($request);
  23. if (!$user) {
  24. Log::error('notification auth failed {request}', ['request' => $request]);
  25. return $this->error(__('auth.failed'), 401, 401);
  26. }
  27. switch ($request->get('view')) {
  28. case 'all':
  29. $table = AiModel::whereNotNull('owner_id');
  30. break;
  31. case 'studio':
  32. $studioId = StudioApi::getIdByName($request->get('name'));
  33. $table = AiModel::where('owner_id', $studioId);
  34. break;
  35. case 'usable':
  36. $table = AiModel::where('owner_id', $request->get('user_id'))
  37. ->orWhere('privacy', 'public');
  38. break;
  39. }
  40. if ($request->has('keyword')) {
  41. $table = $table->where('name', 'like', '%' . $request->get('keyword') . '%');
  42. }
  43. $count = $table->count();
  44. $table = $table->orderBy(
  45. $request->get('order', 'created_at'),
  46. $request->get('dir', 'asc')
  47. );
  48. $table = $table->skip($request->get("offset", 0))
  49. ->take($request->get('limit', 1000));
  50. $result = $table->get();
  51. return $this->ok(
  52. [
  53. "rows" => AiModelResource::collection(resource: $result),
  54. "count" => $count,
  55. ]
  56. );
  57. }
  58. /**
  59. * Store a newly created resource in storage.
  60. *
  61. * @param \App\Http\Requests\StoreAiModelRequest $request
  62. * @return \Illuminate\Http\Response
  63. */
  64. public function store(StoreAiModelRequest $request)
  65. {
  66. //
  67. $user = AuthApi::current($request);
  68. if (!$user) {
  69. return $this->error(__('auth.failed'), 401, 401);
  70. }
  71. $studioId = StudioApi::getIdByName($request->get('studio_name'));
  72. Log::debug('store', ['studioId' => $studioId, 'user' => $user]);
  73. if (!self::canEdit($user['user_uid'], $studioId)) {
  74. return $this->error(__('auth.failed'), 403, 403);
  75. }
  76. $new = new AiModel();
  77. $new->name = $request->get('name');
  78. $new->uid = Str::uuid();
  79. $new->real_name = Str::uuid();
  80. $new->owner_id = $studioId;
  81. $new->editor_id = $user['user_uid'];
  82. $new->save();
  83. return $this->ok(new AiModelResource($new));
  84. }
  85. /**
  86. * Display the specified resource.
  87. *
  88. * @param \App\Models\AiModel $aiModel
  89. * @return \Illuminate\Http\Response
  90. */
  91. public function show(AiModel $aiModel)
  92. {
  93. //
  94. return $this->ok(new AiModelResource($aiModel));
  95. }
  96. /**
  97. * Update the specified resource in storage.
  98. *
  99. * @param \App\Http\Requests\UpdateAiModelRequest $request
  100. * @param \App\Models\AiModel $aiModel
  101. * @return \Illuminate\Http\Response
  102. */
  103. public function update(UpdateAiModelRequest $request, AiModel $aiModel)
  104. {
  105. //
  106. $user = AuthApi::current($request);
  107. if (!$user) {
  108. return $this->error(__('auth.failed'), 401, 401);
  109. }
  110. if (!self::canEdit($user['user_uid'], $aiModel->owner_id)) {
  111. return $this->error(__('auth.failed'), 403, 403);
  112. }
  113. $aiModel->name = $request->get('name');
  114. $aiModel->description = $request->get('description');
  115. $aiModel->system_prompt = $request->get('system_prompt');
  116. $aiModel->url = $request->get('url');
  117. $aiModel->model = $request->get('model');
  118. $aiModel->key = $request->get('key');
  119. $aiModel->privacy = $request->get('privacy');
  120. $aiModel->editor_id = $user['user_uid'];
  121. $aiModel->save();
  122. return $this->ok(new AiModelResource($aiModel));
  123. }
  124. /**
  125. * Remove the specified resource from storage.
  126. *
  127. * @param \App\Models\AiModel $aiModel
  128. * @return \Illuminate\Http\Response
  129. */
  130. public function destroy(Request $request, AiModel $aiModel)
  131. {
  132. //
  133. $user = AuthApi::current($request);
  134. if (!$user) {
  135. return $this->error(__('auth.failed'), 401, 401);
  136. }
  137. if (!self::canEdit($user['user_uid'], $aiModel->owner_id)) {
  138. return $this->error(__('auth.failed'), 403, 403);
  139. }
  140. $del = $aiModel->delete();
  141. return $this->ok($del);
  142. }
  143. public static function canEdit($user_uid, $owner_uid)
  144. {
  145. return $user_uid === $owner_uid;
  146. }
  147. }