Fix query endpoint for cif files

soedinglab · May 15, 2024 · fe64840 · fe64840
1 parent 8364a21
commit fe64840
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/backend/server.go b/backend/server.go
@@ -598,14 +598,23 @@ func server(jobsystem JobSystem, config ConfigRoot) {
 			http.Error(w, err.Error(), http.StatusBadRequest)
 			return
 		}
-		var suffix string
+		var queryPath string
 		switch config.App {
 		case "foldseek":
-			suffix = ".pdb"
+			pdbPath := filepath.Join(config.Paths.Results, string(ticket.Id), "job.pdb")
+			cifPath := filepath.Join(config.Paths.Results, string(ticket.Id), "job.cif")
+			if fileExists(pdbPath) {
+				queryPath = pdbPath
+			} else if fileExists(cifPath) {
+				queryPath = cifPath
+			} else {
+				http.Error(w, "File not found", http.StatusBadRequest)
+				return
+			}
 		default:
-			suffix = ".fasta"
+			queryPath = filepath.Join(config.Paths.Results, string(ticket.Id), "job.fasta")
 		}
-		query, err := os.ReadFile(filepath.Join(config.Paths.Results, string(ticket.Id), "job"+suffix))
+		query, err := os.ReadFile(queryPath)
 		if err != nil {
 			http.Error(w, err.Error(), http.StatusBadRequest)
 			return