lib/controller/router/response.go

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 package router
   6
   7 import (
   8         "encoding/json"
   9         "fmt"
  10         "net/http"
  11         "regexp"
  12         "strings"
  13         "time"
  14
  15         "git.arvados.org/arvados.git/sdk/go/arvados"
  16         "git.arvados.org/arvados.git/sdk/go/httpserver"
  17 )
  18
  19 const rfc3339NanoFixed = "2006-01-02T15:04:05.000000000Z07:00"
  20
  21 type responseOptions struct {
  22         Select []string
  23         Count  string
  24 }
  25
  26 func (rtr *router) responseOptions(opts interface{}) (responseOptions, error) {
  27         var rOpts responseOptions
  28         switch opts := opts.(type) {
  29         case *arvados.GetOptions:
  30                 rOpts.Select = opts.Select
  31         case *arvados.ListOptions:
  32                 rOpts.Select = opts.Select
  33                 rOpts.Count = opts.Count
  34         }
  35         return rOpts, nil
  36 }
  37
  38 func applySelectParam(selectParam []string, orig map[string]interface{}) map[string]interface{} {
  39         if len(selectParam) == 0 {
  40                 return orig
  41         }
  42         selected := map[string]interface{}{}
  43         for _, attr := range selectParam {
  44                 if v, ok := orig[attr]; ok {
  45                         selected[attr] = v
  46                 }
  47         }
  48         // Some keys are always preserved, even if not requested
  49         for _, k := range []string{"etag", "kind", "writable_by"} {
  50                 if v, ok := orig[k]; ok {
  51                         selected[k] = v
  52                 }
  53         }
  54         return selected
  55 }
  56
  57 func (rtr *router) sendResponse(w http.ResponseWriter, req *http.Request, resp interface{}, opts responseOptions) {
  58         var tmp map[string]interface{}
  59
  60         if resp, ok := resp.(http.Handler); ok {
  61                 // resp knows how to write its own http response
  62                 // header and body.
  63                 resp.ServeHTTP(w, req)
  64                 return
  65         }
  66
  67         err := rtr.transcode(resp, &tmp)
  68         if err != nil {
  69                 rtr.sendError(w, err)
  70                 return
  71         }
  72
  73         respKind := kind(resp)
  74         if respKind == "arvados#sharedGroupList" {
  75                 tmp["kind"] = "arvados#groupList"
  76         } else if respKind != "" {
  77                 tmp["kind"] = respKind
  78         }
  79         defaultItemKind := ""
  80         if strings.HasSuffix(respKind, "List") {
  81                 defaultItemKind = strings.TrimSuffix(respKind, "List")
  82         }
  83
  84         if items, ok := tmp["items"].([]interface{}); ok {
  85                 for i, item := range items {
  86                         // Fill in "kind" by inspecting UUID/PDH if
  87                         // possible; fall back on assuming each
  88                         // Items[] entry in an "arvados#fooList"
  89                         // response should have kind="arvados#foo".
  90                         item, _ := item.(map[string]interface{})
  91                         infix := ""
  92                         if uuid, _ := item["uuid"].(string); len(uuid) == 27 {
  93                                 infix = uuid[6:11]
  94                         }
  95                         if k := kind(infixMap[infix]); k != "" {
  96                                 item["kind"] = k
  97                         } else if pdh, _ := item["portable_data_hash"].(string); pdh != "" {
  98                                 item["kind"] = "arvados#collection"
  99                         } else if defaultItemKind != "" {
 100                                 item["kind"] = defaultItemKind
 101                         }
 102                         item = applySelectParam(opts.Select, item)
 103                         rtr.mungeItemFields(item)
 104                         items[i] = item
 105                 }
 106                 if opts.Count == "none" {
 107                         delete(tmp, "items_available")
 108                 }
 109         } else {
 110                 tmp = applySelectParam(opts.Select, tmp)
 111                 rtr.mungeItemFields(tmp)
 112         }
 113
 114         w.Header().Set("Content-Type", "application/json")
 115         enc := json.NewEncoder(w)
 116         enc.SetEscapeHTML(false)
 117         enc.Encode(tmp)
 118 }
 119
 120 func (rtr *router) sendError(w http.ResponseWriter, err error) {
 121         code := http.StatusInternalServerError
 122         if err, ok := err.(interface{ HTTPStatus() int }); ok {
 123                 code = err.HTTPStatus()
 124         }
 125         httpserver.Error(w, err.Error(), code)
 126 }
 127
 128 var infixMap = map[string]interface{}{
 129         "4zz18": arvados.Collection{},
 130         "j7d0g": arvados.Group{},
 131 }
 132
 133 var mungeKind = regexp.MustCompile(`\..`)
 134
 135 func kind(resp interface{}) string {
 136         t := fmt.Sprintf("%T", resp)
 137         if !strings.HasPrefix(t, "arvados.") {
 138                 return ""
 139         }
 140         return mungeKind.ReplaceAllStringFunc(t, func(s string) string {
 141                 // "arvados.CollectionList" => "arvados#collectionList"
 142                 return "#" + strings.ToLower(s[1:])
 143         })
 144 }
 145
 146 func (rtr *router) mungeItemFields(tmp map[string]interface{}) {
 147         for k, v := range tmp {
 148                 if strings.HasSuffix(k, "_at") {
 149                         // Format non-nil timestamps as
 150                         // rfc3339NanoFixed (otherwise they would use
 151                         // the default time encoding, which omits
 152                         // trailing zeroes).
 153                         switch tv := v.(type) {
 154                         case *time.Time:
 155                                 if tv == nil || tv.IsZero() {
 156                                         tmp[k] = nil
 157                                 } else {
 158                                         tmp[k] = tv.Format(rfc3339NanoFixed)
 159                                 }
 160                         case time.Time:
 161                                 if tv.IsZero() {
 162                                         tmp[k] = nil
 163                                 } else {
 164                                         tmp[k] = tv.Format(rfc3339NanoFixed)
 165                                 }
 166                         case string:
 167                                 if tv == "" {
 168                                         tmp[k] = nil
 169                                 } else if t, err := time.Parse(time.RFC3339Nano, tv); err != nil {
 170                                         // pass through an invalid time value (?)
 171                                 } else if t.IsZero() {
 172                                         tmp[k] = nil
 173                                 } else {
 174                                         tmp[k] = t.Format(rfc3339NanoFixed)
 175                                 }
 176                         }
 177                 }
 178                 // Arvados API spec says when these fields are empty
 179                 // they appear in responses as null, rather than a
 180                 // zero value.
 181                 switch k {
 182                 case "output_uuid", "output_name", "log_uuid", "description", "requesting_container_uuid", "container_uuid":
 183                         if v == "" {
 184                                 tmp[k] = nil
 185                         }
 186                 case "container_count_max":
 187                         if v == float64(0) {
 188                                 tmp[k] = nil
 189                         }
 190                 }
 191         }
 192 }