aboutsummaryrefslogtreecommitdiff
path: root/files/sync-media
blob: 64e3f02cb0449aa6b35111b0b24c15102473c9e5 (plain)
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
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
#!/bin/bash
#
# sync-media assets using git-annex
#

REMOTE="$1"
VOLUME="/media/$REMOTE"
DOMAIN="`facter domain`"
HOST="`facter hostname`"
CACHE="/var/cache/$HOST/media"
MEDIA="media.$DOMAIN"
INCOMING="$CACHE/incoming"
WHOAMI="`whoami`"
OPTIONS="$*"

# Fix identity
function sync_media_identity {
  if [ -z "`git config --local user.email`" ] || [ -z "`git config --local user.name`" ]; then
    repo="$(basename `pwd`)"
    git config user.name "${repo^} Archive"
    git config user.email "$repo@localhost"
  fi
}

# Add files into the annex
function sync_media_add {
  git annex add .

  # Adding hidden files and symlinks, find version
  #find -name '.*' -exec git annex add {} \;
  #find -type l -exec git add {} \;

  # Adding hidden files and symlinks, git+for version
  #for file in `git status --porcelain -u | sed -e 's/?? //'`; do
  #  if [ -h "$file" ]; then
  #    git add $file
  #  else
  #    git annex add $file
  #  fi
  #done

  # Adding hidden files and symlinks, git+while version
  git status --porcelain -u | sed -e 's/?? //' | while read file; do
    if [ -h "$file" ]; then
      git add "$file"
    else
      git annex add "$file"
    fi
  done
}

# Set sudo config
if [ "$WHOAMI" != 'root' ]; then
  sudo="sudo"
else
  echo "Sorry, cannot run as root"
  exit 1
fi

# Set fsck config
if echo $OPTIONS | grep -q -- "--fsck"; then
  FSCK="true"
fi

# Set unused config
if echo $OPTIONS | grep -q -- "--dropunused"; then
  DROPUNUSED="true"
fi

# Commit changes
if [ -d "$CACHE" ]; then
  # Fix cache permissions
  #echo "Fixing $CACHE permissions..."
  #$sudo find $CACHE -type f -exec chmod 644 {} \;
  #$sudo find $CACHE -type d -exec chmod 755 {} \;

  # Fix incoming permissions
  if [ -d "$INCOMING" ]; then
    echo "Fixing $INCOMING permissions..."
    $sudo find $INCOMING -type f -exec chmod 664 {} \;
    $sudo find $INCOMING -type d -exec chmod 775 {} \;
    $sudo chown -R $WHOAMI.incoming $INCOMING
  fi

  # Add and update local repositories
  for folder in `ls $CACHE`; do
    if [ -d "$CACHE/$folder/.git/annex" ]; then
      if [ "`git -C $CACHE/$folder config sync-media.skip`" == "true" ]; then
        continue
      fi

      #if [ "`git -C $CACHE/$folder config sync-media.ready`" != "true" ]; then
      #  echo "Skipping $CACHE/$folder: not sync-media ready, please config your repo."
      #  continue
      #fi

      (
      cd $CACHE/$folder
      echo "Syncing $CACHE/$folder..."

      # If there is a playlists folder, make sure mpd user can write to it
      if [ -d "playlists" ]; then
        $sudo chmod 775 playlists
        $sudo chown -R mpd.audio playlists
        find playlists -type f -exec sudo chmod 664 {} \;
        find playlists -type d -exec sudo chmod 775 {} \;
      fi

      # Ensure we have a reference to the remote repository
      if [ -d "$VOLUME/$MEDIA/$folder/git/.annex" ]; then
        # Check remote
        if ! git remote | grep -q "^$REMOTE$"; then
          git remote add $REMOTE $VOLUME/$MEDIA/$folder
        fi
      fi

      sync_media_identity
      sync_media_add
      git annex sync

      # Control whether the repository should have a copy of everything
      if [ "`git -C $CACHE/$folder config sync-media.getall`" == "true" ]; then
        git annex get .
      fi

      if [ "$FSCK" == "true" ]; then
        git annex fsck --fast
      fi

      if [ "$DROPUNUSED" == "true" ]; then
        git annex unused
        git annex dropunused 1-1000
      fi
      )
    fi
  done

  if [ ! -z "$REMOTE" ]; then
    # Check storage media
    MOUNT="`mount | grep $VOLUME`"
    if [ ! -z "$MOUNT" ]; then
      DRIVE="$(basename `echo $MOUNT | awk '{ print $1 }'`)"
    fi

    if [ ! -z "$DRIVE" ]; then
      if [ ! -d "$VOLUME/$MEDIA" ]; then
        echo "Folder $VOLUME/$MEDIA does not exist..."
      else
        for folder in `ls $CACHE`; do
          if [ -d "$CACHE/$folder/.git/annex" ]; then
            if [ "`git -C $CACHE/$folder config sync-media.skip`" == "true" ]; then
              continue
            fi

            #if [ "`git -C $CACHE/$folder config sync-media.ready`" != "true" ]; then
            #  echo "Skipping $CACHE/$folder: not sync-media ready, please config your repo."
            #  continue
            #fi

            if [ ! -d "$VOLUME/$MEDIA/$folder" ]; then
              (
              cd $VOLUME/$MEDIA
              echo "Initializing $VOLUME/$MEDIA/$folder..."
              git clone $CACHE/$folder && cd $folder && sync_media_identity && git annex init $DRIVE && \
                git remote rename origin $HOST && cd $CACHE/$folder && git remote add $DRIVE $VOLUME/$MEDIA/$folder
              )
            fi
          elif [ ! -d "$VOLUME/$MEDIA/$folder" ]; then
            if [ ! -e "$CACHE/$folder/.sync-media/skip" ]; then
              echo "Syncing $VOLUME/$MEDIA/$folder..."
              rsync -av --delete --exclude=.sync-media $CACHE/$folder/ $VOLUME/$MEDIA/$folder/
            fi
          fi
        done
      fi
    fi
  fi
