You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

62 lines
1.4KB

  1. // Copyright 2018 The Macaron Authors
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License"): you may
  4. // not use this file except in compliance with the License. You may obtain
  5. // a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  11. // WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  12. // License for the specific language governing permissions and limitations
  13. // under the License.
  14. package session
  15. import (
  16. "net/url"
  17. "gopkg.in/macaron.v1"
  18. )
  19. type Flash struct {
  20. ctx *macaron.Context
  21. url.Values
  22. ErrorMsg, WarningMsg, InfoMsg, SuccessMsg string
  23. }
  24. func (f *Flash) set(name, msg string, current ...bool) {
  25. isShow := false
  26. if (len(current) == 0 && macaron.FlashNow) ||
  27. (len(current) > 0 && current[0]) {
  28. isShow = true
  29. }
  30. if isShow {
  31. f.ctx.Data["Flash"] = f
  32. } else {
  33. f.Set(name, msg)
  34. }
  35. }
  36. func (f *Flash) Error(msg string, current ...bool) {
  37. f.ErrorMsg = msg
  38. f.set("error", msg, current...)
  39. }
  40. func (f *Flash) Warning(msg string, current ...bool) {
  41. f.WarningMsg = msg
  42. f.set("warning", msg, current...)
  43. }
  44. func (f *Flash) Info(msg string, current ...bool) {
  45. f.InfoMsg = msg
  46. f.set("info", msg, current...)
  47. }
  48. func (f *Flash) Success(msg string, current ...bool) {
  49. f.SuccessMsg = msg
  50. f.set("success", msg, current...)
  51. }