-
Notifications
You must be signed in to change notification settings - Fork 307
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
7 changed files
with
194 additions
and
59 deletions.
There are no files selected for viewing
149 changes: 149 additions & 0 deletions
149
internal/database/migrations/20240607113052_modify_functions.sql
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,149 @@ | ||
| -- +goose Up | ||
| -- +goose StatementBegin | ||
| CREATE OR REPLACE FUNCTION teldrive.get_file_from_path(full_path text,u_id bigint) | ||
| RETURNS setof teldrive.files AS $$ | ||
| DECLARE | ||
| target_id text; | ||
| begin | ||
|
|
||
| IF full_path = '/' then | ||
| RETURN QUERY select * from teldrive.files as root where root.parent_id = 'root'; | ||
| END IF; | ||
|
|
||
| WITH RECURSIVE dir_hierarchy AS ( | ||
| SELECT | ||
| root.id, | ||
| root.name, | ||
| root.parent_id, | ||
| 0 AS depth, | ||
| '' as path | ||
| FROM | ||
| teldrive.files as root | ||
| WHERE | ||
| root.parent_id = 'root' AND root.user_id = u_id | ||
|
|
||
| UNION ALL | ||
|
|
||
| SELECT | ||
| f.id, | ||
| f.name, | ||
| f.parent_id, | ||
| dh.depth + 1 AS depth, | ||
| dh.path || '/' || f.name | ||
| FROM | ||
| teldrive.files f | ||
| JOIN | ||
| dir_hierarchy dh ON dh.id = f.parent_id | ||
| WHERE f.type = 'folder' AND f.user_id = u_id | ||
| ) | ||
|
|
||
| SELECT id into target_id FROM dir_hierarchy dh | ||
| WHERE dh.path = full_path | ||
| ORDER BY dh.depth DESC | ||
| LIMIT 1; | ||
|
|
||
| RETURN QUERY select * from teldrive.files where id=target_id; | ||
|
|
||
| END; | ||
| $$ LANGUAGE plpgsql; | ||
|
|
||
| CREATE OR REPLACE FUNCTION teldrive.create_directories(u_id bigint, long_path text) | ||
| RETURNS SETOF teldrive.files AS $$ | ||
| DECLARE | ||
| path_parts TEXT[]; | ||
| current_directory_id TEXT; | ||
| new_directory_id TEXT; | ||
| directory_name TEXT; | ||
| path_so_far TEXT; | ||
| depth_dir INTEGER; | ||
| BEGIN | ||
| path_parts := string_to_array(regexp_replace(long_path, '^/+', ''), '/'); | ||
| path_so_far := ''; | ||
| depth_dir := 0; | ||
|
|
||
| SELECT id INTO current_directory_id FROM teldrive.files WHERE parent_id = 'root'; | ||
|
|
||
| FOR directory_name IN SELECT unnest(path_parts) LOOP | ||
| depth_dir := depth_dir + 1; | ||
|
|
||
| SELECT id INTO new_directory_id | ||
| FROM teldrive.files | ||
| WHERE parent_id = current_directory_id | ||
| AND "name" = directory_name | ||
| AND "user_id" = u_id; | ||
|
|
||
| IF new_directory_id IS NULL THEN | ||
| INSERT INTO teldrive.files ("name", "type", mime_type, parent_id, "user_id", starred, "depth") | ||
| VALUES (directory_name, 'folder', 'drive/folder', current_directory_id, u_id, false, depth_dir) | ||
| RETURNING id INTO new_directory_id; | ||
| END IF; | ||
|
|
||
| current_directory_id := new_directory_id; | ||
| END LOOP; | ||
|
|
||
| RETURN QUERY SELECT * FROM teldrive.files WHERE id = current_directory_id; | ||
| END; | ||
| $$ LANGUAGE plpgsql; | ||
|
|
||
| CREATE OR REPLACE FUNCTION teldrive.move_directory(src text, dest text, u_id bigint) | ||
| RETURNS void AS $$ | ||
| DECLARE | ||
| src_parent TEXT; | ||
| src_base TEXT; | ||
| dest_parent TEXT; | ||
| dest_base TEXT; | ||
| dest_id text; | ||
| dest_parent_id text; | ||
| src_id text; | ||
| begin | ||
|
|
||
| select id into src_id from teldrive.get_file_from_path(src,u_id); | ||
|
|
||
| select id into dest_id from teldrive.get_file_from_path(dest,u_id); | ||
|
|
||
| IF src_id is NULL THEN | ||
| RAISE EXCEPTION 'source directory not found'; | ||
| END IF; | ||
|
|
||
| IF dest_id is not NULL then | ||
| RAISE EXCEPTION 'destination directory exists'; | ||
| END IF; | ||
|
|
||
| SELECT parent, base INTO src_parent,src_base FROM teldrive.split_path(src); | ||
|
|
||
| SELECT parent, base INTO dest_parent, dest_base FROM teldrive.split_path(dest); | ||
|
|
||
| IF src_parent != dest_parent then | ||
| select id into dest_id from teldrive.create_directories(u_id,dest); | ||
| UPDATE teldrive.files SET parent_id = dest_id WHERE parent_id = src_id; | ||
| delete from teldrive.files where id = src_id; | ||
| END IF; | ||
|
|
||
| IF src_base != dest_base and src_parent = dest_parent then | ||
| UPDATE teldrive.files SET name = dest_base WHERE id = src_id; | ||
| END IF; | ||
|
|
||
| END; | ||
| $$ LANGUAGE plpgsql; | ||
|
|
||
| CREATE OR REPLACE FUNCTION teldrive.move_items(file_ids text[], dest text, u_id bigint) | ||
| RETURNS void AS $$ | ||
| declare | ||
| dest_id TEXT; | ||
| BEGIN | ||
|
|
||
| select id into dest_id from teldrive.get_file_from_path(dest,u_id); | ||
|
|
||
| IF dest_id is NULL then | ||
| select id into dest_id from teldrive.create_directories(u_id,dest); | ||
| END IF; | ||
|
|
||
| UPDATE teldrive.files | ||
| SET parent_id = dest_id | ||
| WHERE id = ANY(file_ids); | ||
| END; | ||
| $$ LANGUAGE plpgsql; | ||
|
|
||
| DROP INDEX IF EXISTS "path_idx"; | ||
| ALTER TABLE "teldrive"."files" DROP COLUMN IF EXISTS "path"; | ||
| -- +goose StatementEnd |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters