integration.go 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452
  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. rssbridge_enabled,
  162. rssbridge_url
  163. FROM
  164. integrations
  165. WHERE
  166. user_id=$1
  167. `
  168. var integration model.Integration
  169. err := s.db.QueryRow(query, userID).Scan(
  170. &integration.UserID,
  171. &integration.PinboardEnabled,
  172. &integration.PinboardToken,
  173. &integration.PinboardTags,
  174. &integration.PinboardMarkAsUnread,
  175. &integration.InstapaperEnabled,
  176. &integration.InstapaperUsername,
  177. &integration.InstapaperPassword,
  178. &integration.FeverEnabled,
  179. &integration.FeverUsername,
  180. &integration.FeverToken,
  181. &integration.GoogleReaderEnabled,
  182. &integration.GoogleReaderUsername,
  183. &integration.GoogleReaderPassword,
  184. &integration.WallabagEnabled,
  185. &integration.WallabagOnlyURL,
  186. &integration.WallabagURL,
  187. &integration.WallabagClientID,
  188. &integration.WallabagClientSecret,
  189. &integration.WallabagUsername,
  190. &integration.WallabagPassword,
  191. &integration.NotionEnabled,
  192. &integration.NotionToken,
  193. &integration.NotionPageID,
  194. &integration.NunuxKeeperEnabled,
  195. &integration.NunuxKeeperURL,
  196. &integration.NunuxKeeperAPIKey,
  197. &integration.EspialEnabled,
  198. &integration.EspialURL,
  199. &integration.EspialAPIKey,
  200. &integration.EspialTags,
  201. &integration.ReadwiseEnabled,
  202. &integration.ReadwiseAPIKey,
  203. &integration.PocketEnabled,
  204. &integration.PocketAccessToken,
  205. &integration.PocketConsumerKey,
  206. &integration.TelegramBotEnabled,
  207. &integration.TelegramBotToken,
  208. &integration.TelegramBotChatID,
  209. &integration.TelegramBotTopicID,
  210. &integration.TelegramBotDisableWebPagePreview,
  211. &integration.TelegramBotDisableNotification,
  212. &integration.TelegramBotDisableButtons,
  213. &integration.LinkdingEnabled,
  214. &integration.LinkdingURL,
  215. &integration.LinkdingAPIKey,
  216. &integration.LinkdingTags,
  217. &integration.LinkdingMarkAsUnread,
  218. &integration.MatrixBotEnabled,
  219. &integration.MatrixBotUser,
  220. &integration.MatrixBotPassword,
  221. &integration.MatrixBotURL,
  222. &integration.MatrixBotChatID,
  223. &integration.AppriseEnabled,
  224. &integration.AppriseURL,
  225. &integration.AppriseServicesURL,
  226. &integration.ShioriEnabled,
  227. &integration.ShioriURL,
  228. &integration.ShioriUsername,
  229. &integration.ShioriPassword,
  230. &integration.ShaarliEnabled,
  231. &integration.ShaarliURL,
  232. &integration.ShaarliAPISecret,
  233. &integration.WebhookEnabled,
  234. &integration.WebhookURL,
  235. &integration.WebhookSecret,
  236. &integration.RSSBridgeEnabled,
  237. &integration.RSSBridgeURL,
  238. )
  239. switch {
  240. case err == sql.ErrNoRows:
  241. return &integration, nil
  242. case err != nil:
  243. return &integration, fmt.Errorf(`store: unable to fetch integration row: %v`, err)
  244. default:
  245. return &integration, nil
  246. }
  247. }
  248. // UpdateIntegration saves user integration settings.
  249. func (s *Storage) UpdateIntegration(integration *model.Integration) error {
  250. query := `
  251. UPDATE
  252. integrations
  253. SET
  254. pinboard_enabled=$1,
  255. pinboard_token=$2,
  256. pinboard_tags=$3,
  257. pinboard_mark_as_unread=$4,
  258. instapaper_enabled=$5,
  259. instapaper_username=$6,
  260. instapaper_password=$7,
  261. fever_enabled=$8,
  262. fever_username=$9,
  263. fever_token=$10,
  264. wallabag_enabled=$11,
  265. wallabag_only_url=$12,
  266. wallabag_url=$13,
  267. wallabag_client_id=$14,
  268. wallabag_client_secret=$15,
  269. wallabag_username=$16,
  270. wallabag_password=$17,
  271. nunux_keeper_enabled=$18,
  272. nunux_keeper_url=$19,
  273. nunux_keeper_api_key=$20,
  274. pocket_enabled=$21,
  275. pocket_access_token=$22,
  276. pocket_consumer_key=$23,
  277. googlereader_enabled=$24,
  278. googlereader_username=$25,
  279. googlereader_password=$26,
  280. telegram_bot_enabled=$27,
  281. telegram_bot_token=$28,
  282. telegram_bot_chat_id=$29,
  283. telegram_bot_topic_id=$30,
  284. telegram_bot_disable_web_page_preview=$31,
  285. telegram_bot_disable_notification=$32,
  286. telegram_bot_disable_buttons=$33,
  287. espial_enabled=$34,
  288. espial_url=$35,
  289. espial_api_key=$36,
  290. espial_tags=$37,
  291. linkding_enabled=$38,
  292. linkding_url=$39,
  293. linkding_api_key=$40,
  294. linkding_tags=$41,
  295. linkding_mark_as_unread=$42,
  296. matrix_bot_enabled=$43,
  297. matrix_bot_user=$44,
  298. matrix_bot_password=$45,
  299. matrix_bot_url=$46,
  300. matrix_bot_chat_id=$47,
  301. notion_enabled=$48,
  302. notion_token=$49,
  303. notion_page_id=$50,
  304. readwise_enabled=$51,
  305. readwise_api_key=$52,
  306. apprise_enabled=$53,
  307. apprise_url=$54,
  308. apprise_services_url=$55,
  309. shiori_enabled=$56,
  310. shiori_url=$57,
  311. shiori_username=$58,
  312. shiori_password=$59,
  313. shaarli_enabled=$60,
  314. shaarli_url=$61,
  315. shaarli_api_secret=$62,
  316. webhook_enabled=$63,
  317. webhook_url=$64,
  318. webhook_secret=$65,
  319. rssbridge_enabled=$66,
  320. rssbridge_url=$67
  321. WHERE
  322. user_id=$68
  323. `
  324. _, err := s.db.Exec(
  325. query,
  326. integration.PinboardEnabled,
  327. integration.PinboardToken,
  328. integration.PinboardTags,
  329. integration.PinboardMarkAsUnread,
  330. integration.InstapaperEnabled,
  331. integration.InstapaperUsername,
  332. integration.InstapaperPassword,
  333. integration.FeverEnabled,
  334. integration.FeverUsername,
  335. integration.FeverToken,
  336. integration.WallabagEnabled,
  337. integration.WallabagOnlyURL,
  338. integration.WallabagURL,
  339. integration.WallabagClientID,
  340. integration.WallabagClientSecret,
  341. integration.WallabagUsername,
  342. integration.WallabagPassword,
  343. integration.NunuxKeeperEnabled,
  344. integration.NunuxKeeperURL,
  345. integration.NunuxKeeperAPIKey,
  346. integration.PocketEnabled,
  347. integration.PocketAccessToken,
  348. integration.PocketConsumerKey,
  349. integration.GoogleReaderEnabled,
  350. integration.GoogleReaderUsername,
  351. integration.GoogleReaderPassword,
  352. integration.TelegramBotEnabled,
  353. integration.TelegramBotToken,
  354. integration.TelegramBotChatID,
  355. integration.TelegramBotTopicID,
  356. integration.TelegramBotDisableWebPagePreview,
  357. integration.TelegramBotDisableNotification,
  358. integration.TelegramBotDisableButtons,
  359. integration.EspialEnabled,
  360. integration.EspialURL,
  361. integration.EspialAPIKey,
  362. integration.EspialTags,
  363. integration.LinkdingEnabled,
  364. integration.LinkdingURL,
  365. integration.LinkdingAPIKey,
  366. integration.LinkdingTags,
  367. integration.LinkdingMarkAsUnread,
  368. integration.MatrixBotEnabled,
  369. integration.MatrixBotUser,
  370. integration.MatrixBotPassword,
  371. integration.MatrixBotURL,
  372. integration.MatrixBotChatID,
  373. integration.NotionEnabled,
  374. integration.NotionToken,
  375. integration.NotionPageID,
  376. integration.ReadwiseEnabled,
  377. integration.ReadwiseAPIKey,
  378. integration.AppriseEnabled,
  379. integration.AppriseURL,
  380. integration.AppriseServicesURL,
  381. integration.ShioriEnabled,
  382. integration.ShioriURL,
  383. integration.ShioriUsername,
  384. integration.ShioriPassword,
  385. integration.ShaarliEnabled,
  386. integration.ShaarliURL,
  387. integration.ShaarliAPISecret,
  388. integration.WebhookEnabled,
  389. integration.WebhookURL,
  390. integration.WebhookSecret,
  391. integration.RSSBridgeEnabled,
  392. integration.RSSBridgeURL,
  393. integration.UserID,
  394. )
  395. if err != nil {
  396. return fmt.Errorf(`store: unable to update integration record: %v`, err)
  397. }
  398. return nil
  399. }
  400. // HasSaveEntry returns true if the given user can save articles to third-parties.
  401. func (s *Storage) HasSaveEntry(userID int64) (result bool) {
  402. query := `
  403. SELECT
  404. true
  405. FROM
  406. integrations
  407. WHERE
  408. user_id=$1
  409. AND
  410. (
  411. pinboard_enabled='t' OR
  412. instapaper_enabled='t' OR
  413. wallabag_enabled='t' OR
  414. notion_enabled='t' OR
  415. nunux_keeper_enabled='t' OR
  416. espial_enabled='t' OR
  417. readwise_enabled='t' OR
  418. pocket_enabled='t' OR
  419. linkding_enabled='t' OR
  420. apprise_enabled='t' OR
  421. shiori_enabled='t' OR
  422. shaarli_enabled='t' OR
  423. webhook_enabled='t'
  424. )
  425. `
  426. if err := s.db.QueryRow(query, userID).Scan(&result); err != nil {
  427. result = false
  428. }
  429. return result
  430. }