files.go 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287
  1. package mxj
  2. import (
  3. "fmt"
  4. "io"
  5. "os"
  6. )
  7. type Maps []Map
  8. func NewMaps() Maps {
  9. return make(Maps, 0)
  10. }
  11. type MapRaw struct {
  12. M Map
  13. R []byte
  14. }
  15. // NewMapsFromXmlFile - creates an array from a file of JSON values.
  16. func NewMapsFromJsonFile(name string) (Maps, error) {
  17. fi, err := os.Stat(name)
  18. if err != nil {
  19. return nil, err
  20. }
  21. if !fi.Mode().IsRegular() {
  22. return nil, fmt.Errorf("file %s is not a regular file", name)
  23. }
  24. fh, err := os.Open(name)
  25. if err != nil {
  26. return nil, err
  27. }
  28. defer fh.Close()
  29. am := make([]Map, 0)
  30. for {
  31. m, raw, err := NewMapJsonReaderRaw(fh)
  32. if err != nil && err != io.EOF {
  33. return am, fmt.Errorf("error: %s - reading: %s", err.Error(), string(raw))
  34. }
  35. if len(m) > 0 {
  36. am = append(am, m)
  37. }
  38. if err == io.EOF {
  39. break
  40. }
  41. }
  42. return am, nil
  43. }
  44. // ReadMapsFromJsonFileRaw - creates an array of MapRaw from a file of JSON values.
  45. func NewMapsFromJsonFileRaw(name string) ([]MapRaw, error) {
  46. fi, err := os.Stat(name)
  47. if err != nil {
  48. return nil, err
  49. }
  50. if !fi.Mode().IsRegular() {
  51. return nil, fmt.Errorf("file %s is not a regular file", name)
  52. }
  53. fh, err := os.Open(name)
  54. if err != nil {
  55. return nil, err
  56. }
  57. defer fh.Close()
  58. am := make([]MapRaw, 0)
  59. for {
  60. mr := new(MapRaw)
  61. mr.M, mr.R, err = NewMapJsonReaderRaw(fh)
  62. if err != nil && err != io.EOF {
  63. return am, fmt.Errorf("error: %s - reading: %s", err.Error(), string(mr.R))
  64. }
  65. if len(mr.M) > 0 {
  66. am = append(am, *mr)
  67. }
  68. if err == io.EOF {
  69. break
  70. }
  71. }
  72. return am, nil
  73. }
  74. // NewMapsFromXmlFile - creates an array from a file of XML values.
  75. func NewMapsFromXmlFile(name string) (Maps, error) {
  76. fi, err := os.Stat(name)
  77. if err != nil {
  78. return nil, err
  79. }
  80. if !fi.Mode().IsRegular() {
  81. return nil, fmt.Errorf("file %s is not a regular file", name)
  82. }
  83. fh, err := os.Open(name)
  84. if err != nil {
  85. return nil, err
  86. }
  87. defer fh.Close()
  88. am := make([]Map, 0)
  89. for {
  90. m, raw, err := NewMapXmlReaderRaw(fh)
  91. if err != nil && err != io.EOF {
  92. return am, fmt.Errorf("error: %s - reading: %s", err.Error(), string(raw))
  93. }
  94. if len(m) > 0 {
  95. am = append(am, m)
  96. }
  97. if err == io.EOF {
  98. break
  99. }
  100. }
  101. return am, nil
  102. }
  103. // NewMapsFromXmlFileRaw - creates an array of MapRaw from a file of XML values.
  104. // NOTE: the slice with the raw XML is clean with no extra capacity - unlike NewMapXmlReaderRaw().
  105. // It is slow at parsing a file from disk and is intended for relatively small utility files.
  106. func NewMapsFromXmlFileRaw(name string) ([]MapRaw, error) {
  107. fi, err := os.Stat(name)
  108. if err != nil {
  109. return nil, err
  110. }
  111. if !fi.Mode().IsRegular() {
  112. return nil, fmt.Errorf("file %s is not a regular file", name)
  113. }
  114. fh, err := os.Open(name)
  115. if err != nil {
  116. return nil, err
  117. }
  118. defer fh.Close()
  119. am := make([]MapRaw, 0)
  120. for {
  121. mr := new(MapRaw)
  122. mr.M, mr.R, err = NewMapXmlReaderRaw(fh)
  123. if err != nil && err != io.EOF {
  124. return am, fmt.Errorf("error: %s - reading: %s", err.Error(), string(mr.R))
  125. }
  126. if len(mr.M) > 0 {
  127. am = append(am, *mr)
  128. }
  129. if err == io.EOF {
  130. break
  131. }
  132. }
  133. return am, nil
  134. }
  135. // ------------------------ Maps writing -------------------------
  136. // These are handy-dandy methods for dumping configuration data, etc.
  137. // JsonString - analogous to mv.Json()
  138. func (mvs Maps) JsonString(safeEncoding ...bool) (string, error) {
  139. var s string
  140. for _, v := range mvs {
  141. j, err := v.Json()
  142. if err != nil {
  143. return s, err
  144. }
  145. s += string(j)
  146. }
  147. return s, nil
  148. }
  149. // JsonStringIndent - analogous to mv.JsonIndent()
  150. func (mvs Maps) JsonStringIndent(prefix, indent string, safeEncoding ...bool) (string, error) {
  151. var s string
  152. var haveFirst bool
  153. for _, v := range mvs {
  154. j, err := v.JsonIndent(prefix, indent)
  155. if err != nil {
  156. return s, err
  157. }
  158. if haveFirst {
  159. s += "\n"
  160. } else {
  161. haveFirst = true
  162. }
  163. s += string(j)
  164. }
  165. return s, nil
  166. }
  167. // XmlString - analogous to mv.Xml()
  168. func (mvs Maps) XmlString() (string, error) {
  169. var s string
  170. for _, v := range mvs {
  171. x, err := v.Xml()
  172. if err != nil {
  173. return s, err
  174. }
  175. s += string(x)
  176. }
  177. return s, nil
  178. }
  179. // XmlStringIndent - analogous to mv.XmlIndent()
  180. func (mvs Maps) XmlStringIndent(prefix, indent string) (string, error) {
  181. var s string
  182. for _, v := range mvs {
  183. x, err := v.XmlIndent(prefix, indent)
  184. if err != nil {
  185. return s, err
  186. }
  187. s += string(x)
  188. }
  189. return s, nil
  190. }
  191. // JsonFile - write Maps to named file as JSON
  192. // Note: the file will be created, if necessary; if it exists it will be truncated.
  193. // If you need to append to a file, open it and use JsonWriter method.
  194. func (mvs Maps) JsonFile(file string, safeEncoding ...bool) error {
  195. var encoding bool
  196. if len(safeEncoding) == 1 {
  197. encoding = safeEncoding[0]
  198. }
  199. s, err := mvs.JsonString(encoding)
  200. if err != nil {
  201. return err
  202. }
  203. fh, err := os.Create(file)
  204. if err != nil {
  205. return err
  206. }
  207. defer fh.Close()
  208. fh.WriteString(s)
  209. return nil
  210. }
  211. // JsonFileIndent - write Maps to named file as pretty JSON
  212. // Note: the file will be created, if necessary; if it exists it will be truncated.
  213. // If you need to append to a file, open it and use JsonIndentWriter method.
  214. func (mvs Maps) JsonFileIndent(file, prefix, indent string, safeEncoding ...bool) error {
  215. var encoding bool
  216. if len(safeEncoding) == 1 {
  217. encoding = safeEncoding[0]
  218. }
  219. s, err := mvs.JsonStringIndent(prefix, indent, encoding)
  220. if err != nil {
  221. return err
  222. }
  223. fh, err := os.Create(file)
  224. if err != nil {
  225. return err
  226. }
  227. defer fh.Close()
  228. fh.WriteString(s)
  229. return nil
  230. }
  231. // XmlFile - write Maps to named file as XML
  232. // Note: the file will be created, if necessary; if it exists it will be truncated.
  233. // If you need to append to a file, open it and use XmlWriter method.
  234. func (mvs Maps) XmlFile(file string) error {
  235. s, err := mvs.XmlString()
  236. if err != nil {
  237. return err
  238. }
  239. fh, err := os.Create(file)
  240. if err != nil {
  241. return err
  242. }
  243. defer fh.Close()
  244. fh.WriteString(s)
  245. return nil
  246. }
  247. // XmlFileIndent - write Maps to named file as pretty XML
  248. // Note: the file will be created,if necessary; if it exists it will be truncated.
  249. // If you need to append to a file, open it and use XmlIndentWriter method.
  250. func (mvs Maps) XmlFileIndent(file, prefix, indent string) error {
  251. s, err := mvs.XmlStringIndent(prefix, indent)
  252. if err != nil {
  253. return err
  254. }
  255. fh, err := os.Create(file)
  256. if err != nil {
  257. return err
  258. }
  259. defer fh.Close()
  260. fh.WriteString(s)
  261. return nil
  262. }