render.go 4.4 KB

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