1 files changed, 9 insertions, 1 deletions
diff --git a/data.go b/data.go
index df85090..140f914 100644
--- a/data.go
+++ b/data.go
@@ -2,7 +2,9 @@ package main
 
 import (
 	"encoding/json"
+	"errors"
 	"fmt"
+	"github.com/BurntSushi/toml"
 	"github.com/anhgelus/golatt"
 	"html/template"
 	"log/slog"
@@ -125,7 +127,13 @@ func (c *Config) LoadCustomPages() ([]*CustomPage, error) {
 			return nil, err
 		}
 		var p CustomPage
-		err = json.Unmarshal(b, &p)
+		if strings.HasSuffix(cp, ".json") {
+			err = json.Unmarshal(b, &p)
+		} else if strings.HasSuffix(cp, ".toml") {
+			err = toml.Unmarshal(b, &p)
+		} else {
+			return nil, errors.New("custom page file must be .json or .toml")
+		}
 		if err != nil {
 			return nil, err
 		}