Files
sshwifty-udp-telnet-http/application/controller/base.go
2022-02-13 17:33:52 +08:00

133 lines
3.6 KiB
Go

// Sshwifty - A Web SSH client
//
// Copyright (C) 2019-2022 Ni Rui <ranqus@gmail.com>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
package controller
import (
"net/http"
"strings"
"github.com/nirui/sshwifty/application/log"
)
// Error
var (
ErrControllerNotImplemented = NewError(
http.StatusNotImplemented, "Server does not know how to handle the "+
"request")
)
type controller interface {
Get(w http.ResponseWriter, r *http.Request, l log.Logger) error
Head(w http.ResponseWriter, r *http.Request, l log.Logger) error
Post(w http.ResponseWriter, r *http.Request, l log.Logger) error
Put(w http.ResponseWriter, r *http.Request, l log.Logger) error
Delete(w http.ResponseWriter, r *http.Request, l log.Logger) error
Connect(w http.ResponseWriter, r *http.Request, l log.Logger) error
Options(w http.ResponseWriter, r *http.Request, l log.Logger) error
Trace(w http.ResponseWriter, r *http.Request, l log.Logger) error
Patch(w http.ResponseWriter, r *http.Request, l log.Logger) error
Other(
method string,
w http.ResponseWriter,
r *http.Request,
l log.Logger,
) error
}
type baseController struct{}
func (b baseController) Get(
w http.ResponseWriter, r *http.Request, l log.Logger) error {
return ErrControllerNotImplemented
}
func (b baseController) Head(
w http.ResponseWriter, r *http.Request, l log.Logger) error {
return ErrControllerNotImplemented
}
func (b baseController) Post(
w http.ResponseWriter, r *http.Request, l log.Logger) error {
return ErrControllerNotImplemented
}
func (b baseController) Put(
w http.ResponseWriter, r *http.Request, l log.Logger) error {
return ErrControllerNotImplemented
}
func (b baseController) Delete(
w http.ResponseWriter, r *http.Request, l log.Logger) error {
return ErrControllerNotImplemented
}
func (b baseController) Connect(
w http.ResponseWriter, r *http.Request, l log.Logger) error {
return ErrControllerNotImplemented
}
func (b baseController) Options(
w http.ResponseWriter, r *http.Request, l log.Logger) error {
return ErrControllerNotImplemented
}
func (b baseController) Trace(
w http.ResponseWriter, r *http.Request, l log.Logger) error {
return ErrControllerNotImplemented
}
func (b baseController) Patch(
w http.ResponseWriter, r *http.Request, l log.Logger) error {
return ErrControllerNotImplemented
}
func (b baseController) Other(
method string, w http.ResponseWriter, r *http.Request, l log.Logger) error {
return ErrControllerNotImplemented
}
func serveController(
c controller,
w http.ResponseWriter,
r *http.Request,
l log.Logger,
) error {
switch strings.ToUpper(r.Method) {
case "GET":
return c.Get(w, r, l)
case "HEAD":
return c.Head(w, r, l)
case "POST":
return c.Post(w, r, l)
case "PUT":
return c.Put(w, r, l)
case "DELETE":
return c.Delete(w, r, l)
case "CONNECT":
return c.Connect(w, r, l)
case "OPTIONS":
return c.Options(w, r, l)
case "TRACE":
return c.Trace(w, r, l)
case "PATCH":
return c.Patch(w, r, l)
default:
return c.Other(r.Method, w, r, l)
}
}