From 6a3828167c5aee01b698b851db8c446bd8267b2b Mon Sep 17 00:00:00 2001
From: Andrew Kwon <andrewshky@gmail.com>
Date: Sat, 15 Jan 2022 22:47:49 -0800
Subject: [PATCH] Add setup option to configure custom file assets

Resolve #11
---
 README.md                    |  1 +
 lua/presence/file_assets.lua | 13 ++++++++++++-
 lua/presence/init.lua        | 11 +++++++++--
 3 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 4646bbf..9d170c1 100644
--- a/README.md
+++ b/README.md
@@ -43,6 +43,7 @@ require("presence"):setup({
     enable_line_number  = false,                      -- Displays the current line number instead of the current project
     blacklist           = {},                         -- A list of strings or Lua patterns that disable Rich Presence if the current file name, path, or workspace matches
     buttons             = true,                       -- Configure Rich Presence button(s), either a boolean to enable/disable, a static table (`{{ label = "<label>", url = "<url>" }, ...}`, or a function(buffer: string, repo_url: string|nil): table)
+    file_assets         = {},                         -- Custom file asset definitions keyed by file names and extensions (see default config at `lua/presence/file_assets.lua` for reference)
 
     -- Rich Presence text options
     editing_text        = "Editing %s",               -- Format string rendered when an editable file is loaded in the buffer (either string or function(filename: string): string)
diff --git a/lua/presence/file_assets.lua b/lua/presence/file_assets.lua
index 6b557b7..3a5907f 100644
--- a/lua/presence/file_assets.lua
+++ b/lua/presence/file_assets.lua
@@ -1,4 +1,15 @@
--- Discord application asset file names (name, key[, description]) by file extension
+-- File asset definitions (name, source) keyed by file name or extension
+-- * name - the name of the asset shown as the title of the file in Discord
+-- * source - the source of the asset, either an art asset key or the URL of an image asset
+--
+-- Example: {
+--     -- Use art assets uploaded in Discord application for the configured client id
+--     js = { "JavaScript", "javascript" },
+--     ts = { "TypeScript", "typescript" },
+--     -- Use image URLs
+--     rs = { "Rust", "https://www.rust-lang.org/logos/rust-logo-512x512.png" },
+--     go = { "Go", "https://go.dev/blog/go-brand/Go-Logo/PNG/Go-Logo_Aqua.png" },
+-- }
 return {
     [".aliases"] = { ".aliases", "shell" },
     [".appveyor.yml"] = { "AppVeyor config", "appveyor" },
diff --git a/lua/presence/init.lua b/lua/presence/init.lua
index aebdf22..8bb5a1c 100644
--- a/lua/presence/init.lua
+++ b/lua/presence/init.lua
@@ -63,8 +63,8 @@ Presence.workspaces = {}
 local log = require("lib.log")
 local msgpack = require("deps.msgpack")
 local serpent = require("deps.serpent")
-local file_assets = require("presence.file_assets")
 local file_explorers = require("presence.file_explorers")
+local default_file_assets = require("presence.file_assets")
 local plugin_managers = require("presence.plugin_managers")
 local Discord = require("presence.discord")
 
@@ -121,6 +121,13 @@ function Presence:setup(options)
     self:set_option("line_number_text", "Line %s out of %s")
     self:set_option("blacklist", {})
     self:set_option("buttons", true)
+    -- File assets options
+    self:set_option("file_assets", {})
+    for name, asset in pairs(default_file_assets) do
+        if not self.options.file_assets[name] then
+            self.options.file_assets[name] = asset
+        end
+    end
 
     -- Get and check discord socket path
     local discord_socket_path = self:get_discord_socket_path()
@@ -790,7 +797,7 @@ function Presence:update_for_buffer(buffer, should_debounce)
     local name = filename
     local asset_key = "code"
     local description = filename
-    local file_asset = file_assets[filename] or file_assets[extension]
+    local file_asset = self.options.file_assets[filename] or self.options.file_assets[extension]
     if file_asset then
         name, asset_key, description = unpack(file_asset)
         self.log:debug(string.format("Using file asset: %s", vim.inspect(file_asset)))