forked from SpeedyCoder/go-onfido
/
document.go
209 lines (175 loc) · 6.04 KB
/
document.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
package onfido
import (
"bytes"
"context"
"encoding/json"
"fmt"
"io"
"mime/multipart"
"net/http"
"net/textproto"
//"os"
"strings"
"time"
)
// Supported document types and sides
const (
DocumentTypeUnknown DocumentType = "unknown"
DocumentTypePassport DocumentType = "passport"
DocumentTypeIDCard DocumentType = "national_identity_card"
DocumentTypeDrivingLicense DocumentType = "driving_licence"
DocumentTypeUKBRP DocumentType = "uk_biometric_residence_permit"
DocumentTypeTaxID DocumentType = "tax_id"
DocumentTypeVoterID DocumentType = "voter_id"
DocumentTypeBankStatement DocumentType = "bank_statement"
DocumentSideFront DocumentSide = "front"
DocumentSideBack DocumentSide = "back"
)
// DocumentType represents a document type (passport, ID, etc)
type DocumentType string
// DocumentSide represents a document side (front, back)
type DocumentSide string
// DocumentRequest represents a document request to Onfido API
type DocumentRequest struct {
File io.ReadSeeker `json:"file,omitempty"`
Type DocumentType `json:"type,omitempty"`
Side DocumentSide `json:"side,omitempty"`
IssuingCountry string `json:"issuing_country,omitempty"`
}
// Document represents a document in Onfido API
type Document struct {
ID string `json:"id,omitempty"`
CreatedAt *time.Time `json:"created_at,omitempty"`
Href string `json:"href,omitempty"`
DownloadHref string `json:"download_href,omitempty"`
FileName string `json:"file_name,omitempty"`
FileType string `json:"file_type,omitempty"`
FileSize int `json:"file_size,omitempty"`
Type DocumentType `json:"type,omitempty"`
Side DocumentSide `json:"side,omitempty"`
IssuingCountry string `json:"issuing_country,omitempty"`
// Messages []string `json:"messages,omitempty"`
}
// Documents represents a list of documents from the Onfido API
type Documents struct {
Documents []*Document `json:"documents"`
}
// Document content represents the content of the document download from Onfido API
type DocumentDownload struct {
Size int
Content []byte
}
func (d *DocumentDownload) Write(data []byte) (n int, err error) {
d.Content = data
d.Size = len(data)
return len(data), nil
}
var quoteEscaper = strings.NewReplacer("\\", "\\\\", `"`, "\\\"")
func escapeQuotes(s string) string {
return quoteEscaper.Replace(s)
}
// createFormFile creates a new form-data header with the provided field name,
// file name, and file content type.
// this is used instead of multipart.Writer.CreateFormFile because Onfido API
// doesn't accept 'application/octet-stream' as content-type.
func createFormFile(writer *multipart.Writer, fieldname string, file io.ReadSeeker, filename string) (io.Writer, error) {
buffer := make([]byte, 512)
if _, err := file.Read(buffer); err != nil {
return nil, err
}
if _, err := file.Seek(0, 0); err != nil {
return nil, err
}
// var filename string
// if f, ok := file.(*os.File); ok {
// filename = f.Name()
// }
//filename = "foo.jpg"
h := make(textproto.MIMEHeader)
h.Set("Content-Disposition",
fmt.Sprintf(`form-data; name="%s"; filename="%s"`,
escapeQuotes(fieldname), escapeQuotes(filename)))
h.Set("Content-Type", http.DetectContentType(buffer))
//fmt.Printf("Header is %+v\n", h)
return writer.CreatePart(h)
}
// UploadDocument uploads a document for the provided applicant.
// see https://documentation.onfido.com/?shell#upload-document
func (c *Client) UploadDocument(ctx context.Context, applicantID string, dr DocumentRequest, filename string) (*Document, error) {
body := &bytes.Buffer{}
writer := multipart.NewWriter(body)
part, err := createFormFile(writer, "file", dr.File, filename)
if err != nil {
return nil, err
}
if _, err := io.Copy(part, dr.File); err != nil {
return nil, err
}
if err := writer.WriteField("type", string(dr.Type)); err != nil {
return nil, err
}
if err := writer.WriteField("side", string(dr.Side)); err != nil {
return nil, err
}
if err := writer.WriteField("issuing_country", string(dr.IssuingCountry)); err != nil {
return nil, err
}
if err := writer.Close(); err != nil {
return nil, err
}
req, err := c.newRequest("POST", "/applicants/"+applicantID+"/documents", body)
req.Header.Set("Content-Type", writer.FormDataContentType())
if err != nil {
return nil, err
}
var resp Document
_, err = c.do(ctx, req, &resp)
return &resp, err
}
// GetDocument retrieves a single document for the provided applicant by its ID.
// see https://documentation.onfido.com/?shell#retrieve-document
func (c *Client) GetDocument(ctx context.Context, applicantID, id string) (*Document, error) {
req, err := c.newRequest("GET", "/applicants/"+applicantID+"/documents/"+id, nil)
if err != nil {
return nil, err
}
var resp Document
_, err = c.do(ctx, req, &resp)
return &resp, err
}
func (c *Client) DownloadDocument(ctx context.Context, applicantID, id string) (*DocumentDownload, error) {
req, err := c.newRequest("GET", "/applicants/"+applicantID+"/documents/"+id+"/download", nil)
if err != nil {
return nil, err
}
blob, err := c.download(ctx, req)
return &DocumentDownload{Content: blob, Size: len(blob)}, err
}
// DocumentIter represents a document iterator
type DocumentIter struct {
*iter
}
// Document returns the current item in the iterator as a Document.
func (i *DocumentIter) Document() *Document {
return i.Current().(*Document)
}
// ListDocuments retrieves the list of documents for the provided applicant.
// see https://documentation.onfido.com/?shell#list-documents
func (c *Client) ListDocuments(applicantID string) *DocumentIter {
handler := func(body []byte) ([]interface{}, error) {
var d Documents
if err := json.Unmarshal(body, &d); err != nil {
return nil, err
}
values := make([]interface{}, len(d.Documents))
for i, v := range d.Documents {
values[i] = v
}
return values, nil
}
return &DocumentIter{&iter{
c: c,
nextURL: "/applicants/" + applicantID + "/documents",
handler: handler,
}}
}