This is the second tutorial for Bubble Tea covering commands, which deal with I/O. The tutorial assumes you have a working knowledge of Go and a decent understanding of the first tutorial.
You can find the non-annotated version of this program on GitHub.
For this tutorial we're building a very simple program that makes an HTTP request to a server and reports the status code of the response.
We'll import a few necessary packages and put the URL we're going to check in
a const.
package main
import (
"fmt"
"net/http"
"os"
"time"
tea "github.com/charmbracelet/bubbletea"
)
const url = "https://charm.sh/"Next we'll define our model. The only things we need to store are the status code of the HTTP response and a possible error.
type model struct {
status int
err error
}Cmds are functions that perform some I/O and then return a Msg. Checking the
time, ticking a timer, reading from the disk, and network stuff are all I/O and
should be run through commands. That might sound harsh, but it will keep your
Bubble Tea program straightforward and simple.
Anyway, let's write a Cmd that makes a request to a server and returns the
result as a Msg.
func checkServer() tea.Msg {
// Create an HTTP client and make a GET request.
c := &http.Client{Timeout: 10 * time.Second}
res, err := c.Get(url)
if err != nil {
// There was an error making our request. Wrap the error we received
// in a message and return it.
return errMsg{err}
}
// We received a response from the server. Return the HTTP status code
// as a message.
return statusMsg(res.StatusCode)
}
type statusMsg int
type errMsg struct{ err error }
// For messages that contain errors it's often handy to also implement the
// error interface on the message.
func (e errMsg) Error() string { return e.err.Error() }And notice that we've defined two new Msg types. They can be any type, even
an empty struct. We'll come back to them later in our update function.
First, let's write our initialization function.
The initialization method is very simple: we return the Cmd we made earlier.
Note that we don't call the function; the Bubble Tea runtime will do that when
the time is right.
func (m model) Init() (tea.Cmd) {
return checkServer
}Internally, Cmds run asynchronously in a goroutine. The Msg they return is
collected and sent to our update function for handling. Remember those message
types we made earlier when we were making the checkServer command? We handle
them here. This makes dealing with many asynchronous operations very easy.
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
switch msg := msg.(type) {
case statusMsg:
// The server returned a status message. Save it to our model. Also
// tell the Bubble Tea runtime we want to exit because we have nothing
// else to do. We'll still be able to render a final view with our
// status message.
m.status = int(msg)
return m, tea.Quit
case errMsg:
// There was an error. Note it in the model. And tell the runtime
// we're done and want to quit.
m.err = msg
return m, tea.Quit
case tea.KeyMsg:
// Ctrl+c exits. Even with short running programs it's good to have
// a quit key, just in case your logic is off. Users will be very
// annoyed if they can't exit.
if msg.Type == tea.KeyCtrlC {
return m, tea.Quit
}
}
// If we happen to get any other messages, don't do anything.
return m, nil
}Our view is very straightforward. We look at the current model and build a string accordingly:
func (m model) View() string {
// If there's an error, print it out and don't do anything else.
if m.err != nil {
return fmt.Sprintf("\nWe had some trouble: %v\n\n", m.err)
}
// Tell the user we're doing something.
s := fmt.Sprintf("Checking %s ... ", url)
// When the server responds with a status, add it to the current line.
if m.status > 0 {
s += fmt.Sprintf("%d %s!", m.status, http.StatusText(m.status))
}
// Send off whatever we came up with above for rendering.
return "\n" + s + "\n\n"
}The only thing left to do is run the program, so let's do that! Our initial
model doesn't need any data at all in this case, we just initialize it with
a model struct with default values.
func main() {
if _, err := tea.NewProgram(model{}).Run(); err != nil {
fmt.Printf("Uh oh, there was an error: %v\n", err)
os.Exit(1)
}
}And that's that. There's one more thing that is helpful to know about
Cmds, though.
Cmds are defined in Bubble Tea as type Cmd func() Msg. So they're just
functions that don't take any arguments and return a Msg, which can be
any type. If you need to pass arguments to a command, you just make a function
that returns a command. For example:
func cmdWithArg(id int) tea.Cmd {
return func() tea.Msg {
return someMsg{id: id}
}
}A more real-world example looks like:
func checkSomeUrl(url string) tea.Cmd {
return func() tea.Msg {
c := &http.Client{Timeout: 10 * time.Second}
res, err := c.Get(url)
if err != nil {
return errMsg{err}
}
return statusMsg(res.StatusCode)
}
}Anyway, just make sure you do as much stuff as you can in the innermost function, because that's the one that runs asynchronously.
After doing this tutorial and the previous one you should be ready to build a Bubble Tea program of your own. We also recommend that you look at the Bubble Tea example programs as well as Bubbles, a component library for Bubble Tea.
And, of course, check out the Go Docs.
We'd love to hear your thoughts on this tutorial. Feel free to drop us a note!
Part of Charm.
Charm热爱开源 • Charm loves open source