render.go 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175
  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. doc_data_set := map[string]string{
  58. "doc_name": doc_name,
  59. "data": data,
  60. "render_name": render_name,
  61. "render_type": render_type,
  62. "from": from,
  63. "include": include,
  64. }
  65. render_data := make(map[string]interface{})
  66. if markup == "namumark" {
  67. render_data = Namumark()
  68. } else if markup == "markdown" {
  69. render_data = Markdown(db, db_set, doc_data_set)
  70. } else {
  71. render_data["data"] = data
  72. render_data["js_data"] = ""
  73. render_data["backlink"] = [][]string{}
  74. }
  75. if backlink == "1" {
  76. stmt, err := db.Prepare(DB_change(db_set, "delete from back where link = ?"))
  77. if err != nil {
  78. log.Fatal(err)
  79. }
  80. defer stmt.Close()
  81. _, err = stmt.Exec(doc_name)
  82. if err != nil {
  83. log.Fatal(err)
  84. }
  85. stmt, err = db.Prepare(DB_change(db_set, "delete from back where title = ? and type = 'no'"))
  86. if err != nil {
  87. log.Fatal(err)
  88. }
  89. defer stmt.Close()
  90. _, err = stmt.Exec(doc_name)
  91. if err != nil {
  92. log.Fatal(err)
  93. }
  94. stmt, err = db.Prepare(DB_change(db_set, "delete from data_set where doc_name = ? and set_name = 'link_count'"))
  95. if err != nil {
  96. log.Fatal(err)
  97. }
  98. defer stmt.Close()
  99. _, err = stmt.Exec(doc_name)
  100. if err != nil {
  101. log.Fatal(err)
  102. }
  103. stmt, err = db.Prepare(DB_change(db_set, "delete from data_set where doc_name = ? and set_name = 'doc_type'"))
  104. if err != nil {
  105. log.Fatal(err)
  106. }
  107. defer stmt.Close()
  108. _, err = stmt.Exec(doc_name)
  109. if err != nil {
  110. log.Fatal(err)
  111. }
  112. end_backlink := render_data["backlink"].([][]string)
  113. for for_a := 0; for_a < len(end_backlink); for_a++ {
  114. stmt, err := db.Prepare(DB_change(db_set, "insert into back (link, title, type, data) values (?, ?, ?, ?)"))
  115. if err != nil {
  116. log.Fatal(err)
  117. }
  118. defer stmt.Close()
  119. _, err = stmt.Exec(end_backlink[0], end_backlink[1], end_backlink[2])
  120. if err != nil {
  121. log.Fatal(err)
  122. }
  123. }
  124. stmt, err = db.Prepare(DB_change(db_set, "insert into data_set (doc_name, doc_rev, set_name, set_data) values (?, '', 'link_count', ?)"))
  125. if err != nil {
  126. log.Fatal(err)
  127. }
  128. defer stmt.Close()
  129. _, err = stmt.Exec(doc_name, render_data["link_count"].(int))
  130. if err != nil {
  131. log.Fatal(err)
  132. }
  133. stmt, err = db.Prepare(DB_change(db_set, "insert into data_set (doc_name, doc_rev, set_name, set_data) values (?, '', 'doc_type', ?)"))
  134. if err != nil {
  135. log.Fatal(err)
  136. }
  137. defer stmt.Close()
  138. _, err = stmt.Exec(doc_name, "")
  139. if err != nil {
  140. log.Fatal(err)
  141. }
  142. }
  143. return map[string]string{
  144. "data": "<div id=\"opennamu_render_complete\">" + render_data["data"].(string) + "</div>",
  145. "js_data": render_data["js_data"].(string),
  146. }
  147. }