render.go 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178
  1. package tool
  2. import (
  3. "database/sql"
  4. "log"
  5. "strconv"
  6. "time"
  7. )
  8. func Get_render(db *sql.DB, db_set map[string]string, doc_name string, data string, render_type string) map[string]string {
  9. var markup string
  10. if render_type == "api_view" || render_type == "api_from" || render_type == "api_include" || render_type == "backlink" {
  11. stmt, err := db.Prepare(DB_change(db_set, "select set_data from data_set where doc_name = ? and set_name = 'document_markup'"))
  12. if err != nil {
  13. log.Fatal(err)
  14. }
  15. defer stmt.Close()
  16. err = stmt.QueryRow(doc_name).Scan(&markup)
  17. if err != nil {
  18. if err == sql.ErrNoRows {
  19. markup = ""
  20. } else {
  21. log.Fatal(err)
  22. }
  23. }
  24. }
  25. if markup == "" {
  26. err := db.QueryRow(DB_change(db_set, "select data from other where name = 'markup'")).Scan(&markup)
  27. if err != nil {
  28. if err == sql.ErrNoRows {
  29. markup = ""
  30. } else {
  31. log.Fatal(err)
  32. }
  33. }
  34. }
  35. if markup == "" {
  36. markup = "namumark"
  37. }
  38. now_time := time.Now().UnixNano()
  39. render_name := strconv.Itoa(int(now_time))
  40. render_data := Get_render_direct(db, db_set, doc_name, data, markup, render_name, render_type)
  41. return render_data
  42. }
  43. func Get_render_direct(db *sql.DB, db_set map[string]string, doc_name string, data string, markup string, render_name string, render_type string) map[string]string {
  44. from := ""
  45. include := ""
  46. backlink := ""
  47. if render_type == "api_include" {
  48. include = "1"
  49. } else if render_type == "api_from" {
  50. from = "1"
  51. } else if render_type == "backlink" {
  52. backlink = "1"
  53. }
  54. if render_type == "api_view" || render_type == "api_from" || render_type == "api_include" || render_type == "backlink" {
  55. render_type = "view"
  56. }
  57. log.Fatal(data)
  58. doc_data_set := map[string]string{
  59. "doc_name": doc_name,
  60. "data": data,
  61. "render_name": render_name,
  62. "render_type": render_type,
  63. "from": from,
  64. "include": include,
  65. }
  66. render_data := make(map[string]interface{})
  67. if markup == "namumark" {
  68. render_data_class := Namumark_new(db, db_set, doc_data_set)
  69. render_data = render_data_class.main()
  70. } else if markup == "markdown" {
  71. render_data = Markdown(db, db_set, doc_data_set)
  72. } else {
  73. render_data["data"] = data
  74. render_data["js_data"] = ""
  75. render_data["backlink"] = [][]string{}
  76. }
  77. if backlink == "1" {
  78. stmt, err := db.Prepare(DB_change(db_set, "delete from back where link = ?"))
  79. if err != nil {
  80. log.Fatal(err)
  81. }
  82. defer stmt.Close()
  83. _, err = stmt.Exec(doc_name)
  84. if err != nil {
  85. log.Fatal(err)
  86. }
  87. stmt, err = db.Prepare(DB_change(db_set, "delete from back where title = ? and type = 'no'"))
  88. if err != nil {
  89. log.Fatal(err)
  90. }
  91. defer stmt.Close()
  92. _, err = stmt.Exec(doc_name)
  93. if err != nil {
  94. log.Fatal(err)
  95. }
  96. stmt, err = db.Prepare(DB_change(db_set, "delete from data_set where doc_name = ? and set_name = 'link_count'"))
  97. if err != nil {
  98. log.Fatal(err)
  99. }
  100. defer stmt.Close()
  101. _, err = stmt.Exec(doc_name)
  102. if err != nil {
  103. log.Fatal(err)
  104. }
  105. stmt, err = db.Prepare(DB_change(db_set, "delete from data_set where doc_name = ? and set_name = 'doc_type'"))
  106. if err != nil {
  107. log.Fatal(err)
  108. }
  109. defer stmt.Close()
  110. _, err = stmt.Exec(doc_name)
  111. if err != nil {
  112. log.Fatal(err)
  113. }
  114. end_backlink := render_data["backlink"].([][]string)
  115. for for_a := 0; for_a < len(end_backlink); for_a++ {
  116. stmt, err := db.Prepare(DB_change(db_set, "insert into back (link, title, type, data) values (?, ?, ?, ?)"))
  117. if err != nil {
  118. log.Fatal(err)
  119. }
  120. defer stmt.Close()
  121. _, err = stmt.Exec(end_backlink[0], end_backlink[1], end_backlink[2])
  122. if err != nil {
  123. log.Fatal(err)
  124. }
  125. }
  126. stmt, err = db.Prepare(DB_change(db_set, "insert into data_set (doc_name, doc_rev, set_name, set_data) values (?, '', 'link_count', ?)"))
  127. if err != nil {
  128. log.Fatal(err)
  129. }
  130. defer stmt.Close()
  131. _, err = stmt.Exec(doc_name, render_data["link_count"].(int))
  132. if err != nil {
  133. log.Fatal(err)
  134. }
  135. stmt, err = db.Prepare(DB_change(db_set, "insert into data_set (doc_name, doc_rev, set_name, set_data) values (?, '', 'doc_type', ?)"))
  136. if err != nil {
  137. log.Fatal(err)
  138. }
  139. defer stmt.Close()
  140. _, err = stmt.Exec(doc_name, "")
  141. if err != nil {
  142. log.Fatal(err)
  143. }
  144. }
  145. return map[string]string{
  146. "data": "<div id=\"opennamu_render_complete\">" + render_data["data"].(string) + "</div>",
  147. "js_data": render_data["js_data"].(string),
  148. }
  149. }