feat: saved configs for generators

database/badger/badger.go

@@ -211,3 +211,24 @@ func (b *Badger) GetSources() ([]database.DataSourceEntry, error) {
 
 	return sources, nil
 }
+
+func (b *Badger) GetKey(key string) (string, error) {
+	var value string
+	err := b.db.View(func(txn *badger.Txn) error {
+		item, err := txn.Get([]byte("KV_" + key))
+		if err != nil {
+			return err
+		}
+		return item.Value(func(val []byte) error {
+			value = string(val)
+			return nil
+		})
+	})
+	return value, err
+}
+
+func (b *Badger) SetKey(key string, value string) error {
+	return b.db.Update(func(txn *badger.Txn) error {
+		return txn.Set([]byte("KV_"+key), []byte(value))
+	})
+}

database/cloud/cloud.go

@@ -4,13 +4,14 @@ import (
 	"bytes"
 	"encoding/json"
 	"errors"
-	"github.com/BigJk/snd"
-	"github.com/BigJk/snd/database"
-	"github.com/BigJk/snd/log"
 	"io"
 	"net/http"
 	"net/url"
 	"time"
+
+	"github.com/BigJk/snd"
+	"github.com/BigJk/snd/database"
+	"github.com/BigJk/snd/log"
 )
 
 type Cloud struct {
@@ -217,3 +218,11 @@ func (c *Cloud) GetSources() ([]database.DataSourceEntry, error) {
 	_, err := c.request(http.MethodGet, "/api/sources", nil, &sources)
 	return sources, err
 }
+
+func (c *Cloud) GetKey(key string) (string, error) {
+	return "", errors.New("not implemented")
+}
+
+func (c *Cloud) SetKey(key string, value string) error {
+	return errors.New("not implemented")
+}

database/database.go

@@ -50,4 +50,7 @@ type Database interface {
 	DeleteSource(id string) error
 	GetSource(id string) (snd.DataSource, error)
 	GetSources() ([]DataSourceEntry, error)
+
+	GetKey(key string) (string, error)
+	SetKey(key string, value string) error
 }

database/memory/memory.go

@@ -12,6 +12,7 @@ type Memory struct {
 	entries    map[string]map[string]snd.Entry
 	generators map[string]snd.Generator
 	sources    map[string]snd.DataSource
+	kv         map[string]string
 }
 
 func New() *Memory {
@@ -159,3 +160,12 @@ func (m *Memory) GetSources() ([]database.DataSourceEntry, error) {
 	}
 	return sources, nil
 }
+
+func (m *Memory) GetKey(key string) (string, error) {
+	return m.kv[key], nil
+}
+
+func (m *Memory) SetKey(key string, value string) error {
+	m.kv[key] = value
+	return nil
+}

database/storm/storm.go

@@ -2,6 +2,7 @@ package storm
 
 import (
 	"bytes"
+	"errors"
 	"time"
 
 	"github.com/BigJk/snd"
@@ -185,3 +186,11 @@ func (s *Storm) GetSources() ([]database.DataSourceEntry, error) {
 
 	return sources, nil
 }
+
+func (s *Storm) GetKey(key string) (string, error) {
+	return "", errors.New("not implemented")
+}
+
+func (s *Storm) SetKey(key, value string) error {
+	return errors.New("not implemented")
+}

frontend/src/js/core/api.ts

@@ -27,6 +27,10 @@ export const SAVE_SETTINGS = 'saveSettings';
 export const GET_VERSION = 'getVersion';
 export const NEW_VERSION = 'newVersion';
 
+// Key Value functions
+export const GET_KEY = 'getKey';
+export const SET_KEY = 'setKey';
+
 // Entries functions
 export const GET_ENTRIES = 'getEntries';
 export const SAVE_ENTRY = 'saveEntry';

frontend/src/js/ui/components/modals/prompt.ts

@@ -0,0 +1,57 @@
+import m from 'mithril';
+
+import Button from 'js/ui/shoelace/button';
+import Input from 'js/ui/shoelace/input';
+import Modal from 'js/ui/shoelace/modal';
+
+import HorizontalProperty from 'js/ui/components/horizontal-property';
+
+import { popPortal, pushPortal } from 'js/ui/portal';
+
+type PromptProps = {
+	title: string;
+	label: string;
+	description: string;
+	buttonText?: string;
+	onSuccess: (answer: string) => void;
+};
+
+const promptModal = (props: PromptProps) => (): m.Component => {
+	let state = '';
+
+	return {
+		view: () =>
+			m(
+				Modal,
+				{ title: props.title, onClose: () => popPortal() },
+				m('div', [
+					m(
+						HorizontalProperty,
+						{
+							label: props.label,
+							description: props.description,
+							bottomBorder: false,
+							centered: true,
+						},
+						m(Input, { value: state, placeholder: '', onChange: (val: string) => (state = val) }),
+					),
+					m(
+						Button,
+						{
+							intend: 'success',
+							className: '.mt2',
+							onClick: () => {
+								props.onSuccess(state);
+								popPortal();
+							},
+						},
+						props.buttonText || 'Submit',
+					),
+				]),
+			),
+	};
+};
+
+export function openPromptModal(props: PromptProps) {
+	pushPortal(promptModal(props));
+}

frontend/src/js/ui/views/generator/single.ts

@@ -1,4 +1,8 @@
 import m from 'mithril';
+import { cloneDeep, map } from 'lodash-es';
+
+import HorizontalProperty from '../../components/horizontal-property';
+import { openPromptModal } from '../../components/modals/prompt';
 
 import { buildId } from 'js/types/basic-info';
 import Generator, { sanitizeConfig } from 'js/types/generator';
@@ -34,6 +38,7 @@ type SingleGeneratorState = {
 	printCount: number;
 	lastRendered: string;
 	hidePreview: boolean;
+	savedConfigs: Record<string, any[]>;
 };
 
 export default (): m.Component<SingleGeneratorProps> => {
@@ -43,6 +48,7 @@ export default (): m.Component<SingleGeneratorProps> => {
 		printCount: 1,
 		lastRendered: '',
 		hidePreview: false,
+		savedConfigs: {},
 	};
 
 	const print = () => {
@@ -95,12 +101,51 @@ export default (): m.Component<SingleGeneratorProps> => {
 		});
 	};
 
+	const saveConfigs = () => {
+		if (!state.generator) {
+			return;
+		}
+		return API.exec<void>(API.SET_KEY, `${buildId('generator', state.generator)}_saved_configs`, JSON.stringify(state.savedConfigs));
+	};
+
+	const saveConfig = () => {
+		openPromptModal({
+			title: 'Save Config',
+			label: 'Name',
+			description: 'Enter a name for the config',
+			onSuccess: (name) => {
+				if (!state.generator) return;
+				state.savedConfigs[name] = cloneDeep(state.config);
+				saveConfigs()?.catch(error);
+				success('Saved config');
+			},
+		});
+	};
+
+	const deleteSavedConfig = (name: string) => {
+		dialogWarning('Are you sure you want to delete this config?').then(() => {
+			if (!state.generator) return;
+			delete state.savedConfigs[name];
+			saveConfigs()?.catch(error);
+		});
+	};
+
+	const loadSavedConfig = (name: string) => {
+		if (!state.generator) return;
+		state.config = cloneDeep(state.savedConfigs[name]);
+	};
+
 	return {
 		oninit({ attrs }) {
 			API.exec<Generator>(API.GET_GENERATOR, attrs.id).then((generator) => {
 				state.generator = generator;
 				state.config = sanitizeConfig(generator, {});
 			});
+			API.exec<string>(API.GET_KEY, `${attrs.id}_saved_configs`)
+				.then((configs) => {
+					state.savedConfigs = JSON.parse(configs);
+				})
+				.catch(console.error);
 		},
 		onupdate({ attrs }) {},
 		view({ attrs }) {
@@ -187,7 +232,7 @@ export default (): m.Component<SingleGeneratorProps> => {
 										m.redraw();
 									},
 								}),
-								m(Flex, { className: '.bt.b--black-10.pv2.ph3', justify: 'end', items: 'center', gap: 2 }, [
+								m(Flex, { className: '.bt.b--black-10.pv2.ph3', justify: 'between', items: 'center', gap: 2 }, [
 									/*
 									Disabled for now!
 
@@ -203,9 +248,18 @@ export default (): m.Component<SingleGeneratorProps> => {
 										}),
 									),
 									*/
-									m(Tooltip, { content: 'Hide Preview' }, m(Checkbox, { checked: state.hidePreview, onChange: (val) => (state.hidePreview = val) })),
-									m(IconButton, { icon: 'camera', intend: 'primary', onClick: screenshot }, 'Screenshot'),
-									m(IconButton, { icon: 'print', intend: 'success', onClick: print }, 'Print'),
+									m(Flex, { gap: 2 }, [
+										m(IconButton, { icon: 'save', intend: 'primary', onClick: saveConfig }, 'Save Config'),
+										m(IconButton, { icon: 'camera', intend: 'primary', onClick: screenshot }, 'Screenshot'),
+									]),
+									m(Flex, { gap: 2, items: 'center' }, [
+										m(
+											Tooltip,
+											{ content: 'Hide Preview' },
+											m(Checkbox, { checked: state.hidePreview, onChange: (val) => (state.hidePreview = val) }),
+										),
+										m(IconButton, { icon: 'print', intend: 'success', onClick: print }, 'Print'),
+									]),
 								]),
 							]),
 						Information: () =>
@@ -216,7 +270,36 @@ export default (): m.Component<SingleGeneratorProps> => {
 									? [m('div.f5.mb2.b.mt3', 'Copyright Notice'), m('div', { style: { whiteSpace: 'break-spaces' } }, state.generator.copyrightNotice)]
 									: []),
 							]),
