9862a57b73
There are no consumers of any of this logic nor is it complete. We're better off without it in the meantime until we have a use case and a more cohesive model for expressing it.
120 lines
3.1 KiB
Go
120 lines
3.1 KiB
Go
// Copyright 2015 The Chihaya Authors. All rights reserved.
|
|
// Use of this source code is governed by the BSD 2-Clause license,
|
|
// which can be found in the LICENSE file.
|
|
|
|
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/url"
|
|
"runtime"
|
|
|
|
"github.com/julienschmidt/httprouter"
|
|
|
|
"github.com/chihaya/chihaya/stats"
|
|
"github.com/chihaya/chihaya/tracker/models"
|
|
)
|
|
|
|
const jsonContentType = "application/json; charset=UTF-8"
|
|
|
|
func handleError(err error) (int, error) {
|
|
if err == nil {
|
|
return http.StatusOK, nil
|
|
} else if _, ok := err.(models.NotFoundError); ok {
|
|
stats.RecordEvent(stats.ClientError)
|
|
return http.StatusNotFound, nil
|
|
} else if _, ok := err.(models.ClientError); ok {
|
|
stats.RecordEvent(stats.ClientError)
|
|
return http.StatusBadRequest, nil
|
|
}
|
|
return http.StatusInternalServerError, err
|
|
}
|
|
|
|
func (s *Server) check(w http.ResponseWriter, r *http.Request, p httprouter.Params) (int, error) {
|
|
_, err := w.Write([]byte("STILL-ALIVE"))
|
|
return handleError(err)
|
|
}
|
|
|
|
func (s *Server) stats(w http.ResponseWriter, r *http.Request, p httprouter.Params) (int, error) {
|
|
w.Header().Set("Content-Type", jsonContentType)
|
|
|
|
var err error
|
|
var val interface{}
|
|
query := r.URL.Query()
|
|
|
|
stats.DefaultStats.GoRoutines = runtime.NumGoroutine()
|
|
|
|
if _, flatten := query["flatten"]; flatten {
|
|
val = stats.DefaultStats.Flattened()
|
|
} else {
|
|
val = stats.DefaultStats
|
|
}
|
|
|
|
if _, pretty := query["pretty"]; pretty {
|
|
var buf []byte
|
|
buf, err = json.MarshalIndent(val, "", " ")
|
|
|
|
if err == nil {
|
|
_, err = w.Write(buf)
|
|
}
|
|
} else {
|
|
err = json.NewEncoder(w).Encode(val)
|
|
}
|
|
|
|
return handleError(err)
|
|
}
|
|
|
|
func (s *Server) getTorrent(w http.ResponseWriter, r *http.Request, p httprouter.Params) (int, error) {
|
|
infohash, err := url.QueryUnescape(p.ByName("infohash"))
|
|
if err != nil {
|
|
return http.StatusNotFound, err
|
|
}
|
|
|
|
torrent, err := s.tracker.FindTorrent(infohash)
|
|
if err != nil {
|
|
return handleError(err)
|
|
}
|
|
|
|
w.Header().Set("Content-Type", jsonContentType)
|
|
e := json.NewEncoder(w)
|
|
return handleError(e.Encode(torrent))
|
|
}
|
|
|
|
func (s *Server) putTorrent(w http.ResponseWriter, r *http.Request, p httprouter.Params) (int, error) {
|
|
var torrent models.Torrent
|
|
err := json.NewDecoder(r.Body).Decode(&torrent)
|
|
if err != nil {
|
|
return http.StatusBadRequest, err
|
|
}
|
|
|
|
s.tracker.PutTorrent(&torrent)
|
|
return http.StatusOK, nil
|
|
}
|
|
|
|
func (s *Server) delTorrent(w http.ResponseWriter, r *http.Request, p httprouter.Params) (int, error) {
|
|
infohash, err := url.QueryUnescape(p.ByName("infohash"))
|
|
if err != nil {
|
|
return http.StatusNotFound, err
|
|
}
|
|
|
|
s.tracker.DeleteTorrent(infohash)
|
|
return http.StatusOK, nil
|
|
}
|
|
|
|
func (s *Server) getClient(w http.ResponseWriter, r *http.Request, p httprouter.Params) (int, error) {
|
|
if err := s.tracker.ClientApproved(p.ByName("clientID")); err != nil {
|
|
return http.StatusNotFound, err
|
|
}
|
|
return http.StatusOK, nil
|
|
}
|
|
|
|
func (s *Server) putClient(w http.ResponseWriter, r *http.Request, p httprouter.Params) (int, error) {
|
|
s.tracker.PutClient(p.ByName("clientID"))
|
|
return http.StatusOK, nil
|
|
}
|
|
|
|
func (s *Server) delClient(w http.ResponseWriter, r *http.Request, p httprouter.Params) (int, error) {
|
|
s.tracker.DeleteClient(p.ByName("clientID"))
|
|
return http.StatusOK, nil
|
|
}
|