internal: add db operations to api

cmd/geth/consolecmd_test.go

@@ -30,7 +30,7 @@ import (
 )
 
 const (
-	ipcAPIs  = "admin:1.0 debug:1.0 eth:1.0 ethash:1.0 miner:1.0 net:1.0 personal:1.0 rpc:1.0 txpool:1.0 web3:1.0"
+	ipcAPIs  = "admin:1.0 db:1.0 debug:1.0 eth:1.0 ethash:1.0 miner:1.0 net:1.0 personal:1.0 rpc:1.0 txpool:1.0 web3:1.0"
 	httpAPIs = "eth:1.0 net:1.0 rpc:1.0 web3:1.0"
 )
 

internal/ethapi/api.go

@@ -1998,6 +1998,51 @@ func (s *PublicNetAPI) Version() string {
 	return fmt.Sprintf("%d", s.networkVersion)
 }
 
+// PublicDbAPI exposes low-level methods for interacting with the database.
+type PublicDbAPI struct {
+	b Backend
+}
+
+// NewPublicDbAPI creates a new instance of PublicDbAPI.
+func NewPublicDbAPI(b Backend) *PublicDbAPI {
+	return &PublicDbAPI{b: b}
+}
+
+// Get returns the raw value of a key stored in the database.
+func (api *PublicDbAPI) Get(key string) (hexutil.Bytes, error) {
+	blob, err := parseHexOrString(key)
+	if err != nil {
+		return nil, err
+	}
+	return api.b.ChainDb().Get(blob)
+}
+
+// Put stores a raw value under the key in the database.
+func (api *PublicDbAPI) Put(key string, value hexutil.Bytes) (hexutil.Bytes, error) {
+	var (
+		ret []byte
+		db  = api.b.ChainDb()
+	)
+	blob, err := parseHexOrString(key)
+	if err != nil {
+		return nil, err
+	}
+	data, err := db.Get(blob)
+	if err == nil {
+		ret = data
+	}
+	return ret, db.Put(blob, value)
+}
+
+// Delete removes an item from the database.
+func (api *PublicDbAPI) Delete(key string) error {
+	blob, err := parseHexOrString(key)
+	if err != nil {
+		return err
+	}
+	return api.b.ChainDb().Delete(blob)
+}
+
 // checkTxFee is an internal function used to check whether the fee of
 // the given transaction is _reasonable_(under the cap).
 func checkTxFee(gasPrice *big.Int, gas uint64, cap float64) error {
@@ -2021,3 +2066,12 @@ func toHexSlice(b [][]byte) []string {
 	}
 	return r
 }
+
+// ParseHexOrString tries to hexdecode b, but if the prefix is missing, it instead just returns the raw bytes
+func parseHexOrString(str string) ([]byte, error) {
+	b, err := hexutil.Decode(str)
+	if errors.Is(err, hexutil.ErrMissingPrefix) {
+		return []byte(str), nil
+	}
+	return b, err
+}

internal/ethapi/backend.go

@@ -137,6 +137,11 @@ func GetAPIs(apiBackend Backend) []rpc.API {
 			Version:   "1.0",
 			Service:   NewPrivateAccountAPI(apiBackend, nonceLock),
 			Public:    false,
+		}, {
+			Namespace: "db",
+			Version:   "1.0",
+			Service:   NewPublicDbAPI(apiBackend),
+			Public:    true,
 		},
 	}
 }

internal/web3ext/web3ext.go

@@ -30,6 +30,7 @@ var Modules = map[string]string{
 	"txpool":   TxpoolJs,
 	"les":      LESJs,
 	"vflux":    VfluxJs,
+	"db":       DbJS,
 }
 
 const CliqueJs = `
@@ -853,3 +854,28 @@ web3._extend({
 	]
 });
 `
+
+const DbJS = `
+web3._extend({
+	property: 'db',
+	methods:
+	[
+		new web3._extend.Method({
+			name: 'get',
+			call: 'db_get',
+			params: 1
+		}),
+		new web3._extend.Method({
+			name: 'put',
+			call: 'db_put',
+			params: 2
+		}),
+		new web3._extend.Method({
+			name: 'delete',
+			call: 'db_delete',
+			params: 1
+		}),
+	],
+	properties: []
+});
+`