]> BookStack Code Mirror - bookstack/blob - database/seeders/DummyContentSeeder.php
Vectors: Got basic LLM querying working using vector search context
[bookstack] / database / seeders / DummyContentSeeder.php
1 <?php
2
3 namespace Database\Seeders;
4
5 use BookStack\Api\ApiToken;
6 use BookStack\Entities\Models\Book;
7 use BookStack\Entities\Models\Bookshelf;
8 use BookStack\Entities\Models\Chapter;
9 use BookStack\Entities\Models\Page;
10 use BookStack\Permissions\JointPermissionBuilder;
11 use BookStack\Permissions\Models\RolePermission;
12 use BookStack\Search\SearchIndex;
13 use BookStack\Users\Models\Role;
14 use BookStack\Users\Models\User;
15 use Illuminate\Database\Seeder;
16 use Illuminate\Support\Facades\Hash;
17 use Illuminate\Support\Str;
18
19 class DummyContentSeeder extends Seeder
20 {
21     /**
22      * Run the database seeds.
23      *
24      * @return void
25      */
26     public function run()
27     {
28         // Create an editor user
29         $editorUser = User::factory()->create();
30         $editorRole = Role::getRole('editor');
31         $additionalEditorPerms = ['receive-notifications', 'comment-create-all'];
32         $editorRole->permissions()->syncWithoutDetaching(RolePermission::whereIn('name', $additionalEditorPerms)->pluck('id'));
33         $editorUser->attachRole($editorRole);
34
35         // Create a viewer user
36         $viewerUser = User::factory()->create();
37         $role = Role::getRole('viewer');
38         $viewerUser->attachRole($role);
39
40         $byData = ['created_by' => $editorUser->id, 'updated_by' => $editorUser->id, 'owned_by' => $editorUser->id];
41
42         Book::factory()->count(5)->create($byData)
43             ->each(function ($book) use ($byData) {
44                 $chapters = Chapter::factory()->count(3)->create($byData)
45                     ->each(function ($chapter) use ($book, $byData) {
46                         $pages = Page::factory()->count(3)->make(array_merge($byData, ['book_id' => $book->id]));
47                         $chapter->pages()->saveMany($pages);
48                     });
49                 $pages = Page::factory()->count(3)->make($byData);
50                 $book->chapters()->saveMany($chapters);
51                 $book->pages()->saveMany($pages);
52             });
53
54         $largeBook = Book::factory()->create(array_merge($byData, ['name' => 'Large book' . Str::random(10)]));
55         $pages = Page::factory()->count(200)->make($byData);
56         $chapters = Chapter::factory()->count(50)->make($byData);
57         $largeBook->pages()->saveMany($pages);
58         $largeBook->chapters()->saveMany($chapters);
59
60         $shelves = Bookshelf::factory()->count(10)->create($byData);
61         $largeBook->shelves()->attach($shelves->pluck('id'));
62
63         // Assign API permission to editor role and create an API key
64         $apiPermission = RolePermission::getByName('access-api');
65         $editorRole->attachPermission($apiPermission);
66         $token = (new ApiToken())->forceFill([
67             'user_id'    => $editorUser->id,
68             'name'       => 'Testing API key',
69             'expires_at' => ApiToken::defaultExpiry(),
70             'secret'     => Hash::make('password'),
71             'token_id'   => 'apitoken',
72         ]);
73         $token->save();
74
75         app(JointPermissionBuilder::class)->rebuildForAll();
76         app(SearchIndex::class)->indexAllEntities();
77     }
78 }