Skip to content

Commit d11bd55

Browse files
committed
Moved methods.
1 parent 709fcfa commit d11bd55

File tree

2 files changed

+45
-51
lines changed

2 files changed

+45
-51
lines changed

services/lfs/locks.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ func GetListLockHandler(ctx *context.Context) {
6565
}
6666
ctx.Resp.Header().Set("Content-Type", lfs_module.MediaType)
6767

68-
rv, _ := unpack(ctx)
68+
rv := getRequestContext(ctx)
6969

7070
repository, err := models.GetRepositoryByOwnerAndName(rv.User, rv.Repo)
7171
if err != nil {

services/lfs/server.go

Lines changed: 44 additions & 50 deletions
Original file line numberDiff line numberDiff line change
@@ -65,49 +65,12 @@ func CheckAcceptMediaType(ctx *context.Context) {
6565
}
6666
}
6767

68-
func getAuthenticatedRepoAndMeta(ctx *context.Context, rc *requestContext, p lfs_module.Pointer, requireWrite bool) (*models.LFSMetaObject, *models.Repository) {
69-
if !p.IsValid() {
70-
log.Info("Attempt to access invalid LFS OID[%s] in %s/%s", p.Oid, rc.User, rc.Repo)
71-
writeStatus(ctx, http.StatusUnprocessableEntity)
72-
return nil, nil
73-
}
74-
75-
repository := getAuthenticatedRepository(ctx, rc, requireWrite)
76-
if repository == nil {
77-
return nil, nil
78-
}
79-
80-
meta, err := repository.GetLFSMetaObjectByOid(p.Oid)
81-
if err != nil {
82-
log.Error("Unable to get LFS OID[%s] Error: %v", p.Oid, err)
83-
writeStatus(ctx, http.StatusNotFound)
84-
return nil, nil
85-
}
86-
87-
return meta, repository
88-
}
89-
90-
func getAuthenticatedRepository(ctx *context.Context, rc *requestContext, requireWrite bool) *models.Repository {
91-
repository, err := models.GetRepositoryByOwnerAndName(rc.User, rc.Repo)
92-
if err != nil {
93-
log.Error("Unable to get repository: %s/%s Error: %v", rc.User, rc.Repo, err)
94-
writeStatus(ctx, http.StatusNotFound)
95-
return nil
96-
}
97-
98-
if !authenticate(ctx, repository, rc.Authorization, requireWrite) {
99-
requireAuth(ctx)
100-
return nil
101-
}
102-
103-
return repository
104-
}
105-
10668
// DownloadHandler gets the content from the content store
10769
func DownloadHandler(ctx *context.Context) {
108-
rc, p := unpack(ctx)
70+
rc := getRequestContext(ctx)
71+
p := lfs_module.Pointer{Oid: ctx.Params("oid")}
10972

110-
meta, _ := getAuthenticatedRepoAndMeta(ctx, rc, p, false)
73+
meta := getAuthenticatedMeta(ctx, rc, p, false)
11174
if meta == nil {
11275
return
11376
}
@@ -253,7 +216,7 @@ func BatchHandler(ctx *context.Context) {
253216
if meta == nil {
254217
_, err := models.NewLFSMetaObject(&models.LFSMetaObject{Pointer: p, RepositoryID: repository.ID})
255218
if err != nil {
256-
log.Error("Unable to create LFS MetaObject [%s] for %s/%s. Error: %v", p.Oid, rc.User, rc.Repo, metaErr)
219+
log.Error("Unable to create LFS MetaObject [%s] for %s/%s. Error: %v", p.Oid, rc.User, rc.Repo, err)
257220
writeStatus(ctx, http.StatusInternalServerError)
258221
return
259222
}
@@ -322,7 +285,7 @@ func UploadHandler(ctx *context.Context) {
322285
writeStatus(ctx, http.StatusInternalServerError)
323286
return
324287
}
325-
if m.Existing || exisits {
288+
if meta.Existing || exists {
326289
ctx.Resp.WriteHeader(http.StatusOK)
327290
return
328291
}
@@ -351,7 +314,7 @@ func VerifyHandler(ctx *context.Context) {
351314

352315
rc := getRequestContext(ctx)
353316

354-
meta, _ := getAuthenticatedRepoAndMeta(ctx, rc, p, true)
317+
meta := getAuthenticatedMeta(ctx, rc, p, true)
355318
if meta == nil {
356319
return
357320
}
@@ -385,6 +348,44 @@ func getRequestContext(ctx *context.Context) *requestContext {
385348
}
386349
}
387350

351+
func getAuthenticatedMeta(ctx *context.Context, rc *requestContext, p lfs_module.Pointer, requireWrite bool) *models.LFSMetaObject {
352+
if !p.IsValid() {
353+
log.Info("Attempt to access invalid LFS OID[%s] in %s/%s", p.Oid, rc.User, rc.Repo)
354+
writeStatus(ctx, http.StatusUnprocessableEntity)
355+
return nil
356+
}
357+
358+
repository := getAuthenticatedRepository(ctx, rc, requireWrite)
359+
if repository == nil {
360+
return nil
361+
}
362+
363+
meta, err := repository.GetLFSMetaObjectByOid(p.Oid)
364+
if err != nil {
365+
log.Error("Unable to get LFS OID[%s] Error: %v", p.Oid, err)
366+
writeStatus(ctx, http.StatusNotFound)
367+
return nil
368+
}
369+
370+
return meta
371+
}
372+
373+
func getAuthenticatedRepository(ctx *context.Context, rc *requestContext, requireWrite bool) *models.Repository {
374+
repository, err := models.GetRepositoryByOwnerAndName(rc.User, rc.Repo)
375+
if err != nil {
376+
log.Error("Unable to get repository: %s/%s Error: %v", rc.User, rc.Repo, err)
377+
writeStatus(ctx, http.StatusNotFound)
378+
return nil
379+
}
380+
381+
if !authenticate(ctx, repository, rc.Authorization, requireWrite) {
382+
requireAuth(ctx)
383+
return nil
384+
}
385+
386+
return repository
387+
}
388+
388389
func buildObjectResponse(rc *requestContext, pointer lfs_module.Pointer, download, upload bool, err *lfs_module.ObjectError) *lfs_module.ObjectResponse {
389390
rep := &lfs_module.ObjectResponse{Pointer: pointer}
390391
if err != nil {
@@ -411,13 +412,6 @@ func buildObjectResponse(rc *requestContext, pointer lfs_module.Pointer, downloa
411412
return rep
412413
}
413414

414-
func unpack(ctx *context.Context) (*requestContext, lfs_module.Pointer) {
415-
rc := getRequestContext(ctx)
416-
p := lfs_module.Pointer{Oid: ctx.Params("oid")}
417-
418-
return rc, p
419-
}
420-
421415
func writeStatus(ctx *context.Context, status int) {
422416
writeStatusMessage(ctx, status, http.StatusText(status))
423417
}

0 commit comments

Comments
 (0)