108 lines
3.5 KiB
TypeScript
108 lines
3.5 KiB
TypeScript
|
/*
|
||
|
|
* Vencord, a Discord client mod
|
||
|
|
* Copyright (c) 2024 Vendicated and contributors
|
||
|
|
* SPDX-License-Identifier: GPL-3.0-or-later
|
||
|
|
*/
|
||
|
|
|
||
|
|
import { findGroupChildrenByChildId, NavContextMenuPatchCallback } from "@api/ContextMenu";
|
||
|
|
import { definePluginSettings } from "@api/Settings";
|
||
|
|
import { Devs } from "@utils/constants";
|
||
|
|
import definePlugin, { OptionType } from "@utils/types";
|
||
|
|
import { Flex, Menu } from "@webpack/common";
|
||
|
|
|
||
|
|
const DefaultEngines = {
|
||
|
|
Google: "https://www.google.com/search?q=",
|
||
|
|
DuckDuckGo: "https://duckduckgo.com/",
|
||
|
|
Bing: "https://www.bing.com/search?q=",
|
||
|
|
Yahoo: "https://search.yahoo.com/search?p=",
|
||
|
|
Github: "https://github.com/search?q=",
|
||
|
|
Kagi: "https://kagi.com/search?q=",
|
||
|
|
Yandex: "https://yandex.com/search/?text=",
|
||
|
|
AOL: "https://search.aol.com/aol/search?q=",
|
||
|
|
Baidu: "https://www.baidu.com/s?wd=",
|
||
|
|
Wikipedia: "https://wikipedia.org/w/index.php?search=",
|
||
|
|
} as const;
|
||
|
|
|
||
|
|
const settings = definePluginSettings({
|
||
|
|
customEngineName: {
|
||
|
|
description: "Name of the custom search engine",
|
||
|
|
type: OptionType.STRING,
|
||
|
|
placeholder: "Google"
|
||
|
|
},
|
||
|
|
customEngineURL: {
|
||
|
|
description: "The URL of your Engine",
|
||
|
|
type: OptionType.STRING,
|
||
|
|
placeholder: "https://google.com/search?q="
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
function search(src: string, engine: string) {
|
||
|
|
open(engine + encodeURIComponent(src), "_blank");
|
||
|
|
}
|
||
|
|
|
||
|
|
function makeSearchItem(src: string) {
|
||
|
|
let Engines = {};
|
||
|
|
|
||
|
|
if (settings.store.customEngineName && settings.store.customEngineURL) {
|
||
|
|
Engines[settings.store.customEngineName] = settings.store.customEngineURL;
|
||
|
|
}
|
||
|
|
|
||
|
|
Engines = { ...Engines, ...DefaultEngines };
|
||
|
|
|
||
|
|
return (
|
||
|
|
<Menu.MenuItem
|
||
|
|
label="Search Text"
|
||
|
|
key="search-text"
|
||
|
|
id="vc-search-text"
|
||
|
|
>
|
||
|
|
{Object.keys(Engines).map((engine, i) => {
|
||
|
|
const key = "vc-search-content-" + engine;
|
||
|
|
return (
|
||
|
|
<Menu.MenuItem
|
||
|
|
key={key}
|
||
|
|
id={key}
|
||
|
|
label={
|
||
|
|
<Flex style={{ alignItems: "center", gap: "0.5em" }}>
|
||
|
|
<img
|
||
|
|
style={{
|
||
|
|
borderRadius: "50%"
|
||
|
|
}}
|
||
|
|
aria-hidden="true"
|
||
|
|
height={16}
|
||
|
|
width={16}
|
||
|
|
src={`https://www.google.com/s2/favicons?domain=${Engines[engine]}`}
|
||
|
|
/>
|
||
|
|
{engine}
|
||
|
|
</Flex>
|
||
|
|
}
|
||
|
|
action={() => search(src, Engines[engine])}
|
||
|
|
/>
|
||
|
|
);
|
||
|
|
})}
|
||
|
|
</Menu.MenuItem>
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
const messageContextMenuPatch: NavContextMenuPatchCallback = (children, _props) => {
|
||
|
|
const selection = document.getSelection()?.toString();
|
||
|
|
if (!selection) return;
|
||
|
|
|
||
|
|
const group = findGroupChildrenByChildId("search-google", children);
|
||
|
|
if (group) {
|
||
|
|
const idx = group.findIndex(c => c?.props?.id === "search-google");
|
||
|
|
if (idx !== -1) group[idx] = makeSearchItem(selection);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
export default definePlugin({
|
||
|
|
name: "ReplaceGoogleSearch",
|
||
|
|
description: "Replaces the Google search with different Engines",
|
||
|
|
authors: [Devs.Moxxie, Devs.Ethan],
|
||
|
|
|
||
|
|
settings,
|
||
|
|
|
||
|
|
contextMenus: {
|
||
|
|
"message": messageContextMenuPatch
|
||
|
|
}
|
||
|
|
});
|