ocr_request.go

package ocrworker

import (
	"encoding/base64"
	"fmt"
)

type OcrRequest struct {
	ImgUrl            string                 `json:"img_url"`
	ImgBase64         string                 `json:"img_base64"`
	EngineType        OcrEngineType          `json:"engine"`
	ImgBytes          []byte                 `json:"img_bytes"`
	PreprocessorChain []string               `json:"preprocessors"`
	PreprocessorArgs  map[string]interface{} `json:"preprocessor-args"`
	EngineArgs        map[string]interface{} `json:"engine_args"`
	Deferred          bool                   `json:"deferred"`
	ReplyTo           string                 `json:"reply_to"`
	DocType           string                 `json:"doc_type"`
	RequestID         string                 `json:"req_id"`
	PageNumber        uint16                 `json:"page_number"`
	UserAgent         string                 `json:"user_agent"`
	TimeOut           uint                   `json:"time_out"`
	ReferenceID       string                 `json:"reference_id"`
	// decode ocr in http handler rather than putting in queue
	InplaceDecode bool `json:"inplace_decode"`
}

// figure out the next pre-processor routing key to use (if any).
// if we have finished with the pre-processors, then use the processorRoutingKey
func (ocrRequest *OcrRequest) nextPreprocessor(processorRoutingKey string) string {
	if len(ocrRequest.PreprocessorChain) == 0 {
		return processorRoutingKey
	} else {
		var x string
		s := ocrRequest.PreprocessorChain
		x, s = s[len(s)-1], s[:len(s)-1]
		ocrRequest.PreprocessorChain = s
		return x
	}
}

func (ocrRequest *OcrRequest) decodeBase64() error {
	bytes, decodeError := base64.StdEncoding.DecodeString(ocrRequest.ImgBase64)

	if decodeError != nil {
		return decodeError
	}

	ocrRequest.ImgBytes = bytes
	ocrRequest.ImgBase64 = ""

	return nil
}

func (ocrRequest *OcrRequest) hasBase64() bool {
	return ocrRequest.ImgBase64 != ""
}

func (ocrRequest *OcrRequest) downloadImgUrl() error {
	bytes, err := url2bytes(ocrRequest.ImgUrl)
	if err != nil {
		return err
	}
	ocrRequest.ImgBytes = bytes
	ocrRequest.ImgUrl = ""
	return nil
}

func (ocrRequest *OcrRequest) String() string {
	return fmt.Sprintf("ImgUrl: %s, EngineType: %s, Preprocessors: %s, Request ID: %s", ocrRequest.ImgUrl, ocrRequest.EngineType, ocrRequest.PreprocessorChain, ocrRequest.RequestID)
}