Refactored main.go into cmd/jobsearch/main.go (in case separate commands are introduced later) which has required the options mechanism to change slightly as well as the logging.
This commit is contained in:
1 parent
de3ebf68c5
commit
6436253629
14 files changed
+273
-224
No files matched your search
@@ -0,0 +1,85 @@
|
|||||||
|
/*
|
||||||
|
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"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"log/slog"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"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/version"
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
)
|
||||||
|
|
||||||
|
// 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.SetupIntialLogging(cmd)
|
||||||
|
|
||||||
|
slog.Debug("root.PersistentPreRun")
|
||||||
|
slog.Debug(
|
||||||
|
"command: \"" +
|
||||||
|
strings.Join(strings.Split(cmd.CommandPath(), " ")[1:], " ") +
|
||||||
|
"\"",
|
||||||
|
)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
PersistentPostRunE: func(cmd *cobra.Command, args []string) error {
|
||||||
|
slog.Debug("root.PersistentPostRun")
|
||||||
|
|
||||||
|
return opts.Finish()
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
cobra.EnableTraverseRunHooks = true
|
||||||
|
|
||||||
|
opts.Bind(root)
|
||||||
|
|
||||||
|
root.AddCommand(
|
||||||
|
version.NewCommand(),
|
||||||
|
config.NewCommand(opts),
|
||||||
|
)
|
||||||
|
|
||||||
|
return root
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
if err := newRootCmd().Execute(); err != nil {
|
||||||
|
fmt.Fprintln(os.Stderr, "job-search:", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
-118
@@ -1,118 +0,0 @@
|
|||||||
/*
|
|
||||||
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/>.
|
|
||||||
*/
|
|
||||||
package cmd
|
|
||||||
|
|
||||||
import (
|
|
||||||
"log/slog"
|
|
||||||
"os"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"git.avsdev.uk/Go/job-search/internal/cli"
|
|
||||||
cmdConfig "git.avsdev.uk/Go/job-search/internal/cli/config"
|
|
||||||
cmdVersion "git.avsdev.uk/Go/job-search/internal/cli/version"
|
|
||||||
"git.avsdev.uk/Go/job-search/internal/config"
|
|
||||||
"git.avsdev.uk/Go/job-search/internal/log"
|
|
||||||
"github.com/spf13/cobra"
|
|
||||||
)
|
|
||||||
|
|
||||||
var rootCmd = &cobra.Command{
|
|
||||||
Use: "job-search",
|
|
||||||
Short: "JobSearch - AVSDev Future Opportunities!",
|
|
||||||
Long: `JobSearch is a scraping tool for searching for possible contract
|
|
||||||
opportunities.
|
|
||||||
`,
|
|
||||||
CompletionOptions: cobra.CompletionOptions{
|
|
||||||
HiddenDefaultCmd: true,
|
|
||||||
},
|
|
||||||
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
|
|
||||||
// Check if the command is expecting to use terminal for messages
|
|
||||||
if val, ok := cmd.Annotations["terminal_log"]; ok && val == "1" {
|
|
||||||
log.LogToTerminal()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check for debug level via args
|
|
||||||
dbgCount, _ := cmd.Flags().GetCount("debug")
|
|
||||||
conf := config.JobSearchConfig{
|
|
||||||
Logging: config.JobSearchConfigLogging{},
|
|
||||||
}
|
|
||||||
if dbgCount > 1 {
|
|
||||||
conf.Logging.Level = "debug"
|
|
||||||
} else if dbgCount > 0 {
|
|
||||||
conf.Logging.Level = "info"
|
|
||||||
}
|
|
||||||
if conf.Logging.LogLevel() <= slog.LevelDebug {
|
|
||||||
log.EnableDebug()
|
|
||||||
} else if conf.Logging.LogLevel() <= slog.LevelInfo {
|
|
||||||
log.EnableInfo()
|
|
||||||
}
|
|
||||||
|
|
||||||
// With logging set up, these will be visible if trace is active
|
|
||||||
slog.Debug("root.PersistentPreRun")
|
|
||||||
slog.Debug(
|
|
||||||
"command: \"" +
|
|
||||||
strings.Join(strings.Split(cmd.CommandPath(), " ")[1:], " ") +
|
|
||||||
"\"",
|
|
||||||
)
|
|
||||||
|
|
||||||
// Always load config unless command indicates not to
|
|
||||||
if val, ok := cmd.Annotations["no_config"]; !ok || val == "0" {
|
|
||||||
// Load config - from specified file if provided
|
|
||||||
cfgFile, _ := cmd.Flags().GetString("config")
|
|
||||||
if err := config.Init(cfgFile, &conf); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Attach config to command context
|
|
||||||
ctx := cli.WithConfig(cmd.Context(), conf)
|
|
||||||
cmd.SetContext(ctx)
|
|
||||||
|
|
||||||
// Change logging level based off newly loaded config
|
|
||||||
if conf.Logging.LogLevel() <= slog.LevelDebug {
|
|
||||||
log.EnableDebug()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
},
|
|
||||||
PersistentPostRun: func(cmd *cobra.Command, args []string) {
|
|
||||||
slog.Debug("root.PersistentPostRun")
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
// Execute, called by main.main() handles all child commands and appropriate
|
|
||||||
// flags.
|
|
||||||
func Execute() {
|
|
||||||
err := rootCmd.Execute()
|
|
||||||
if err != nil {
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
cobra.EnableTraverseRunHooks = true
|
|
||||||
|
|
||||||
log.Init()
|
|
||||||
|
|
||||||
pFlags := rootCmd.PersistentFlags()
|
|
||||||
pFlags.StringP(
|
|
||||||
"config", "c", "",
|
|
||||||
"config file (default: "+config.EtcConfigPath+"/config.yaml)")
|
|
||||||
pFlags.CountP("debug", "d", "enable debug output (repeat for trace)")
|
|
||||||
|
|
||||||
rootCmd.AddCommand(cmdVersion.NewCommand())
|
|
||||||
rootCmd.AddCommand(cmdConfig.NewCommand())
|
|
||||||
}
|
|
||||||
@@ -19,10 +19,11 @@ package config
|
|||||||
import (
|
import (
|
||||||
"log/slog"
|
"log/slog"
|
||||||
|
|
||||||
|
"git.avsdev.uk/Go/job-search/internal/cli"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
func NewCommand() *cobra.Command {
|
func NewCommand(opts *cli.Options) *cobra.Command {
|
||||||
cmd := &cobra.Command{
|
cmd := &cobra.Command{
|
||||||
Use: "config",
|
Use: "config",
|
||||||
Short: "Configuration management",
|
Short: "Configuration management",
|
||||||
@@ -33,9 +34,9 @@ func NewCommand() *cobra.Command {
|
|||||||
slog.Debug("config.PersistentPostRun")
|
slog.Debug("config.PersistentPostRun")
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
cmd.AddCommand(commandShow())
|
cmd.AddCommand(commandShow(opts))
|
||||||
cmd.AddCommand(commandGet())
|
cmd.AddCommand(commandGet(opts))
|
||||||
cmd.AddCommand(commandSet())
|
cmd.AddCommand(commandSet(opts))
|
||||||
cmd.AddCommand(commandInit())
|
cmd.AddCommand(commandInit())
|
||||||
return cmd
|
return cmd
|
||||||
}
|
}
|
||||||
@@ -17,7 +17,6 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|||||||
package config
|
package config
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
|
|
||||||
@@ -25,7 +24,7 @@ import (
|
|||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
func commandGet() *cobra.Command {
|
func commandGet(opts *cli.Options) *cobra.Command {
|
||||||
cmd := &cobra.Command{
|
cmd := &cobra.Command{
|
||||||
Use: "get [config_key]",
|
Use: "get [config_key]",
|
||||||
Args: cobra.ExactArgs(1),
|
Args: cobra.ExactArgs(1),
|
||||||
@@ -43,9 +42,9 @@ func commandGet() *cobra.Command {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
conf, ok := cli.ConfigFromContext(cmd.Context())
|
conf, err := opts.RunSetup(cmd, true, true)
|
||||||
if !ok {
|
if err != nil {
|
||||||
return errors.New("no configuration loaded")
|
return fmt.Errorf("no configuration loaded (%w)", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
v, err := conf.ValueOf(args[0])
|
v, err := conf.ValueOf(args[0])
|
||||||
@@ -57,9 +56,6 @@ func commandGet() *cobra.Command {
|
|||||||
|
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
Annotations: map[string]string{
|
|
||||||
"terminal_log": "1",
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
return cmd
|
return cmd
|
||||||
}
|
}
|
||||||
@@ -87,10 +87,6 @@ func commandInit() *cobra.Command {
|
|||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
Annotations: map[string]string{
|
|
||||||
"no_config": "1",
|
|
||||||
"terminal_log": "1",
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
return cmd
|
return cmd
|
||||||
}
|
}
|
||||||
@@ -29,7 +29,7 @@ import (
|
|||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
)
|
)
|
||||||
|
|
||||||
func setConfig(conf config.JobSearchConfig, key string, value string) error {
|
func setConfig(conf *config.JobSearchConfig, key string, value string) error {
|
||||||
var confValue any = nil
|
var confValue any = nil
|
||||||
var confType reflect.Kind
|
var confType reflect.Kind
|
||||||
var err error
|
var err error
|
||||||
@@ -65,7 +65,7 @@ func setConfig(conf config.JobSearchConfig, key string, value string) error {
|
|||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
func unsetConfig(conf config.JobSearchConfig, key string) error {
|
func unsetConfig(conf *config.JobSearchConfig, key string) error {
|
||||||
if _, err := conf.TypeOf(key); err != nil {
|
if _, err := conf.TypeOf(key); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -88,7 +88,7 @@ func unsetConfig(conf config.JobSearchConfig, key string) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func commandSet() *cobra.Command {
|
func commandSet(opts *cli.Options) *cobra.Command {
|
||||||
cmd := &cobra.Command{
|
cmd := &cobra.Command{
|
||||||
Use: "set [config_key] [config_value]",
|
Use: "set [config_key] [config_value]",
|
||||||
Args: cobra.MatchAll(cobra.MinimumNArgs(1), cobra.MaximumNArgs(2)),
|
Args: cobra.MatchAll(cobra.MinimumNArgs(1), cobra.MaximumNArgs(2)),
|
||||||
@@ -117,9 +117,9 @@ func commandSet() *cobra.Command {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
conf, ok := cli.ConfigFromContext(cmd.Context())
|
conf, err := opts.RunSetup(cmd, true, true)
|
||||||
if !ok {
|
if err != nil {
|
||||||
return errors.New("no configuration loaded")
|
return fmt.Errorf("no configuration loaded (%w)", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if viper.ConfigFileUsed() == "" {
|
if viper.ConfigFileUsed() == "" {
|
||||||
@@ -132,9 +132,6 @@ func commandSet() *cobra.Command {
|
|||||||
return setConfig(conf, args[0], args[1])
|
return setConfig(conf, args[0], args[1])
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
Annotations: map[string]string{
|
|
||||||
"terminal_log": "1",
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
cmd.Flags().BoolP("unset", "x", false, "clear configuration setting")
|
cmd.Flags().BoolP("unset", "x", false, "clear configuration setting")
|
||||||
return cmd
|
return cmd
|
||||||
|
|||||||
@@ -17,14 +17,14 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|||||||
package config
|
package config
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"fmt"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
|
|
||||||
"git.avsdev.uk/Go/job-search/internal/cli"
|
"git.avsdev.uk/Go/job-search/internal/cli"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
func commandShow() *cobra.Command {
|
func commandShow(opts *cli.Options) *cobra.Command {
|
||||||
cmd := &cobra.Command{
|
cmd := &cobra.Command{
|
||||||
Use: "show",
|
Use: "show",
|
||||||
Short: "Print config information",
|
Short: "Print config information",
|
||||||
@@ -37,9 +37,9 @@ func commandShow() *cobra.Command {
|
|||||||
RunE: func(cmd *cobra.Command, args []string) error {
|
RunE: func(cmd *cobra.Command, args []string) error {
|
||||||
slog.Debug("config.show.Run")
|
slog.Debug("config.show.Run")
|
||||||
|
|
||||||
conf, ok := cli.ConfigFromContext(cmd.Context())
|
conf, err := opts.RunSetup(cmd, true, true)
|
||||||
if !ok {
|
if err != nil {
|
||||||
return errors.New("no configuration loaded")
|
return fmt.Errorf("no configuration loaded (%w)", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if yaml, _ := cmd.Flags().GetBool("yaml"); yaml {
|
if yaml, _ := cmd.Flags().GetBool("yaml"); yaml {
|
||||||
@@ -52,9 +52,6 @@ func commandShow() *cobra.Command {
|
|||||||
|
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
Annotations: map[string]string{
|
|
||||||
"terminal_log": "1",
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
cmd.Flags().Bool("json", false, "format configuration as json")
|
cmd.Flags().Bool("json", false, "format configuration as json")
|
||||||
cmd.Flags().Bool("yaml", false, "format configuration as yaml")
|
cmd.Flags().Bool("yaml", false, "format configuration as yaml")
|
||||||
|
|||||||
@@ -14,12 +14,28 @@ GNU Lesser General Public License for more details.
|
|||||||
You should have received a copy of the GNU Lesser General Public License
|
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/>.
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
*/
|
*/
|
||||||
package main
|
package cli
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"git.avsdev.uk/Go/job-search/cmd"
|
"log/slog"
|
||||||
|
|
||||||
|
"git.avsdev.uk/Go/job-search/internal/log"
|
||||||
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
func main() {
|
func SetupIntialLogging(cmd *cobra.Command) {
|
||||||
cmd.Execute()
|
// Start logging to terminal first - any commands that want to use file
|
||||||
|
// logging should activate it themselves
|
||||||
|
log.LogToTerminal()
|
||||||
|
|
||||||
|
// Check for quiet flag
|
||||||
|
if quiet, _ := cmd.Flags().GetBool("quiet"); quiet {
|
||||||
|
log.SetLevel(slog.LevelError)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check for debug logging enabled (overrides quiet flag)
|
||||||
|
debugEnabled, _ := cmd.Flags().GetBool("debug")
|
||||||
|
if debugEnabled {
|
||||||
|
log.EnableDebug()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -0,0 +1,89 @@
|
|||||||
|
/*
|
||||||
|
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/>.
|
||||||
|
*/
|
||||||
|
package cli
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log/slog"
|
||||||
|
|
||||||
|
"git.avsdev.uk/Go/job-search/internal/config"
|
||||||
|
"git.avsdev.uk/Go/job-search/internal/log"
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Options struct {
|
||||||
|
configFile string
|
||||||
|
debug bool
|
||||||
|
quiet bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// Binds common options through the root command
|
||||||
|
func (o *Options) Bind(cmd *cobra.Command) {
|
||||||
|
pf := cmd.PersistentFlags()
|
||||||
|
pf.StringVarP(&o.configFile, "config", "c", "", "configuration file")
|
||||||
|
pf.BoolVarP(&o.debug, "debug", "d", false, "enable debug messages")
|
||||||
|
pf.BoolVarP(
|
||||||
|
&o.quiet, "quiet", "q", false, "don't report info or warning messages",
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Setup loads the config and ensure logging is switch to file if needed
|
||||||
|
func (o *Options) RunSetup(
|
||||||
|
cmd *cobra.Command, loadConfig bool, terminalLog bool,
|
||||||
|
) (*config.JobSearchConfig, error) {
|
||||||
|
// Check for debug level or quiet supression via args
|
||||||
|
conf := &config.JobSearchConfig{
|
||||||
|
Logging: config.JobSearchConfigLogging{},
|
||||||
|
}
|
||||||
|
if o.debug {
|
||||||
|
conf.Logging.Level = "debug"
|
||||||
|
} else if o.quiet {
|
||||||
|
conf.Logging.Level = "error"
|
||||||
|
}
|
||||||
|
|
||||||
|
// Only load config if command requests it (nearly always true)
|
||||||
|
if loadConfig {
|
||||||
|
// Load config - from specified file if provided
|
||||||
|
if err := config.Init(o.configFile, conf); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Change logging level based off newly loaded config
|
||||||
|
if conf.Logging.LogLevel() <= slog.LevelDebug && !log.IsDebugEnabled() {
|
||||||
|
log.EnableDebug()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if the logging should be switched to file
|
||||||
|
if !terminalLog {
|
||||||
|
if len(conf.Logging.File) == 0 {
|
||||||
|
slog.Warn("No log file configured, remaining in console logging mode!")
|
||||||
|
} else {
|
||||||
|
log.LogToFile(conf.Logging.File)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return conf, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Finish shuts down any open file logger
|
||||||
|
func (o *Options) Finish() error {
|
||||||
|
if log.IsFileLogger() {
|
||||||
|
log.Shutdown()
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@@ -37,7 +37,7 @@ func NewCommand() *cobra.Command {
|
|||||||
Run: func(cmd *cobra.Command, args []string) {
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
slog.Debug("version.Run")
|
slog.Debug("version.Run")
|
||||||
|
|
||||||
fmt.Printf("PBCNinja - %s\n", version.Version)
|
fmt.Printf("JobSearch - %s\n", version.Version)
|
||||||
if version.BuildTime != "unknown" {
|
if version.BuildTime != "unknown" {
|
||||||
fmt.Printf("Built: %s\n", version.BuildTime)
|
fmt.Printf("Built: %s\n", version.BuildTime)
|
||||||
}
|
}
|
||||||
@@ -45,10 +45,6 @@ func NewCommand() *cobra.Command {
|
|||||||
fmt.Printf("Commit: %s\n", version.GitCommit)
|
fmt.Printf("Commit: %s\n", version.GitCommit)
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
Annotations: map[string]string{
|
|
||||||
"no_config": "1",
|
|
||||||
"terminal_log": "1",
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
return cmd
|
return cmd
|
||||||
}
|
}
|
||||||
@@ -1,43 +0,0 @@
|
|||||||
/*
|
|
||||||
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/>.
|
|
||||||
*/
|
|
||||||
package cli
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"git.avsdev.uk/Go/job-search/internal/config"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Define a private key type to avoid collisions
|
|
||||||
type configContextKey struct {
|
|
||||||
name string
|
|
||||||
}
|
|
||||||
|
|
||||||
// Define the key
|
|
||||||
var configKey = &configContextKey{"config"}
|
|
||||||
|
|
||||||
// Public accessor to store a value to ctx
|
|
||||||
func WithConfig(ctx context.Context, conf config.JobSearchConfig) context.Context {
|
|
||||||
// No allocation here since userIDKey is a pointer to a struct
|
|
||||||
return context.WithValue(ctx, configKey, conf)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Public accessor to fetch a value from ctx
|
|
||||||
func ConfigFromContext(ctx context.Context) (config.JobSearchConfig, bool) {
|
|
||||||
v, ok := ctx.Value(configKey).(config.JobSearchConfig)
|
|
||||||
return v, ok
|
|
||||||
}
|
|
||||||
@@ -24,6 +24,7 @@ import (
|
|||||||
|
|
||||||
type JobSearchConfigLogging struct {
|
type JobSearchConfigLogging struct {
|
||||||
Level string
|
Level string
|
||||||
|
File string
|
||||||
}
|
}
|
||||||
|
|
||||||
// Default configuration
|
// Default configuration
|
||||||
@@ -37,6 +38,7 @@ func defaultConfigLogging() JobSearchConfigLogging {
|
|||||||
func (conf JobSearchConfigLogging) Map() map[string]any {
|
func (conf JobSearchConfigLogging) Map() map[string]any {
|
||||||
return map[string]any{
|
return map[string]any{
|
||||||
"level": conf.Level,
|
"level": conf.Level,
|
||||||
|
"file": conf.File,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -55,6 +57,8 @@ func (c JobSearchConfigLogging) TypeOf(key string) (reflect.Kind, error) {
|
|||||||
switch key {
|
switch key {
|
||||||
case "level":
|
case "level":
|
||||||
return reflect.String, nil
|
return reflect.String, nil
|
||||||
|
case "file":
|
||||||
|
return reflect.String, nil
|
||||||
default:
|
default:
|
||||||
return reflect.Invalid, errors.New("unknown configuration key")
|
return reflect.Invalid, errors.New("unknown configuration key")
|
||||||
}
|
}
|
||||||
@@ -65,6 +69,8 @@ func (c JobSearchConfigLogging) ValueOf(key string) (any, error) {
|
|||||||
switch key {
|
switch key {
|
||||||
case "level":
|
case "level":
|
||||||
return c.Level, nil
|
return c.Level, nil
|
||||||
|
case "file":
|
||||||
|
return c.File, nil
|
||||||
default:
|
default:
|
||||||
return "", errors.New("unknown configuration key")
|
return "", errors.New("unknown configuration key")
|
||||||
}
|
}
|
||||||
@@ -82,12 +88,15 @@ func (c *JobSearchConfigLogging) SetValue(key string, value any) error {
|
|||||||
return errors.New(
|
return errors.New(
|
||||||
"invalid log level. Must be one of: debug, info, warn, error")
|
"invalid log level. Must be one of: debug, info, warn, error")
|
||||||
}
|
}
|
||||||
|
case "file":
|
||||||
|
c.File = value.(string)
|
||||||
default:
|
default:
|
||||||
return errors.New("unknown configuration key")
|
return errors.New("unknown configuration key")
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Utility methods for converting level strings to slog levels
|
||||||
func ConfigLevelToLogLevel(level string) slog.Level {
|
func ConfigLevelToLogLevel(level string) slog.Level {
|
||||||
switch level {
|
switch level {
|
||||||
case "debug":
|
case "debug":
|
||||||
@@ -102,7 +111,6 @@ func ConfigLevelToLogLevel(level string) slog.Level {
|
|||||||
return 99
|
return 99
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *JobSearchConfigLogging) LogLevel() slog.Level {
|
func (c *JobSearchConfigLogging) LogLevel() slog.Level {
|
||||||
return ConfigLevelToLogLevel(c.Level)
|
return ConfigLevelToLogLevel(c.Level)
|
||||||
}
|
}
|
||||||
@@ -81,10 +81,14 @@ func Init(cfgFile string, conf *JobSearchConfig) error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
oldLogLevel := conf.Logging.Level
|
// Load configuration from Viper, ensure -debug and -quiet are retained
|
||||||
|
oldLogLevel := conf.Logging.LogLevel()
|
||||||
LoadFromViper(conf)
|
LoadFromViper(conf)
|
||||||
if ConfigLevelToLogLevel(oldLogLevel) < conf.Logging.LogLevel() {
|
if oldLogLevel == slog.LevelDebug {
|
||||||
conf.Logging.Level = oldLogLevel
|
conf.Logging.Level = "debug"
|
||||||
|
}
|
||||||
|
if oldLogLevel == slog.LevelError {
|
||||||
|
conf.Logging.Level = "error"
|
||||||
}
|
}
|
||||||
|
|
||||||
if conf.Logging.LogLevel() <= slog.LevelDebug {
|
if conf.Logging.LogLevel() <= slog.LevelDebug {
|
||||||
|
|||||||
+42
-17
@@ -26,24 +26,11 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
var logLevel slog.LevelVar
|
var logLevel slog.LevelVar
|
||||||
|
var fileLogger *os.File
|
||||||
|
|
||||||
// Set up the logger using defaults
|
// Set the logging to a particular level, quietly
|
||||||
func Init() {
|
func SetLevel(level slog.Level) {
|
||||||
logLevel.Set(slog.LevelWarn)
|
logLevel.Set(level)
|
||||||
opts := &slog.HandlerOptions{
|
|
||||||
Level: &logLevel,
|
|
||||||
}
|
|
||||||
handler := slog.NewTextHandler(os.Stdout, opts)
|
|
||||||
logger := slog.New(handler)
|
|
||||||
slog.SetDefault(logger)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Enable info level logging
|
|
||||||
func EnableInfo() {
|
|
||||||
if logLevel.Level() > slog.LevelInfo {
|
|
||||||
logLevel.Set(slog.LevelInfo)
|
|
||||||
slog.Info("Info logging enabled")
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Enable debug level logging
|
// Enable debug level logging
|
||||||
@@ -54,6 +41,36 @@ func EnableDebug() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func LogToFile(logfile string) error {
|
||||||
|
if fileLogger != nil {
|
||||||
|
(*fileLogger).Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Open the log file
|
||||||
|
fileLogger, err := os.OpenFile(logfile, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
opts := &slog.HandlerOptions{
|
||||||
|
Level: &logLevel,
|
||||||
|
}
|
||||||
|
handler := slog.NewJSONHandler(fileLogger, opts)
|
||||||
|
logger := slog.New(handler)
|
||||||
|
slog.SetDefault(logger)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func Shutdown() error {
|
||||||
|
var err error = nil
|
||||||
|
if fileLogger != nil {
|
||||||
|
err = (*fileLogger).Close()
|
||||||
|
fileLogger = nil
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
func LogToTerminal() {
|
func LogToTerminal() {
|
||||||
opts := &tint.Options{
|
opts := &tint.Options{
|
||||||
Level: &logLevel,
|
Level: &logLevel,
|
||||||
@@ -94,3 +111,11 @@ func IsJSONHandler(h slog.Handler) bool {
|
|||||||
// Read the boolean value using unsafe
|
// Read the boolean value using unsafe
|
||||||
return *(*bool)(unsafe.Pointer(jsonField.UnsafeAddr()))
|
return *(*bool)(unsafe.Pointer(jsonField.UnsafeAddr()))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func IsFileLogger() bool {
|
||||||
|
return fileLogger != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func IsDebugEnabled() bool {
|
||||||
|
return logLevel.Level() <= slog.LevelDebug
|
||||||
|
}
|
||||||
Reference in new issue
Block a user