1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.config.server.http;
import com.yahoo.config.application.api.ApplicationFile;
import com.yahoo.container.jdisc.HttpResponse;
import ai.vespa.http.HttpURL.Path;
import com.yahoo.restapi.SlimeJsonResponse;
import com.yahoo.slime.Cursor;
import com.yahoo.slime.Slime;
import java.io.InputStreamReader;
import java.util.Collections;
import java.util.List;
/**
* Requests for content and content status, both for prepared and active sessions,
* are handled by this class.
*
* @author hmusum
*/
public class ContentHandler {
public HttpResponse get(ContentRequest request) {
ContentRequest.ReturnType returnType = request.getReturnType();
String urlBase = request.getUrlBase("/content/");
if (ContentRequest.ReturnType.STATUS.equals(returnType)) {
return status(request, urlBase);
} else {
return content(request, urlBase);
}
}
public HttpResponse put(ContentRequest request) {
ApplicationFile file = request.getFile();
if (request.getPath().hasTrailingSlash()) {
createDirectory(request, file);
} else {
createFile(request, file);
}
return createResponse(request);
}
public HttpResponse delete(ContentRequest request) {
ApplicationFile file = request.getExistingFile();
deleteFile(file);
return createResponse(request);
}
private HttpResponse content(ContentRequest request, String urlBase) {
ApplicationFile file = request.getExistingFile();
if (file.isDirectory()) {
return new SessionContentListResponse(urlBase, listSortedFiles(file, request.getPath(), request.isRecursive()));
}
return new SessionContentReadResponse(file);
}
private HttpResponse status(ContentRequest request, String urlBase) {
ApplicationFile file = request.getFile();
if (file.isDirectory()) {
return new SessionContentStatusListResponse(urlBase, listSortedFiles(file, request.getPath(), request.isRecursive()));
}
return new SessionContentStatusResponse(file, urlBase);
}
private static List<ApplicationFile> listSortedFiles(ApplicationFile file, Path path, boolean recursive) {
if (path.length() > 0 && ! path.hasTrailingSlash()) {
return List.of(file);
}
List<ApplicationFile> files = file.listFiles(recursive);
Collections.sort(files);
return files;
}
private void createFile(ContentRequest request, ApplicationFile file) {
if ( ! request.hasRequestBody()) {
throw new BadRequestException("Request must contain body when creating a file");
}
try {
file.writeFile(new InputStreamReader(request.getData()));
} catch (Exception e) {
throw new RuntimeException(e);
}
}
@SuppressWarnings("StatementWithEmptyBody")
private void createDirectory(ContentRequest request, ApplicationFile file) {
if (request.hasRequestBody()) {
// TODO: Enable when we have a good way to check if request contains a body
// return new HttpErrorResponse(HttpResponse.Status.BAD_REQUEST, "Request should not contain a body when creating directories");
}
file.createDirectory();
}
private void deleteFile(ApplicationFile file) {
try {
file.delete();
} catch (RuntimeException e) {
throw new BadRequestException("File '" + file.getPath() + "' is not an empty directory");
}
}
private HttpResponse createResponse(ContentRequest request) {
Slime slime = new Slime();
Cursor root = slime.setObject();
root.setString("prepared", request.getUrlBase("/prepared"));
return new SlimeJsonResponse(slime);
}
}
|