src/ZGoBackend.hs

@@ -1189,7 +1189,22 @@ routes pipe config = do
                              liftAndCatchIO $ run (upsertViewingKey o' q)
                              status created201
                      else status forbidden403
-              else status badRequest400 -- TODO: add Unified VK support
+              else case decodeUfvk (C.pack q) of
+                     Nothing -> status badRequest400
+                     Just fvk ->
+                       if matchOrchardAddress
+                            (C.pack q)
+                            (C.pack . T.unpack $ uaddress u')
+                         then do
+                           owner <-
+                             liftAndCatchIO $ run (findOwner $ uaddress u')
+                           case cast' . Doc =<< owner of
+                             Nothing -> status badRequest400
+                             Just o' -> do
+                               unless (oviewkey o' /= "") $ do
+                                 liftAndCatchIO $ run (upsertViewingKey o' q)
+                                 status created201
+                         else status forbidden403
     --Get items associated with the given address
   get "/api/items" $ do
     session <- param "session"