From b87ba1e6bf827e350ac073b8a5afb30a865de524 Mon Sep 17 00:00:00 2001
From: Derrick Hammer <derrick@derrickhammer.com>
Date: Fri, 1 Mar 2024 22:50:22 -0500
Subject: [PATCH] feat: add support for webapp and dir cid's in dnslink

---
 api/s5/s5.go | 44 ++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 42 insertions(+), 2 deletions(-)

diff --git a/api/s5/s5.go b/api/s5/s5.go
index ae11643..08e7143 100644
--- a/api/s5/s5.go
+++ b/api/s5/s5.go
@@ -276,9 +276,41 @@ func (s *S5API) Handle(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	cid := cidVal.(encoding.CID)
+	cid := cidVal.(*encoding.CID)
 
-	file := s.newFile(s.protocol, cid.Hash.HashBytes())
+	if cid.Type == types.CIDTypeResolver {
+		entry, err := s.getNode().Services().Registry().Get(cid.Hash.FullBytes())
+		if err != nil {
+			s.logger.Error("Error getting registry entry", zap.Error(err))
+			w.WriteHeader(http.StatusInternalServerError)
+			return
+		}
+
+		cid, err = encoding.CIDFromRegistry(entry.Data())
+		if err != nil {
+			s.logger.Error("Error getting CID from registry entry", zap.Error(err))
+			w.WriteHeader(http.StatusInternalServerError)
+			return
+		}
+	}
+
+	switch cid.Type {
+	case types.CIDTypeRaw:
+		s.handleDnsLinkRaw(w, r, cid)
+	case types.CIDTypeMetadataWebapp:
+		s.handleDnsLinkWebapp(w, r, cid)
+	case types.CIDTypeDirectory:
+		s.handleDnsLinkDirectory(w, r, cid)
+	default:
+		w.WriteHeader(http.StatusUnsupportedMediaType)
+	}
+}
+
+func (s *S5API) handleDnsLinkRaw(w http.ResponseWriter, r *http.Request, cid *encoding.CID) {
+	file := s.newFile(FileParams{
+		Hash: cid.Hash.HashBytes(),
+		Type: cid.Type,
+	})
 
 	if !file.Exists() {
 		w.WriteHeader(http.StatusNotFound)
@@ -297,6 +329,14 @@ func (s *S5API) Handle(w http.ResponseWriter, r *http.Request) {
 	http.ServeContent(w, r, file.Name(), file.Modtime(), file)
 }
 
+func (s *S5API) handleDnsLinkWebapp(w http.ResponseWriter, r *http.Request, cid *encoding.CID) {
+	http.FileServer(http.FS(newWebAppFs(cid, s))).ServeHTTP(w, r)
+}
+
+func (s *S5API) handleDnsLinkDirectory(w http.ResponseWriter, r *http.Request, cid *encoding.CID) {
+	http.FileServer(http.FS(newDirFs(cid, s))).ServeHTTP(w, r)
+}
+
 type s5TusJwtResponseWriter struct {
 	http.ResponseWriter
 	req *http.Request