integration.go 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444
  1. // SPDX-FileCopyrightText: Copyright The Miniflux Authors. All rights reserved.
  2. // SPDX-License-Identifier: Apache-2.0
  3. package storage // import "miniflux.app/v2/internal/storage"
  4. import (
  5. "database/sql"
  6. "fmt"
  7. "golang.org/x/crypto/bcrypt"
  8. "miniflux.app/v2/internal/model"
  9. )
  10. // HasDuplicateFeverUsername checks if another user have the same Fever username.
  11. func (s *Storage) HasDuplicateFeverUsername(userID int64, feverUsername string) bool {
  12. query := `SELECT true FROM integrations WHERE user_id != $1 AND fever_username=$2`
  13. var result bool
  14. s.db.QueryRow(query, userID, feverUsername).Scan(&result)
  15. return result
  16. }
  17. // HasDuplicateGoogleReaderUsername checks if another user have the same Google Reader username.
  18. func (s *Storage) HasDuplicateGoogleReaderUsername(userID int64, googleReaderUsername string) bool {
  19. query := `SELECT true FROM integrations WHERE user_id != $1 AND googlereader_username=$2`
  20. var result bool
  21. s.db.QueryRow(query, userID, googleReaderUsername).Scan(&result)
  22. return result
  23. }
  24. // UserByFeverToken returns a user by using the Fever API token.
  25. func (s *Storage) UserByFeverToken(token string) (*model.User, error) {
  26. query := `
  27. SELECT
  28. users.id, users.username, users.is_admin, users.timezone
  29. FROM
  30. users
  31. LEFT JOIN
  32. integrations ON integrations.user_id=users.id
  33. WHERE
  34. integrations.fever_enabled='t' AND lower(integrations.fever_token)=lower($1)
  35. `
  36. var user model.User
  37. err := s.db.QueryRow(query, token).Scan(&user.ID, &user.Username, &user.IsAdmin, &user.Timezone)
  38. switch {
  39. case err == sql.ErrNoRows:
  40. return nil, nil
  41. case err != nil:
  42. return nil, fmt.Errorf("store: unable to fetch user: %v", err)
  43. default:
  44. return &user, nil
  45. }
  46. }
  47. // GoogleReaderUserCheckPassword validates the Google Reader hashed password.
  48. func (s *Storage) GoogleReaderUserCheckPassword(username, password string) error {
  49. var hash string
  50. query := `
  51. SELECT
  52. googlereader_password
  53. FROM
  54. integrations
  55. WHERE
  56. integrations.googlereader_enabled='t' AND integrations.googlereader_username=$1
  57. `
  58. err := s.db.QueryRow(query, username).Scan(&hash)
  59. if err == sql.ErrNoRows {
  60. return fmt.Errorf(`store: unable to find this user: %s`, username)
  61. } else if err != nil {
  62. return fmt.Errorf(`store: unable to fetch user: %v`, err)
  63. }
  64. if err := bcrypt.CompareHashAndPassword([]byte(hash), []byte(password)); err != nil {
  65. return fmt.Errorf(`store: invalid password for "%s" (%v)`, username, err)
  66. }
  67. return nil
  68. }
  69. // GoogleReaderUserGetIntegration returns part of the Google Reader parts of the integration struct.
  70. func (s *Storage) GoogleReaderUserGetIntegration(username string) (*model.Integration, error) {
  71. var integration model.Integration
  72. query := `
  73. SELECT
  74. user_id,
  75. googlereader_enabled,
  76. googlereader_username,
  77. googlereader_password
  78. FROM
  79. integrations
  80. WHERE
  81. integrations.googlereader_enabled='t' AND integrations.googlereader_username=$1
  82. `
  83. err := s.db.QueryRow(query, username).Scan(&integration.UserID, &integration.GoogleReaderEnabled, &integration.GoogleReaderUsername, &integration.GoogleReaderPassword)
  84. if err == sql.ErrNoRows {
  85. return &integration, fmt.Errorf(`store: unable to find this user: %s`, username)
  86. } else if err != nil {
  87. return &integration, fmt.Errorf(`store: unable to fetch user: %v`, err)
  88. }
  89. return &integration, nil
  90. }
  91. // Integration returns user integration settings.
  92. func (s *Storage) Integration(userID int64) (*model.Integration, error) {
  93. query := `
  94. SELECT
  95. user_id,
  96. pinboard_enabled,
  97. pinboard_token,
  98. pinboard_tags,
  99. pinboard_mark_as_unread,
  100. instapaper_enabled,
  101. instapaper_username,
  102. instapaper_password,
  103. fever_enabled,
  104. fever_username,
  105. fever_token,
  106. googlereader_enabled,
  107. googlereader_username,
  108. googlereader_password,
  109. wallabag_enabled,
  110. wallabag_only_url,
  111. wallabag_url,
  112. wallabag_client_id,
  113. wallabag_client_secret,
  114. wallabag_username,
  115. wallabag_password,
  116. notion_enabled,
  117. notion_token,
  118. notion_page_id,
  119. nunux_keeper_enabled,
  120. nunux_keeper_url,
  121. nunux_keeper_api_key,
  122. espial_enabled,
  123. espial_url,
  124. espial_api_key,
  125. espial_tags,
  126. readwise_enabled,
  127. readwise_api_key,
  128. pocket_enabled,
  129. pocket_access_token,
  130. pocket_consumer_key,
  131. telegram_bot_enabled,
  132. telegram_bot_token,
  133. telegram_bot_chat_id,
  134. telegram_bot_topic_id,
  135. telegram_bot_disable_web_page_preview,
  136. telegram_bot_disable_notification,
  137. telegram_bot_disable_buttons,
  138. linkding_enabled,
  139. linkding_url,
  140. linkding_api_key,
  141. linkding_tags,
  142. linkding_mark_as_unread,
  143. matrix_bot_enabled,
  144. matrix_bot_user,
  145. matrix_bot_password,
  146. matrix_bot_url,
  147. matrix_bot_chat_id,
  148. apprise_enabled,
  149. apprise_url,
  150. apprise_services_url,
  151. shiori_enabled,
  152. shiori_url,
  153. shiori_username,
  154. shiori_password,
  155. shaarli_enabled,
  156. shaarli_url,
  157. shaarli_api_secret,
  158. webhook_enabled,
  159. webhook_url,
  160. webhook_secret
  161. FROM
  162. integrations
  163. WHERE
  164. user_id=$1
  165. `
  166. var integration model.Integration
  167. err := s.db.QueryRow(query, userID).Scan(
  168. &integration.UserID,
  169. &integration.PinboardEnabled,
  170. &integration.PinboardToken,
  171. &integration.PinboardTags,
  172. &integration.PinboardMarkAsUnread,
  173. &integration.InstapaperEnabled,
  174. &integration.InstapaperUsername,
  175. &integration.InstapaperPassword,
  176. &integration.FeverEnabled,
  177. &integration.FeverUsername,
  178. &integration.FeverToken,
  179. &integration.GoogleReaderEnabled,
  180. &integration.GoogleReaderUsername,
  181. &integration.GoogleReaderPassword,
  182. &integration.WallabagEnabled,
  183. &integration.WallabagOnlyURL,
  184. &integration.WallabagURL,
  185. &integration.WallabagClientID,
  186. &integration.WallabagClientSecret,
  187. &integration.WallabagUsername,
  188. &integration.WallabagPassword,
  189. &integration.NotionEnabled,
  190. &integration.NotionToken,
  191. &integration.NotionPageID,
  192. &integration.NunuxKeeperEnabled,
  193. &integration.NunuxKeeperURL,
  194. &integration.NunuxKeeperAPIKey,
  195. &integration.EspialEnabled,
  196. &integration.EspialURL,
  197. &integration.EspialAPIKey,
  198. &integration.EspialTags,
  199. &integration.ReadwiseEnabled,
  200. &integration.ReadwiseAPIKey,
  201. &integration.PocketEnabled,
  202. &integration.PocketAccessToken,
  203. &integration.PocketConsumerKey,
  204. &integration.TelegramBotEnabled,
  205. &integration.TelegramBotToken,
  206. &integration.TelegramBotChatID,
  207. &integration.TelegramBotTopicID,
  208. &integration.TelegramBotDisableWebPagePreview,
  209. &integration.TelegramBotDisableNotification,
  210. &integration.TelegramBotDisableButtons,
  211. &integration.LinkdingEnabled,
  212. &integration.LinkdingURL,
  213. &integration.LinkdingAPIKey,
  214. &integration.LinkdingTags,
  215. &integration.LinkdingMarkAsUnread,
  216. &integration.MatrixBotEnabled,
  217. &integration.MatrixBotUser,
  218. &integration.MatrixBotPassword,
  219. &integration.MatrixBotURL,
  220. &integration.MatrixBotChatID,
  221. &integration.AppriseEnabled,
  222. &integration.AppriseURL,
  223. &integration.AppriseServicesURL,
  224. &integration.ShioriEnabled,
  225. &integration.ShioriURL,
  226. &integration.ShioriUsername,
  227. &integration.ShioriPassword,
  228. &integration.ShaarliEnabled,
  229. &integration.ShaarliURL,
  230. &integration.ShaarliAPISecret,
  231. &integration.WebhookEnabled,
  232. &integration.WebhookURL,
  233. &integration.WebhookSecret,
  234. )
  235. switch {
  236. case err == sql.ErrNoRows:
  237. return &integration, nil
  238. case err != nil:
  239. return &integration, fmt.Errorf(`store: unable to fetch integration row: %v`, err)
  240. default:
  241. return &integration, nil
  242. }
  243. }
  244. // UpdateIntegration saves user integration settings.
  245. func (s *Storage) UpdateIntegration(integration *model.Integration) error {
  246. query := `
  247. UPDATE
  248. integrations
  249. SET
  250. pinboard_enabled=$1,
  251. pinboard_token=$2,
  252. pinboard_tags=$3,
  253. pinboard_mark_as_unread=$4,
  254. instapaper_enabled=$5,
  255. instapaper_username=$6,
  256. instapaper_password=$7,
  257. fever_enabled=$8,
  258. fever_username=$9,
  259. fever_token=$10,
  260. wallabag_enabled=$11,
  261. wallabag_only_url=$12,
  262. wallabag_url=$13,
  263. wallabag_client_id=$14,
  264. wallabag_client_secret=$15,
  265. wallabag_username=$16,
  266. wallabag_password=$17,
  267. nunux_keeper_enabled=$18,
  268. nunux_keeper_url=$19,
  269. nunux_keeper_api_key=$20,
  270. pocket_enabled=$21,
  271. pocket_access_token=$22,
  272. pocket_consumer_key=$23,
  273. googlereader_enabled=$24,
  274. googlereader_username=$25,
  275. googlereader_password=$26,
  276. telegram_bot_enabled=$27,
  277. telegram_bot_token=$28,
  278. telegram_bot_chat_id=$29,
  279. telegram_bot_topic_id=$30,
  280. telegram_bot_disable_web_page_preview=$31,
  281. telegram_bot_disable_notification=$32,
  282. telegram_bot_disable_buttons=$33,
  283. espial_enabled=$34,
  284. espial_url=$35,
  285. espial_api_key=$36,
  286. espial_tags=$37,
  287. linkding_enabled=$38,
  288. linkding_url=$39,
  289. linkding_api_key=$40,
  290. linkding_tags=$41,
  291. linkding_mark_as_unread=$42,
  292. matrix_bot_enabled=$43,
  293. matrix_bot_user=$44,
  294. matrix_bot_password=$45,
  295. matrix_bot_url=$46,
  296. matrix_bot_chat_id=$47,
  297. notion_enabled=$48,
  298. notion_token=$49,
  299. notion_page_id=$50,
  300. readwise_enabled=$51,
  301. readwise_api_key=$52,
  302. apprise_enabled=$53,
  303. apprise_url=$54,
  304. apprise_services_url=$55,
  305. shiori_enabled=$56,
  306. shiori_url=$57,
  307. shiori_username=$58,
  308. shiori_password=$59,
  309. shaarli_enabled=$60,
  310. shaarli_url=$61,
  311. shaarli_api_secret=$62,
  312. webhook_enabled=$63,
  313. webhook_url=$64,
  314. webhook_secret=$65
  315. WHERE
  316. user_id=$66
  317. `
  318. _, err := s.db.Exec(
  319. query,
  320. integration.PinboardEnabled,
  321. integration.PinboardToken,
  322. integration.PinboardTags,
  323. integration.PinboardMarkAsUnread,
  324. integration.InstapaperEnabled,
  325. integration.InstapaperUsername,
  326. integration.InstapaperPassword,
  327. integration.FeverEnabled,
  328. integration.FeverUsername,
  329. integration.FeverToken,
  330. integration.WallabagEnabled,
  331. integration.WallabagOnlyURL,
  332. integration.WallabagURL,
  333. integration.WallabagClientID,
  334. integration.WallabagClientSecret,
  335. integration.WallabagUsername,
  336. integration.WallabagPassword,
  337. integration.NunuxKeeperEnabled,
  338. integration.NunuxKeeperURL,
  339. integration.NunuxKeeperAPIKey,
  340. integration.PocketEnabled,
  341. integration.PocketAccessToken,
  342. integration.PocketConsumerKey,
  343. integration.GoogleReaderEnabled,
  344. integration.GoogleReaderUsername,
  345. integration.GoogleReaderPassword,
  346. integration.TelegramBotEnabled,
  347. integration.TelegramBotToken,
  348. integration.TelegramBotChatID,
  349. integration.TelegramBotTopicID,
  350. integration.TelegramBotDisableWebPagePreview,
  351. integration.TelegramBotDisableNotification,
  352. integration.TelegramBotDisableButtons,
  353. integration.EspialEnabled,
  354. integration.EspialURL,
  355. integration.EspialAPIKey,
  356. integration.EspialTags,
  357. integration.LinkdingEnabled,
  358. integration.LinkdingURL,
  359. integration.LinkdingAPIKey,
  360. integration.LinkdingTags,
  361. integration.LinkdingMarkAsUnread,
  362. integration.MatrixBotEnabled,
  363. integration.MatrixBotUser,
  364. integration.MatrixBotPassword,
  365. integration.MatrixBotURL,
  366. integration.MatrixBotChatID,
  367. integration.NotionEnabled,
  368. integration.NotionToken,
  369. integration.NotionPageID,
  370. integration.ReadwiseEnabled,
  371. integration.ReadwiseAPIKey,
  372. integration.AppriseEnabled,
  373. integration.AppriseURL,
  374. integration.AppriseServicesURL,
  375. integration.ShioriEnabled,
  376. integration.ShioriURL,
  377. integration.ShioriUsername,
  378. integration.ShioriPassword,
  379. integration.ShaarliEnabled,
  380. integration.ShaarliURL,
  381. integration.ShaarliAPISecret,
  382. integration.WebhookEnabled,
  383. integration.WebhookURL,
  384. integration.WebhookSecret,
  385. integration.UserID,
  386. )
  387. if err != nil {
  388. return fmt.Errorf(`store: unable to update integration record: %v`, err)
  389. }
  390. return nil
  391. }
  392. // HasSaveEntry returns true if the given user can save articles to third-parties.
  393. func (s *Storage) HasSaveEntry(userID int64) (result bool) {
  394. query := `
  395. SELECT
  396. true
  397. FROM
  398. integrations
  399. WHERE
  400. user_id=$1
  401. AND
  402. (
  403. pinboard_enabled='t' OR
  404. instapaper_enabled='t' OR
  405. wallabag_enabled='t' OR
  406. notion_enabled='t' OR
  407. nunux_keeper_enabled='t' OR
  408. espial_enabled='t' OR
  409. readwise_enabled='t' OR
  410. pocket_enabled='t' OR
  411. linkding_enabled='t' OR
  412. apprise_enabled='t' OR
  413. shiori_enabled='t' OR
  414. shaarli_enabled='t' OR
  415. webhook_enabled='t'
  416. )
  417. `
  418. if err := s.db.QueryRow(query, userID).Scan(&result); err != nil {
  419. result = false
  420. }
  421. return result
  422. }