misc: small fix or general refactoring i did not bother commenting

session/events.go

@@ -32,8 +32,8 @@ func NewEvent(tag string, data interface{}) Event {
 
 func (e Event) Label() string {
 	m := e.Data.(LogMessage)
-	label := log.LevelNames[m.Level]
-	color := log.LevelColors[m.Level]
+	label := log.LevelName(m.Level)
+	color := log.LevelColor(m.Level)
 	return color + label + tui.RESET
 }
 

session/prompt.go

@@ -15,24 +15,7 @@ const (
 )
 
 var (
-	// these are here because if colors are disabled,
-	// we need the updated tui.* variables
-	effects = map[string]string{
-		"{bold}":  tui.BOLD,
-		"{dim}":   tui.DIM,
-		"{r}":     tui.RED,
-		"{g}":     tui.GREEN,
-		"{b}":     tui.BLUE,
-		"{y}":     tui.YELLOW,
-		"{fb}":    tui.FOREBLACK,
-		"{fw}":    tui.FOREWHITE,
-		"{bdg}":   tui.BACKDARKGRAY,
-		"{br}":    tui.BACKRED,
-		"{bg}":    tui.BACKGREEN,
-		"{by}":    tui.BACKYELLOW,
-		"{blb}":   tui.BACKLIGHTBLUE, // Ziggy this is for you <3
-		"{reset}": tui.RESET,
-	}
+	effects         = map[string]string{}
 	PromptCallbacks = map[string]func(s *Session) string{
 		"{cidr}": func(s *Session) string {
 			return s.Interface.CIDR()
@@ -62,6 +45,24 @@ type Prompt struct {
 }
 
 func NewPrompt() Prompt {
+	// these are here because if colors are disabled,
+	// we need the updated tui.* variables
+	effects = map[string]string{
+		"{bold}":  tui.BOLD,
+		"{dim}":   tui.DIM,
+		"{r}":     tui.RED,
+		"{g}":     tui.GREEN,
+		"{b}":     tui.BLUE,
+		"{y}":     tui.YELLOW,
+		"{fb}":    tui.FOREBLACK,
+		"{fw}":    tui.FOREWHITE,
+		"{bdg}":   tui.BACKDARKGRAY,
+		"{br}":    tui.BACKRED,
+		"{bg}":    tui.BACKGREEN,
+		"{by}":    tui.BACKYELLOW,
+		"{blb}":   tui.BACKLIGHTBLUE, // Ziggy this is for you <3
+		"{reset}": tui.RESET,
+	}
 	return Prompt{}
 }
 

session/session.go

@@ -95,29 +95,29 @@ type Session struct {
 }
 
 func New() (*Session, error) {
-	var err error
+	opts, err := core.ParseOptions()
+	if err != nil {
+		return nil, err
+	}
+
+	if *opts.NoColors || !tui.Effects() {
+		tui.Disable()
+		log.NoEffects = true
+	}
 
 	s := &Session{
-		Prompt: NewPrompt(),
-		Env:    nil,
-		Active: false,
-		Queue:  nil,
+		Prompt:  NewPrompt(),
+		Options: opts,
+		Env:     nil,
+		Active:  false,
+		Queue:   nil,
 
 		CoreHandlers:   make([]CommandHandler, 0),
 		Modules:        make([]Module, 0),
 		Events:         nil,
 		UnkCmdCallback: nil,
 	}
 
-	if s.Options, err = core.ParseOptions(); err != nil {
-		return nil, err
-	}
-
-	if *s.Options.NoColors || !tui.Effects() {
-		tui.Disable()
-		log.NoEffects = true
-	}
-
 	if *s.Options.CpuProfile != "" {
 		if f, err := os.Create(*s.Options.CpuProfile); err != nil {
 			return nil, err