Golang rest.Context类代码示例

func Encoder(c *rest.Context, data interface{}) error {
	var resp = new(struct {
		Code     int         `json:"code"`
		Status   string      `json:"status"`
		Success  bool        `json:"success"`
		Location string      `json:"location,omitempty"`
		Error    string      `json:"error,omitempty"`
		Data     interface{} `json:"data,omitempty"`
		*httpfs.FileInfo
	})
	switch data := data.(type) {
	case error:
		resp.Error = data.Error()
	case *rest.RedirectURL:
		resp.Location = string(data.URL)
	case *httpfs.FileInfo:
		resp.FileInfo = data
		resp.Location = data.Name
	case *urlinfo.Info:
		resp.Data = data
	default:
		resp.Data = data
	}
	resp.Code = c.Status()
	resp.Status = http.StatusText(resp.Code)
	resp.Success = resp.Code < 400
	c.SetContentType("application/json; charset=utf-8")
	enc := json.NewEncoder(c.Response)
	enc.SetIndent("", "  ")
	return enc.Encode(resp)
}

// Save сохраняет переданный в запросе файл в хранилище.
func (fs *FileStore) Save(c *rest.Context) error {
	fileInfo, err := fs.Store.Create(c.Request.Body) // сохраняем в хранилище
	if err != nil {
		return err
	}
	// файл успешно сохранен
	var url = c.Request.URL
	if !strings.HasSuffix(url.Path, "/") {
		// добавляем к URL запроса "/" в конце, чтобы корректно обработать
		// объединение относительного пути
		url.Path += "/"
	}
	// объединяем текущий URL запроса с именем сохраненного файла
	url, _ = url.Parse(fileInfo.Name)
	urlStr := url.String() // полный URL для доступа к файлу
	fileInfo.Name = urlStr
	c.SetHeader("Location", urlStr)
	log.WithFields(log.Fields{
		"url":  urlStr,
		"mime": fileInfo.Mimetype,
		"size": fileInfo.Size,
	}).Debug("file saved")
	// отдаем ответ с информацией о файле
	c.SetStatus(http.StatusCreated)
	return c.Write(fileInfo)
}

// URLInfo возвращает информацию об указанном в запросе URL.
func URLInfo(c *rest.Context) error {
	urlStr := c.Param("url")
	if info := urlinfo.Get(urlStr); info != nil {
		return c.Write(info)
	}
	return c.Error(http.StatusBadRequest,
		fmt.Sprintf("bad urlinfo request %s", urlStr))
}

// Get отдает файл с аватаркой
func (a *Avatars) Get(c *rest.Context) error {
	// получаем имя MX-сервера из пути
	mxname := c.Param("mx-name")
	// получаем идентификатор пользователя
	uid := c.Param("filename")
	// вычисляем полный путь к файлу с аватаркой
	avatar := filepath.Join(a.Path, mxname, uid)
	return c.ServeFile(avatar)
}

// Authorize проверяет авторизацию пользователя и только в случае успешной
// авторизации запускает обработчик запроса.
func (a *Auth) Authorize(c *rest.Context) error {
	// получаем имя MX-сервера из пути
	mxname := c.Param("mx-name")
	// проверяем псевдонимы и делаем редирект на основное имя
	if name, ok := a.aliases[mxname]; ok {
		var url = strings.Replace(c.Request.URL.Path, mxname, name, 1)
		var status = http.StatusMovedPermanently
		var method = c.Request.Method
		if method != "GET" && method != "HEAD" {
			status = http.StatusPermanentRedirect
		}
		return c.Redirect(status, url)
	}
	// редирект для данного псевдонима не определен
	auth, ok := a.mxlist[mxname]
	// проверяем, что сервер с таким именем определен
	if !ok {
		return c.Error(http.StatusNotFound, fmt.Sprintf("mx %s not found", mxname))
	}
	// проверяем авторизацию, переданную в запросе
	login, password, ok := c.BasicAuth()
	if !ok {
		realm := fmt.Sprintf("Basic realm=%q", mxname)
		c.SetHeader("WWW-Authenticate", realm)
		return rest.ErrUnauthorized
	}
	// проверяем авторизацию пользователя на сервере MX или в кеше
	ui, err := auth.Login(login, password)
	switch err := err.(type) {
	case nil:
		break
	case *mxlogin.Error: // неверная авторизация
		// if err.Code == 6 {
		// 	break // пользователь уже залогинен
		// }
		return c.Error(http.StatusForbidden, err.Error())
	default: // проблемы с подключением к MX серверу
		return c.Error(http.StatusServiceUnavailable, err.Error())
	}
	// сохраняем идентификатор пользователя
	c.SetData(userID, ui.ID)
	return nil
}

// Get отдает файл из хранилища файлов по запросу HTTP.
func (fs *FileStore) Get(c *rest.Context) (err error) {
	filename := c.Param("filename")                            // имя файла в хранилище
	return fs.Store.ServeFile(c.Response, c.Request, filename) // отдаем как есть
}

// AvatarGenerator возвращает сгенерированную аватарку пользователя
func AvatarGenerator(c *rest.Context) error {
	c.SetContentType("image/png")
	return c.Write(icon.Render([]byte(c.Param("id"))))
}

// Save сохраняет аватарку пользователя в файле
func (a *Avatars) Save(c *rest.Context) error {
	// получаем имя MX-сервера из пути
	mxname := c.Param("mx-name")
	// получаем идентификатор пользователя
	uid, ok := c.Data(userID).(uint64)
	if !ok {
		return c.Error(http.StatusInternalServerError, "bad user ID (JID)")
	}
	// создаем катало для сохранения аватарки
	fldr := filepath.Join(a.Path, mxname)
	if err := os.MkdirAll(fldr, 0700); err != nil {
		return c.Error(http.StatusInternalServerError, err.Error())
	}
	// добавляем идентификатор пользователя, как имя файла
	id := strconv.FormatUint(uid, 10)
	avatar := filepath.Join(fldr, id)
	// создаем файл
	file, err := os.Create(avatar)
	if err != nil {
		return c.Error(http.StatusInternalServerError, err.Error())
	}
	// копируем в файл содержимое запроса
	_, err = io.Copy(file, c.Request.Body)
	file.Close()
	if err != nil {
		return c.Error(http.StatusInternalServerError, err.Error())
	}
	// сохраняем дату файла в списке
	fi, err := os.Stat(avatar)
	if err != nil {
		return c.Error(http.StatusInternalServerError, err.Error())
	}
	a.mu.Lock()
	mxlist := a.list[mxname]
	if mxlist == nil {
		mxlist = make(map[uint64]time.Time)
	}
	mxlist[uid] = fi.ModTime()
	a.list[mxname] = mxlist
	a.mu.Unlock()
	// отдаем в заголовке путь к созданному файлу
	c.SetHeader("Location", path.Join(c.Request.URL.Path, id))
	c.SetStatus(http.StatusCreated)
	return c.Write(nil)
}

// List возвращает список сохраненных аватарок
func (a *Avatars) List(c *rest.Context) error {
	a.mu.RLock()
	defer a.mu.RUnlock()
	return c.Write(a.list[c.Param("mx-name")])
}