From 4954703e33444f07efe5c2d25d742210991128a6 Mon Sep 17 00:00:00 2001
From: Andrew Burdyug <buran83@gmail.com>
Date: Wed, 3 Apr 2024 22:14:37 +0300
Subject: [PATCH] Fix status bar indicator

---
 main.go      |  2 ++
 statusbar.go | 10 ++++++++++
 2 files changed, 12 insertions(+)

diff --git a/main.go b/main.go
index f9c8ac0..b1ee50d 100644
--- a/main.go
+++ b/main.go
@@ -779,6 +779,8 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		m.resBodyLines = formatRespBody(
 			m.res.Header.Get("content-type"), string(buf),
 			m.checkboxes[checkboxIndex(autoformat)].IsOn())
+		sbar.SetResStatusCode(m.res.StatusCode)
+		sbar.SetResProto(m.res.Proto)
 		sbar.Info("request is executed, response taken")
 		if len(redirects) > 0 {
 			sbar.Warning(
diff --git a/statusbar.go b/statusbar.go
index a220f6f..6b6cc34 100644
--- a/statusbar.go
+++ b/statusbar.go
@@ -184,6 +184,16 @@ func (s *StatusBar) SetResTime(t time.Duration) {
 	s.resTime = t
 }
 
+// Set last response status code.
+func (s *StatusBar) SetResStatusCode(c int) {
+	s.resStatusCode = c
+}
+
+// Set last response proto.
+func (s *StatusBar) SetResProto(p string) {
+	s.resProto = p
+}
+
 // Set count of requests.
 func (s *StatusBar) SetReqCount(c int) {
 	s.reqCount = c