💾 Archived View for source.community › ckaznocha › gemini › blob › main › response.go captured on 2024-02-05 at 09:53:06. Gemini links have been rewritten to link to archived content

-=-=-=-=-=-=-

                                                         .
,-. ,-. . . ,-. ,-. ,-.    ,-. ,-. ,-,-. ,-,-. . . ,-. . |- . .
`-. | | | | |   |   |-'    |   | | | | | | | | | | | | | |  | |
`-' `-' `-^ '   `-' `-' :: `-' `-' ' ' ' ' ' ' `-^ ' ' ' `' `-|
                                                             /|
                                                            `-'

ckaznocha / gemini

git clone https://source.community/ckaznocha/gemini.git

Branches

Log

Tree

/response.go (main)

↑ /

blob

View raw contents of /response.go (main)

────╮
   1│ package gemini
   2│ 
   3│ import (
   4│ 	"context"
   5│ 	"io"
   6│ 	"mime"
   7│ )
   8│ 
   9│ // ResponseWriter is interface to interact with a Gemini response. Calling any
  10│ // of its meethods after one has already been called or after
  11│ // Handler.ServeGemini has returned is a no-op.
  12│ type ResponseWriter interface {
  13│ 	Failure(ctx context.Context, code StatusCode, msg string)
  14│ 	Input(ctx context.Context, prompt string, isSensitive bool)
  15│ 	Redirect(ctx context.Context, redirectURL string, isPermanant bool)
  16│ 	Success(ctx context.Context, mimeType string) io.Writer
  17│ }
  18│ 
  19│ type response struct {
  20│ 	conn io.Writer
  21│ 	sent atomicBool
  22│ }
  23│ 
  24│ //nolint:gochecknoglobals // allocate some frequently used byte slices.
  25│ var (
  26│ 	responseHeaderSpace          = []byte{' '}
  27│ 	responseHeaderLineTerminator = []byte("\r\n")
  28│ )
  29│ 
  30│ func (r *response) writeHeader(ctx context.Context, code StatusCode, meta []byte) {
  31│ 	if r.sent.isSet() {
  32│ 		return
  33│ 	}
  34│ 	defer r.sent.set()
  35│ 
  36│ 	logWriter := func(string, bool) {}
  37│ 
  38│ 	srv, ok := ServerFromCtx(ctx)
  39│ 	if ok && srv.LogHandler != nil {
  40│ 		logWriter = srv.LogHandler
  41│ 	}
  42│ 
  43│ 	c, err := code.MarshalText()
  44│ 	if err != nil {
  45│ 		logWriter(err.Error(), true)
  46│ 
  47│ 		return
  48│ 	}
  49│ 
  50│ 	if _, err = r.conn.Write(c); err != nil {
  51│ 		logWriter(err.Error(), true)
  52│ 
  53│ 		return
  54│ 	}
  55│ 
  56│ 	if _, err = r.conn.Write(responseHeaderSpace); err != nil {
  57│ 		logWriter(err.Error(), true)
  58│ 
  59│ 		return
  60│ 	}
  61│ 
  62│ 	if _, err = r.conn.Write(meta); err != nil {
  63│ 		logWriter(err.Error(), true)
  64│ 
  65│ 		return
  66│ 	}
  67│ 
  68│ 	if _, err = r.conn.Write(responseHeaderLineTerminator); err != nil {
  69│ 		logWriter(err.Error(), true)
  70│ 
  71│ 		return
  72│ 	}
  73│ }
  74│ 
  75│ func (r *response) Input(ctx context.Context, prompt string, isSensitive bool) {
  76│ 	code := StatusInput
  77│ 	if isSensitive {
  78│ 		code = StatusSensitiveInput
  79│ 	}
  80│ 
  81│ 	r.writeHeader(ctx, code, []byte(prompt))
  82│ }
  83│ 
  84│ func (r *response) Redirect(ctx context.Context, redirectURL string, isPermanant bool) {
  85│ 	code := StatusTemporaryRedirect
  86│ 	if isPermanant {
  87│ 		code = StatusPermanentRedirect
  88│ 	}
  89│ 
  90│ 	r.writeHeader(ctx, code, []byte(redirectURL))
  91│ }
  92│ 
  93│ func (r *response) Success(ctx context.Context, mimeType string) io.Writer {
  94│ 	if mimeType == "" {
  95│ 		mimeType = mime.FormatMediaType("text/gemini", map[string]string{"charset": "utf-8", "lang": "en"})
  96│ 	}
  97│ 
  98│ 	r.writeHeader(ctx, StatusSuccess, []byte(mimeType))
  99│ 
 100│ 	return r.conn
 101│ }
 102│ 
 103│ func (r *response) Failure(ctx context.Context, code StatusCode, msg string) {
 104│ 	category := code.ToCategory()
 105│ 	if category < StatusCategoryTemporaryFailure || category > StatusCategoryPermanentFailure {
 106│ 		code = StatusPermanentFailure
 107│ 	}
 108│ 
 109│ 	r.writeHeader(ctx, code, []byte(msg))
 110│ }
────╯

· · ·

info@source.community