bwilliams
a279405cae
The serve command runs internal/server: the API under /api (health, and the CSRF token endpoint) and the WebSocket hub under /ws, over HTTP or TLS, with an optional HTTP to HTTPS redirect, the PROXY protocol from trusted sources and a graceful shutdown that closes WebSockets first. Its middleware covers the client IP behind trusted proxies, request IDs, request logging, panic recovery, global and API rate limits, Host and body size checks, security response headers, compression, CSRF protection (cross-origin checks and signed double-submit tokens) and JSON-only API bodies. Middleware that logs takes a WithLogger option. New server config sections: proxy, csrf, tls, headers and frontend (not used yet), plus max_header_bytes, max_body_bytes, global and API rate limits and HTTP read, write and idle timeouts. Every setting documents what 0 or empty means, and internal/config/README.md is reorganised by section. server.timeouts.read_header must be greater than 0, and a header or idle timeout of 0 means no limit rather than the read timeout. WebSocket upgrades are protected by the Origin check: the server's own origin, or one of csrf.cross_origin_protection.trusted_origins, which the hub now takes with WithTrustedOrigins. wsserve becomes "ws serve". "ws dial" opens a WebSocket on /ws to test it: --ping sends a ping and waits for the pong, --message sends messages and prints the replies, --origin tests the Origin check, and with neither --ping nor --message each line of stdin is sent until Ctrl+C.
93 lines
2.5 KiB
Go
93 lines
2.5 KiB
Go
/*
|
|
Copyright © 2026 Craig Williams <craig@avsdev.uk>
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Lesser 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 Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
// Command job-search. Run `job-search --help` for usage.
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log/slog"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"git.avsdev.uk/Go/job-search/internal/cli"
|
|
"git.avsdev.uk/Go/job-search/internal/cli/config"
|
|
"git.avsdev.uk/Go/job-search/internal/cli/fetch"
|
|
"git.avsdev.uk/Go/job-search/internal/cli/migrate"
|
|
"git.avsdev.uk/Go/job-search/internal/cli/serve"
|
|
"git.avsdev.uk/Go/job-search/internal/cli/version"
|
|
"git.avsdev.uk/Go/job-search/internal/cli/ws"
|
|
)
|
|
|
|
// newRootCmd builds the command tree. It returns a fresh tree on each call so
|
|
// tests can run commands in isolation.
|
|
func newRootCmd() *cobra.Command {
|
|
opts := &cli.Options{}
|
|
|
|
root := &cobra.Command{
|
|
Use: "job-search",
|
|
Short: "JobSearch - AVSDev Future Opportunities!",
|
|
Long: `JobSearch is a scraping tool for searching for possible contract
|
|
opportunities.
|
|
`,
|
|
SilenceUsage: true, // don't print usage on runtime errors
|
|
SilenceErrors: true, // main prints the error once
|
|
CompletionOptions: cobra.CompletionOptions{
|
|
HiddenDefaultCmd: true,
|
|
},
|
|
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
|
|
cli.SetupInitialLogging(cmd)
|
|
|
|
slog.Debug("command persistent pre-run", "command", "root")
|
|
slog.Debug(
|
|
"command",
|
|
"path", strings.Join(strings.Split(cmd.CommandPath(), " ")[1:], " "),
|
|
)
|
|
|
|
return nil
|
|
},
|
|
PersistentPostRunE: func(cmd *cobra.Command, args []string) error {
|
|
slog.Debug("command persistent post-run", "command", "root")
|
|
|
|
return opts.Finish()
|
|
},
|
|
}
|
|
|
|
cobra.EnableTraverseRunHooks = true
|
|
|
|
opts.Bind(root)
|
|
|
|
root.AddCommand(
|
|
version.NewCommand(),
|
|
config.NewCommand(opts),
|
|
fetch.NewCommand(opts),
|
|
migrate.NewCommand(opts),
|
|
ws.NewCommand(opts),
|
|
serve.NewCommand(opts),
|
|
)
|
|
|
|
return root
|
|
}
|
|
|
|
func main() {
|
|
if err := newRootCmd().Execute(); err != nil {
|
|
fmt.Fprintln(os.Stderr, "job-search:", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|