github.com/go-kivik/kivik/v4@v4.3.2/pouchdb/db.go

// Licensed under the Apache License, Version 2.0 (the "License"); you may not
// use this file except in compliance with the License. You may obtain a copy of
// the License at
//
//  http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations under
// the License.

//go:build js

package pouchdb

import (
	"bytes"
	"context"
	"encoding/json"
	"io"
	"net/http"
	"sync/atomic"

	"github.com/gopherjs/gopherjs/js"

	"github.com/go-kivik/kivik/v4/driver"
	internal "github.com/go-kivik/kivik/v4/int/errors"
	"github.com/go-kivik/kivik/v4/pouchdb/bindings"
)

type db struct {
	db *bindings.DB

	client *client

	// these are set to 1 when compaction begins, and unset when the
	// callback returns.
	compacting  uint32
	viewCleanup uint32
}

var _ driver.DB = (*db)(nil)

func (d *db) AllDocs(ctx context.Context, options driver.Options) (driver.Rows, error) {
	opts := map[string]interface{}{}
	options.Apply(opts)
	result, err := d.db.AllDocs(ctx, opts)
	if err != nil {
		return nil, err
	}
	return &rows{
		Object: result,
	}, nil
}

func (d *db) Query(ctx context.Context, ddoc, view string, options driver.Options) (driver.Rows, error) {
	opts := map[string]interface{}{}
	options.Apply(opts)
	result, err := d.db.Query(ctx, ddoc, view, opts)
	if err != nil {
		return nil, err
	}
	return &rows{
		Object: result,
	}, nil
}

type document struct {
	id   string
	rev  string
	body io.Reader
	done bool
}

func (d *document) Next(row *driver.Row) error {
	if d.done {
		return io.EOF
	}
	d.done = true
	row.ID = d.id
	row.Rev = d.rev
	row.Doc = d.body
	return nil
}

func (d *document) Close() error {
	d.done = true
	return nil
}

func (d *document) UpdateSeq() string { return "" }
func (d *document) Offset() int64     { return 0 }
func (d *document) TotalRows() int64  { return 0 }

func (d *db) Get(ctx context.Context, docID string, options driver.Options) (*driver.Document, error) {
	opts := map[string]interface{}{}
	options.Apply(opts)
	doc, rev, err := d.db.Get(ctx, docID, opts)
	if err != nil {
		return nil, err
	}
	return &driver.Document{
		Rev:  rev,
		Body: io.NopCloser(bytes.NewReader(doc)),
	}, nil
}

func (d *db) CreateDoc(ctx context.Context, doc interface{}, options driver.Options) (docID, rev string, err error) {
	jsonDoc, err := json.Marshal(doc)
	if err != nil {
		return "", "", err
	}
	jsDoc := js.Global.Get("JSON").Call("parse", string(jsonDoc))
	opts := map[string]interface{}{}
	options.Apply(opts)
	return d.db.Post(ctx, jsDoc, opts)
}

func (d *db) Put(ctx context.Context, docID string, doc interface{}, options driver.Options) (rev string, err error) {
	jsonDoc, err := json.Marshal(doc)
	if err != nil {
		return "", err
	}
	jsDoc := js.Global.Get("JSON").Call("parse", string(jsonDoc))
	if id := jsDoc.Get("_id"); id != js.Undefined {
		if id.String() != docID {
			return "", &internal.Error{Status: http.StatusBadRequest, Message: "id argument must match _id field in document"}
		}
	}
	opts := map[string]interface{}{}
	options.Apply(opts)
	jsDoc.Set("_id", docID)
	return d.db.Put(ctx, jsDoc, opts)
}

func (d *db) Delete(ctx context.Context, docID string, options driver.Options) (newRev string, err error) {
	opts := map[string]interface{}{}
	options.Apply(opts)
	rev, _ := opts["rev"].(string)
	return d.db.Delete(ctx, docID, rev, opts)
}

func (d *db) Stats(ctx context.Context) (*driver.DBStats, error) {
	i, err := d.db.Info(ctx)
	return &driver.DBStats{
		Name:           i.Name,
		CompactRunning: atomic.LoadUint32(&d.compacting) == 1 || atomic.LoadUint32(&d.viewCleanup) == 1,
		DocCount:       i.DocCount,
		UpdateSeq:      i.UpdateSeq,
	}, err
}

func (d *db) Compact(context.Context) error {
	if atomic.LoadUint32(&d.compacting) == 1 {
		return &internal.Error{Status: http.StatusTooManyRequests, Message: "kivik: compaction already running"}
	}
	atomic.StoreUint32(&d.compacting, 1)
	defer atomic.StoreUint32(&d.compacting, 0)
	return d.db.Compact()
}

// CompactView is unimplemented for PouchDB.
func (d *db) CompactView(context.Context, string) error {
	return nil
}

func (d *db) ViewCleanup(context.Context) error {
	if atomic.LoadUint32(&d.viewCleanup) == 1 {
		return &internal.Error{Status: http.StatusTooManyRequests, Message: "kivik: view cleanup already running"}
	}
	atomic.StoreUint32(&d.viewCleanup, 1)
	defer atomic.StoreUint32(&d.viewCleanup, 0)
	return d.db.ViewCleanup()
}

func (d *db) Close() error {
	return d.db.Close()
}

var _ driver.Purger = &db{}

func (d *db) Purge(ctx context.Context, docRevMap map[string][]string) (*driver.PurgeResult, error) {
	result := &driver.PurgeResult{
		Purged: make(map[string][]string, len(docRevMap)),
	}
	for docID, revs := range docRevMap {
		for _, rev := range revs {
			delRevs, err := d.db.Purge(ctx, docID, rev)
			if err != nil {
				return result, err
			}
			if result.Purged[docID] == nil {
				result.Purged[docID] = delRevs
			} else {
				result.Purged[docID] = append(result.Purged[docID], delRevs...)
			}
		}
	}
	return result, nil
}