fi

# Retrieve changes at media volumes
if [ ! -z "$DRIVE" ] && [ -d "$VOLUME/$MEDIA" ]; then
  for folder in `ls $VOLUME/$MEDIA`; do
    if [ -d "$VOLUME/$MEDIA/$folder/.git/annex" ]; then
      if [ "`git -C $VOLUME/$MEDIA/$folder config sync-media.skip`" == "true" ]; then
        continue
      fi

      (
      cd $VOLUME/$MEDIA/$folder
      echo "Syncing $VOLUME/$MEDIA/$folder..."

      # Check remote
      if ! git remote | grep -q "^$HOST$"; then
        git remote add $HOST $CACHE/$folder
      fi

      # If there is a playlists folder, make sure mpd user can write to it
      if [ -d "playlists" ]; then
        $sudo chmod 775 playlists
        $sudo chown -R mpd.audio playlists
        find playlists -type f -exec sudo chmod 664 {} \;
        find playlists -type d -exec sudo chmod 775 {} \;
      fi

      sync_media_identity
      sync_media_add
      git annex sync
      git annex get . --numcopies=3

      if [ "$FSCK" == "true" ]; then
        git annex fsck --fast
      fi

      if [ "$DROPUNUSED" == "true" ]; then
        git annex unused
        git annex dropunused 1-1000
      fi

      #git annex drop --auto --numcopies=2
      )
    elif [ -d "$CACHE/$folder" ]; then
      if [ ! -e "$CACHE/$folder/.sync-media/skip" ]; then
        echo "Syncing $VOLUME/$MEDIA/$folder..."
        rsync -av --delete --exclude=.sync-media $CACHE/$folder/ $VOLUME/$MEDIA/$folder/
      fi
    fi
  done
elif [ ! -z "$REMOTE" ]; then
  # Try to copy to a remote
  for folder in `ls $CACHE`; do
    if [ -d "$CACHE/$folder/.git/annex" ]; then
      if [ "`git -C $CACHE/$folder config sync-media.skip`" == "true" ]; then
        continue
      fi

      if git -C $CACHE/$folder remote | grep -q "^$REMOTE$"; then
        (
        cd $CACHE/$folder
        git annex copy . --to $REMOTE
        git annex sync
        )
      else
        echo "No such remote $REMOTE"
      fi
    else
      if [ ! -e "$CACHE/$folder/.sync-media/skip" ]; then
        echo "Syncing $VOLUME/$MEDIA/$folder..."
        rsync -av --delete --exclude=.sync-media $CACHE/$folder/ $REMOTE.$DOMAIN:$CACHE/$folder/
      fi
    fi
  done
fi