render.go 4.3 KB

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