-						Saved: () => m('div.ph3.pv2', 'coming soon...'),
+						Saved: () =>
+							m('div.ph3.pv2.lh-copy', [
+								m('div.f5.b', 'Saved Configs'),
+								Object.keys(state.savedConfigs).length
+									? m(Flex, { direction: 'column' }, [
+											...map(state.savedConfigs, (config, key) =>
+												m(
+													HorizontalProperty,
+													{
+														label: key,
+														description: '',
+														bottomBorder: true,
+														centered: true,
+													},
+													m(
+														Flex,
+														{
+															gap: 2,
+															justify: 'end',
+														},
+														[
+															m(IconButton, { icon: 'trash', intend: 'error', onClick: () => deleteSavedConfig(key) }),
+															m(IconButton, { icon: 'edit', intend: 'primary', onClick: () => loadSavedConfig(key) }, 'Load'),
+														],
+													),
+												),
+											),
+									  ])
+									: m('div.pv2.text-muted', 'No saved configs yet...'),
+							]),
 					},
 				}),
 			);

rpc/kv.go

@@ -0,0 +1,12 @@
+package rpc
+
+import (
+	"github.com/BigJk/snd/database"
+	"github.com/BigJk/snd/rpc/bind"
+	"github.com/labstack/echo/v4"
+)
+
+func RegisterKeyValue(route *echo.Group, db database.Database) {
+	bind.MustBind(route, "/getKey", db.GetKey)
+	bind.MustBind(route, "/setKey", db.SetKey)
+}

server/server.go

@@ -5,7 +5,6 @@ import (
 	"context"
 	"errors"
 	"fmt"
-	"github.com/BigJk/snd/rpc/bind"
 	"io/ioutil"
 	"mime"
 	"net/http"
@@ -14,6 +13,8 @@ import (
 	"sync"
 	"time"
 
+	"github.com/BigJk/snd/rpc/bind"
+
 	"github.com/BigJk/snd/database"
 	"github.com/labstack/echo/v4/middleware"
 	"github.com/patrickmn/go-cache"
@@ -142,6 +143,7 @@ func (s *Server) Start(bindAddr string) error {
 	extern := api.Group("/extern")
 
 	rpc.RegisterVersion(api)
+	rpc.RegisterKeyValue(api, s.db)
 	rpc.RegisterImageUtilities(api)
 	rpc.RegisterSettings(api, s.db)
 	rpc.RegisterTemplate(api, extern, s.db)